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
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Autonomous Driving | NAVSIM v1 (test) | NC98.7 | 99 | |
| Planning | NAVSIM (navtest) | NC98.3 | 53 | |
| Autonomous Driving Planning | NAVSIM (navtest) | NC98.3 | 50 | |
| Autonomous Driving | NAVSIM (test) | PDMS86.5 | 34 | |
| End-to-end Autonomous Driving | Bench2Drive | Driving Score59.95 | 27 | |
| Closed-loop Autonomous Driving Planning | NAVSIM v1 (test) | NC98.3 | 26 | |
| Closed-loop Autonomous Driving | Bench2Drive | Driving Score (DS)59.95 | 21 | |
| Open-loop Autonomous Driving Planning | NAVSIM 1.0 (test) | NC98.3 | 19 | |
| End-to-end Planning | NAVSIM v1 | NC0.984 | 17 | |
| Autonomous Driving Planning | NAVSIM v1 | NC98.3 | 17 |
Showing 10 of 32 rows