cs.AIMar 5, 2026

K-Gen: A Multimodal Language-Conditioned Approach for Interpretable Keypoint-Guided Trajectory Generation

Authors: Mingxuan Mu, Guo Yang, Lei Chen, Ping Wu, Jianxun Cui

Organizations: Harbin Institute of Technology Chongqing Research Inst. of HIT · Changan Automobile Co., Ltd Chongqing, China

Abstract

Generating realistic and diverse trajectories is a critical challenge in autonomous driving simulation. While Large Language Models (LLMs) show promise, existing methods often rely on structured data like vectorized maps, which fail to capture the rich, unstructured visual context of a scene. To address this, we propose K-Gen, an interpretable keypoint-guided multimodal framework that leverages Multimodal Large Language Models (MLLMs) to unify rasterized BEV map inputs with textual scene descriptions. Instead of directly predicting full trajectories, K-Gen generates interpretable keypoints along with reasoning that reflects agent intentions, which are subsequently refined into accurate trajectories by a refinement module. To further enhance keypoint generation, we apply T-DAPO, a trajectory-aware reinforcement fine-tuning algorithm. Experiments on WOMD and nuPlan demonstrate that K-Gen outperforms existing baselines, highlighting the effectiveness of combining multimodal reasoning with keypoint-guided trajectory generation.

Figures & tables

Explore similar work

CardsList
  1. LAD-Drive: Bridging Language and Trajectory with Action-Aware Diffusion Transformers

    Mar 2, 2026Fabian Schmidt, Karol Fedurko, Markus Enzweiler +1Language-Conditioned Robot PlanningAutonomous Driving Planning

  2. LMGenDrive: Bridging Multimodal Understanding and Generative World Modeling for End-to-End Driving

    Apr 9, 2026Hao Shao, Letian Wang, Yang Zhou +5Autonomous DrivingEnd-to-End Autonomous Driving

  3. DriveWLA: Unifying Future Dynamics and Scene Semantics through World-Language-Action Modeling for Autonomous Driving

    Oct 13, 2026Shuzheng Wang, Haoxuan Li, Shengzhen Li +5Mixture-of-TransformersAutonomous Driving Planning