SAH-Drive: A Scenario-Aware Hybrid Planner for Closed-Loop Vehicle Trajectory Generation

Yuqi Fan, Zhiyong Cui, Zhenning Li, Yilong Ren, Haiyang Yu
2/14/2026

No official code found. Best implementation: richie-live/SAH-Drive (20 stars, MIT, last commit 2025-06-21).

Abstract

Reliable planning is crucial for achieving autonomous driving. Rule-based planners are efficient but lack generalization, while learning-based planners excel in generalization yet have limitations in real-time performance and interpretability. In long-tail scenarios, these challenges make planning particularly difficult. To leverage the strengths of both rule-based and learning-based planners, we proposed the Scenario-Aware Hybrid Planner (SAH-Drive) for closed-loop vehicle trajectory planning. Inspired by human driving behavior, SAH-Drive combines a lightweight rule-based planner and a comprehensive learning-based planner, utilizing a dual-timescale decision neuron to determine the final trajectory. To enhance the computational efficiency and robustness of the hybrid planner, we also employed a diffusion proposal number regulator and a trajectory fusion module. The experimental results show that the proposed method significantly improves the generalization capability of the planning system, achieving state-of-the-art performance in interPlan, while maintaining computational efficiency without incurring substantial additional runtime.

View on arXivView PDF

Code Implementations(1)

MIT

Code for ICML 2025: SAH-Drive: A Scenario-Aware Hybrid Planner for Closed-Loop Vehicle Trajectory Generation

203CSS, HTMLMay 25, 20251 years agoMIT

Similar Papers

Cite this paper

@article{fan2026sahdrive,
  title  = {SAH-Drive: A Scenario-Aware Hybrid Planner for Closed-Loop Vehicle Trajectory Generation},
  author = {Yuqi Fan and Zhiyong Cui and Zhenning Li and Yilong Ren and Haiyang Yu},
  year   = {2026},
  eprint = {2505.24390},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2505.24390},
  journal = {ICML 2025 2025}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.24390.svg)](https://codeofpaper.com/paper/2505.24390)

Discussion