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
Autonomous DrivingNAVSIM v1 (test)
NC98.7
113
Autonomous Driving PlanningNAVSIM v1
NC99.1
86
Autonomous Driving PlanningNAVSIM (navtest)
NC98.3
68
Autonomous Driving PlanningNAVSIM v1 (test)
NC98.4
59
PlanningNAVSIM (navtest)
NC98.3
53
Closed-loop Autonomous DrivingBench2Drive
Driving Score (DS)59.95
49
Autonomous DrivingNAVSIM (test)
PDMS86.5
48
PlanningNAVSIM (test)
PDMS86.5
44
Closed-loop Autonomous Driving PlanningNAVSIM v1 (test)
NC98.3
36
End-to-end PlanningNAVSIM v1
NC98.3
32
Showing 10 of 41 rows

Other info

Code

Follow for update