CLAP: Contrastive Latent-space Prompt Optimization for End-to-end Autonomous Driving

TL;DR

CLAP is a novel framework that enhances end-to-end autonomous driving models' performance in rare, safety-critical scenarios by optimizing location-aware prompts using contrastive learning and crowdsourced data.

Contribution

The paper introduces CLAP, a location-aware prompt optimization method that improves autonomous driving in challenging scenarios without retraining the entire model.

Findings

Reduces challenging scenario planning error by 24% on NAVSIM benchmark.

Maintains performance on normal frames while improving difficult scenarios.

Uses contrastive learning to identify and optimize for hard scenes.

Abstract

End-to-end autonomous driving systems powered by Vision-Language-Action (VLA) models achieve strong performance on common driving scenarios, yet remain brittle in rare but safety-critical long-tail situations such as active construction zones and complex yielding geometries. In this paper, we present a method that addresses the long-tail challenging scenes beyond data scaling and model training. We introduce CLAP (Contrastive Latent-space Prompt optimization), a location-aware adaptation framework that augments a frozen VLA driving model with per-roadblock soft prompts, optimized from crowdsourced data and retrieved on demand via Vehicle-to-Everything (V2X) communication. Our approach rests on two observations from VLAs' latent space: (i) at the VLA's hidden-state layer, scenarios from the same roadblock cluster tightly and occupy compact regions of the latent space; and (ii) within a single roadblock, long-tail and normal frames are heavily intermixed in the latent representation, making it difficult to improve one without disturbing the other. CLAP addresses this via a two-stage pipeline: supervised contrastive learning to discover a roadblock-specific hard-scene direction, followed by directionally regularized prompt optimization that selectively improves challenging frames while preserving normal frame performance. On the NAVSIM benchmark with various state-of-the-art VLA backbones, CLAP reduces challenging scenario planning error by 24% with no regression on normal frames, significantly improving planning performance.