English

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

Computer Vision and Pattern Recognition 2024-09-02 v4

Abstract

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 1st1^{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}.

Keywords

Cite

@article{arxiv.2406.06978,
  title  = {Hydra-MDP: End-to-end Multimodal Planning with Multi-target Hydra-Distillation},
  author = {Zhenxin Li and Kailin Li and Shihao Wang and Shiyi Lan and Zhiding Yu and Yishen Ji and Zhiqi Li and Ziyue Zhu and Jan Kautz and Zuxuan Wu and Yu-Gang Jiang and Jose M. Alvarez},
  journal= {arXiv preprint arXiv:2406.06978},
  year   = {2024}
}

Comments

The 1st place solution of End-to-end Driving at Scale at the CVPR 2024 Autonomous Grand Challenge

R2 v1 2026-06-28T17:00:50.430Z