Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Hydra-MDP: End-to-end Multimodal Planning with Multi-target Hydra-Distillation

About

We propose Hydra-MDP, a novel paradigm employing multiple teachers in a teacher-student model. This approach uses knowledge distillation from both human and rule-based teachers to train the student model, which features a multi-head decoder to learn diverse trajectory candidates tailored to various evaluation metrics. With the knowledge of rule-based teachers, Hydra-MDP learns how the environment influences the planning in an end-to-end manner instead of resorting to non-differentiable post-processing. This method achieves the $1^{st}$ place in the Navsim challenge, demonstrating significant improvements in generalization across diverse driving environments and conditions. More details by visiting \url{https://github.com/NVlabs/Hydra-MDP}.

Zhenxin Li, Kailin Li, Shihao Wang, Shiyi Lan, Zhiding Yu, Yishen Ji, Zhiqi Li, Ziyue Zhu, Jan Kautz, Zuxuan Wu, Yu-Gang Jiang, Jose M. Alvarez• 2024

Related benchmarks

TaskDatasetResultRank
Closed-loop PlanningBench2Drive
Driving Score59.95
194
Autonomous Driving PlanningNAVSIM v1 (test)
NC98.4
151
Autonomous DrivingNAVSIM v1 (test)
NC98.7
147
Autonomous Driving PlanningNAVSIM v1
NC99.1
126
Autonomous Driving PlanningNAVSIM (navtest)
NC98.3
91
PlanningNAVSIM (navtest)
NC98.3
78
Autonomous Driving PlanningNAVSIM v2 (Navtest)
NC97.5
76
PlanningNAVSIM (test)
PDMS86.5
75
Closed-loop Autonomous DrivingBench2Drive
Driving Score (DS)59.95
74
Closed-loop Autonomous Driving PlanningNAVSIM v1 (test)
NC98.3
63
Showing 10 of 59 rows

Other info

Code

Follow for update