cs.ROApr 9, 2026

On-Policy Distillation of Language Models for Autonomous Vehicle Motion Planning

Authors: Amirhossein Afsharrad, Amirhesam Abedsoltan, Ahmadreza Moradipari, Sanjay Lall

Organizations: Stanford University · University of California, San Diego · University of California, Santa Barbara

Abstract

Large language models (LLMs) have recently demonstrated strong potential for autonomous vehicle motion planning by reformulating trajectory prediction as a language generation problem. However, deploying capable LLMs in resource-constrained onboard systems remains a fundamental challenge. In this paper, we study how to effectively transfer motion planning knowledge from a large teacher LLM to a smaller, more deployable student model. We build on the GPT-Driver framework, which represents driving scenes as language prompts and generates waypoint trajectories with chain-of-thought reasoning, and investigate two student training paradigms: (i) on-policy generalized knowledge distillation (GKD), which trains the student on its own self-generated outputs using dense token-level feedback from the teacher, and (ii) a dense-feedback reinforcement learning (RL) baseline that uses the teacher's log-probabilities as per-token reward signals in a policy gradient framework. Experiments on the nuScenes benchmark show that GKD substantially outperforms the RL baseline and closely approaches teacher-level performance despite a 5×\times reduction in model size. These results highlight the practical value of on-policy distillation as a principled and effective approach to deploying LLM-based planners in autonomous driving systems.

Figures & tables

Explore similar work

CardsList
  1. GAPL: Grounded Action-effect Policy Learning for LLM-Based Trajectory Planning

    Aug 18, 2026Zhihong Cui, Hengyu Liu, Zhangkai Wu +5Autonomous Driving PlanningLarge Language Model-Based Robot Planning

  2. Orion-Lite: Distilling LLM Reasoning into Efficient Vision-Only Driving Models

    Apr 9, 2026Jing Gu, Niccolò Cavagnero, Gijs DubbelmanEnd-to-End Autonomous DrivingVLM Distillation