English

MTDrive: Multi-turn Interactive Reinforcement Learning for Autonomous Driving

Robotics 2026-02-02 v1 Artificial Intelligence Machine Learning

Abstract

Trajectory planning is a core task in autonomous driving, requiring the prediction of safe and comfortable paths across diverse scenarios. Integrating Multi-modal Large Language Models (MLLMs) with Reinforcement Learning (RL) has shown promise in addressing "long-tail" scenarios. However, existing methods are constrained to single-turn reasoning, limiting their ability to handle complex tasks requiring iterative refinement. To overcome this limitation, we present MTDrive, a multi-turn framework that enables MLLMs to iteratively refine trajectories based on environmental feedback. MTDrive introduces Multi-Turn Group Relative Policy Optimization (mtGRPO), which mitigates reward sparsity by computing relative advantages across turns. We further construct an interactive trajectory understanding dataset from closed-loop simulation to support multi-turn training. Experiments on the NAVSIM benchmark demonstrate superior performance compared to existing methods, validating the effectiveness of our multi-turn reasoning paradigm. Additionally, we implement system-level optimizations to reduce data transfer overhead caused by high-resolution images and multi-turn sequences, achieving 2.5x training throughput. Our data, models, and code will be made available soon.

Keywords

Cite

@article{arxiv.2601.22930,
  title  = {MTDrive: Multi-turn Interactive Reinforcement Learning for Autonomous Driving},
  author = {Xidong Li and Mingyu Guo and Chenchao Xu and Bailin Li and Wenjing Zhu and Yangang Zou and Rui Chen and Zehuan Wang},
  journal= {arXiv preprint arXiv:2601.22930},
  year   = {2026}
}
R2 v1 2026-07-01T09:27:43.248Z