Long-Horizon Consistent and Interaction-Aware World Models for Multi-Style End-to-End Driving
On the map: World models → World models for autonomous driving
also in 1 more
Abstract (arXiv)
End-to-end autonomous driving has increasingly adopted world model-based reinforcement learning frameworks to improve learning efficiency through textitimagined rollouts. However, existing world models suffer from three key limitations: temporal inconsistency in long-horizon imagined rollouts, inadequate modeling of ego-environment interactions, and limited adaptability to diverse driving styles. To address these challenges, we propose textitStyleDrive, a world-model-based learning framework that jointly enforces long-horizon consistency, explicitly disentangles interactive traffic states, and supports multi-style policy optimization within a unified learning paradigm. First, we introduce a temporal consistency regularization that integrates historical latent states through gated cross-attention, stabilizing long-horizon imagined rollouts and mitigating error accumulation. Second, we design an explicit state disentanglement module that separates ego-relevant from ego-irrelevant interactive states, enabling more interpretable and efficient decision-making in complex traffic scenarios. Third, we enable multi-style driving behaviors through Group Relative Policy Optimization, which replaces per-step reward optimization with trajectory-wise relative advantages, reducing reward variance and supporting diverse driving styles without retraining. We evaluate StyleDrive on the Bench2Drive closed-loop driving benchmark, achieving a driving score of 88.44 (+17.08 over the previous best world model-based method) and a success rate of 66.82 (+16.58). Furthermore, we deploy StyleDrive on a real automated guided vehicle platform and demonstrate promising sim-to-real transfer capability in dynamic driving scenarios.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in World models for autonomous driving, one of 11 directions in World models. Builds on 6 of its 40 references that are in the map. First posted on arXiv 3 Sep 2026.
Builds on
6 of 40 references.
Closest ancestors · 2
Same direction, newest first
- 2026GraphWorld: Long-Horizon Planning with World Models for End-to-End Autonomous Driving15 Jun
- 2026EponaV2: Driving World Model with Comprehensive Future Reasoning14 May
Other references · 4
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2019Lyapunov-based Safe Policy Optimization for Continuous Controlcited by 140
- 2025Poutine: Vision-Language-Trajectory Pre-Training and Reinforcement Learning Post-Training Enable Robust End-to-End Autonomous Drivingcited by 11
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026PLAN-S: Bridging Planning with Latent Style Dynamics for Autonomous Driving World Models
- 2026Kinematics-Aware Latent World Models for Data-Efficient Autonomous Driving
- 2025WorldRFT: Latent World Model Planning with Reinforcement Fine-Tuning for Autonomous Driving
- 2024Imagine-2-Drive: Leveraging High-Fidelity World Models via Multi-Modal Diffusion Policies
- 2025AD-R1: Closed-Loop Reinforcement Learning for End-to-End Autonomous Driving with Impartial World Models
- 2024From Imitation to Exploration: End-to-end Autonomous Driving based on World Model
- 2026Latent World Models for Automated Driving: A Unified Taxonomy, Evaluation Framework, and Open Challenges
- 2026Drive-HWM: Hierarchical World Models for Dynamic-Latent Guided Autonomous Driving