Our new X account is live! Follow @wizwand_team for updates
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
Autonomous DrivingNAVSIM v1 (test)
NC98.7
99
PlanningNAVSIM (navtest)
NC98.3
53
Autonomous Driving PlanningNAVSIM (navtest)
NC98.3
50
Autonomous DrivingNAVSIM (test)
PDMS86.5
34
End-to-end Autonomous DrivingBench2Drive
Driving Score59.95
27
Closed-loop Autonomous Driving PlanningNAVSIM v1 (test)
NC98.3
26
Closed-loop Autonomous DrivingBench2Drive
Driving Score (DS)59.95
21
Open-loop Autonomous Driving PlanningNAVSIM 1.0 (test)
NC98.3
19
End-to-end PlanningNAVSIM v1
NC0.984
17
Autonomous Driving PlanningNAVSIM v1
NC98.3
17
Showing 10 of 32 rows

Other info

Code

Follow for update