Search papers, labs, and topics across Lattice.
This paper introduces StableDrive, a novel approach to long-horizon planning in autonomous driving that utilizes Selective Momentum Memory (SMM) and Motion-Stage Training Scaffold (MSTS) to enhance the reliability of historical context and improve future motion predictions. By selectively controlling the influence of past planning states and training with a focus on motion stages, StableDrive significantly reduces collision rates and improves planning metrics across various benchmarks. The method achieves state-of-the-art performance on the nuScenes dataset, demonstrating a 23.3% reduction in average collision rates and other substantial improvements over previous methods.
StableDrive slashes collision rates by over 23% while redefining how historical context is leveraged in autonomous driving.
Reliable long-horizon planning remains a key challenge in end-to-end autonomous driving. By accounting for future motion evolution and potential consequences, it provides forward-looking guidance for safe and consistent driving in evolving traffic environments. Existing methods use historical planning states as temporal context. Self-generated history may become stale or conflict with the current motion stage, introducing unreliable priors. We propose StableDrive to address cross-cycle historical reliability and within-horizon motion-stage evolution. Selective Momentum Memory (SMM), implemented with a Mamba selective state-space operator, controls the influence of the preceding self-predicted planning state on the current cycle. Motion-Stage Training Scaffold (MSTS) uses motion-stage, long-horizon trajectory, and longitudinal-motion supervision to guide stage-aware future motion learning and is removed before inference. A fixed parameter midpoint between two architecture-aligned endpoints yields a single deployable SMM planner without model ensembling or extra inference-time computation. On nuScenes under the MomAD evaluation protocol, StableDrive achieves SOTA performance across all reported planning metrics from 1 to 6 s, reducing average collision rate by 23.3%, TPC by 30.9%, and L2 by 11.8% over the best previously reported value for each metric. On the curated Longitudinal-Transition nuScenes (LT-nuScenes), StableDrive reduces 6-s collision rate by 23.81%, TPC by 10.90%, and L2 by 6.37%. On NAVSIM v1 and v2, StableDrive achieves the highest PDMS/EPDMS in all three reported settings, including a 5.7-point EPDMS gain on v2 navhard over the previous best.
