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
Fig. 1: Overall workflow of the proposed multimodal trajectory generation framework K-Gen. The MLLM takes multimodal scene data (map images and textual inputs) as input, produces both reasoning and sparse keypoints, and refines them into complete trajectories.
Fig. 2: Overall framework of K-Gen. The framework is composed of two main components: an MLLM for reasoning and keypoint generation, and a Transformer-based module for trajectory refinement. For the MLLM, the training pipeline includes two stages: SFT with CoT outputs, followed by reinforcement RFT using the T-DAPO algorithm.
Dataset
Method
Size
mADE ↓
mFDE ↓
SCR ↓
WOMD
LCTGen [ 13 ] †
-
1.262
2.696
0.072
InteractTraj [ 14 ] †
-
1.067
2.190
0.070
InternVL3 [ 6 ]
2B
1.260
3.147
0.011
InternVL2.5 [ 26 ]
4B
1.187
2.796
0.010
InternVL3 [ 6 ]
8B
1.128
2.770
0.007
Qwen2.5-VL [ 27 ]
7B
0.989
2.602
0.008
TABLE I: Quantitative results on WOMD and nuPlan datasets. Methods marked with † are based on closed-source LLMs (API access only, no model training).
Fig. 3: Qualitative results and attention visualization across diverse scenarios (Case 1: Intersection, Case 2: Merging, Case 3: Curved Lane). Columns (a)-(c) show the comparison between the ground truth and trajectories generated by the baseline and our TrajRefiner. Column (d) visualizes the MLLM’s attention heatmaps, where warmer colors indicate higher focus on safety-critical regions and interacting agents.
Setting
SFT
GRPO
DAPO
TrajRefiner
mADE ↓
mFDE ↓
SCR ↓
Zero-shot
-
-
-
-
9.880
15.369
0.5034
SFT
✓
-
-
-
1.128
2.770
0.0068
SFT + TrajRefiner
✓
-
-
✓
0.926
2.496
0.0073
SFT + GRPO + TrajRefiner
✓
✓
-
✓
0.923
2.447
0.0064
SFT + DAPO + TrajRefiner(ours)
✓
-
✓
✓
0.915
2.422
0.0058
TABLE II: Ablation study on training pipeline components. All experiments are based on the 8B model and WOMD datasets.
Method
mADE ↓
mFDE ↓
SCR ↓
Baseline
1.154
2.660
0.0229
+ RCE
0.499
1.404
0.0116
+ KCL
0.416
1.289
0.0099
+ FPL
0.423
0.798
0.0086
TrajRefiner(ours)
0.369
0.738
0.0078
TABLE III: Ablation study of TrajRefiner. Experiments are based on the WOMD datasets.
While multimodal large language models (MLLMs) provide advanced reasoning for autonomous driving, translating their discrete semantic knowledge into continuous trajectories remains a fundamental challenge. Existing methods often rely on unimodal planning heads that inherently limit their ability to represent multimodal driving behavior. Furthermore, most generative approaches frequently condition on one-hot encoded actions, discarding the nuanced navigational uncertainty critical for complex scenarios. To resolve these limitations, we introduce LAD-Drive, a generative framework that structurally disentangles high-level intention from low-level spatial planning. LAD-Drive employs an action decoder to infer a probabilistic meta-action distribution, establishing an explicit belief state that preserves the nuanced intent typically lost by one-hot encodings. This distribution, fused with the vehicle's kinematic state, conditions an action-aware diffusion decoder that utilizes a truncated denoising process to refine learned motion anchors into safe, kinematically feasible trajectories. Extensive evaluations on the LangAuto benchmark demonstrate that LAD-Drive achieves state-of-the-art results, outperforming competitive baselines by up to 59% in Driving Score while significantly reducing route deviations and collisions. We will publicly release the code and models on https://github.com/iis-esslingen/lad-drive.
Fabian Schmidt, Karol Fedurko, Markus Enzweiler +1
Institute for Intelligent Systems, Esslingen University of Applied Sciences, Germany. · Department of Computer Science, University of Freiburg, Germany.
Recent years have seen remarkable progress in autonomous driving, yet generalization to long-tail and open-world scenarios remains a major bottleneck for large-scale deployment. To address this challenge, some works use LLMs and VLMs for vision-language understanding and reasoning, enabling vehicles to interpret rare and safety-critical situations when generating actions. Others study generative world models to capture the spatio-temporal evolution of driving scenes, allowing agents to imagine possible futures before acting. Inspired by human intelligence, which unifies understanding and imagination, we explore a unified model for autonomous driving. We present LMGenDrive, the first framework that combines LLM-based multimodal understanding with generative world models for end-to-end closed-loop driving. Given multi-view camera inputs and natural-language instructions, LMGenDrive generates both future driving videos and control signals. This design provides complementary benefits: video prediction improves spatio-temporal scene modeling, while the LLM contributes strong semantic priors and instruction grounding from large-scale pretraining. We further propose a progressive three-stage training strategy, from vision pretraining to multi-step long-horizon driving, to improve stability and performance. LMGenDrive supports both low-latency online planning and autoregressive offline video generation. Experiments show that it significantly outperforms prior methods on challenging closed-loop benchmarks, with clear gains in instruction following, spatio-temporal understanding, and robustness to rare scenarios. These results suggest that unifying multimodal understanding and generation is a promising direction for more generalizable and robust embodied decision-making systems.
Vision-Language Models (VLMs) and world models have demonstrated strong capabilities in semantic understanding and future scene evolution for autonomous driving. However, effectively integrating both capabilities for trajectory generation remains challenging. In this work, we propose DriveWLA, a unified World-Language-Action model based on Mixture-of-Transformers (MoT) that jointly models semantic understanding, future scene dynamics, and trajectory generation. DriveWLA consists of three experts: a World Expert for future prediction, a Language Expert for semantic understanding, and an Action Expert for trajectory generation. We further introduce Cross-Expert Attention, which enables controlled information flow across experts while aligning semantic and dynamic representations. To improve trajectory generation beyond imitation learning, we apply Flow-GRPO reinforcement learning post-training to refine the Action Expert. Extensive experiments show that DriveWLA achieves superior performance in both open-loop and closed-loop evaluation. DriveWLA achieves 92.3 PDMS on NAVSIM v1 and 90.1 EPDMS on NAVSIM v2, demonstrating strong performance among VLM-based and world-model-based planners. Without specific fine-tuning, DriveWLA further demonstrates strong zero-shot generalization to HUGSIM, nuScenes, and Bench2Drive. These results validate the effectiveness of our unified World-Language-Action modeling, establishing DriveWLA as a generalizable framework for autonomous driving.
Shuzheng Wang, Haoxuan Li, Shengzhen Li +5
Huazhong University of Science and Technology · Yinwang Intelligent Technology Co. Ltd.