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
152
Autonomous DrivingNAVSIM v1 (test)
NC98.7
147
Autonomous Driving PlanningNAVSIM v1
NC99.1
126
Autonomous Driving PlanningNAVSIM v1 (test)
NC98.4
118
Closed-loop Autonomous DrivingBench2Drive
Driving Score (DS)59.95
74
Autonomous Driving PlanningNAVSIM (navtest)
NC98.3
68
Autonomous DrivingNAVSIM (test)
PDMS86.5
62
End-to-end PlanningNAVSIM v1
NC98.4
61
PlanningNAVSIM (test)
PDMS86.5
59
PlanningNAVSIM (navtest)
NC98.3
53
Showing 10 of 54 rows

Other info

Code

Follow for update