cs.ROSep 15, 2026

EditWM: Event-Decomposed World Modeling with Incremental Correction for End-to-End Autonomous Driving

Authors: Junjie Yang, Qingwei Zeng, Youyou Li, Zicheng Ding, Ziyi Shi, Shuqi Shen, Hongliang Lu, Hai Yang

Organizations: The Hong Kong University of Science and Technology · Southern University of Science and Technology · The Chinese University of Hong Kong, Shenzhen

Abstract

World models support autonomous driving by predicting the scene evolution associated with candidate trajectories. Driving dynamics differ in predictability, motivating a distinction between regular evolution and event-induced deviations that call for selective correction. We propose EditWM, a world model that decomposes future prediction into normal evolution and event-driven incremental correction in compact visual feature space. A trajectory-conditioned normal predictor provides the base forecast and is then frozen for correction learning. A correction decoder compares this forecast with observation history and planned actions, producing a bounded feature update whose contribution is regulated by a learned gate. The corrected future features condition trajectory scoring through candidate-specific cross-attention, linking world modeling to plan selection. At inference, EditWM uses only past and current observations, ego state, and candidate trajectories. Across all 12,146 NAVSIM navtest scenes, expert-trajectory-conditioned evaluation shows a 5.35% reduction in future-feature MSE over Normal, with improvements in 83.54% of scenes. The system achieves 91.05 EPDMS on a 100-point scale using the official EPDMS evaluator. These results demonstrate improved future-feature prediction and competitive trajectory selection when corrected future representations are integrated into planning.

Figures & tables

Explore similar work

CardsList
  1. PlanWAM: Planning-Shaped Future Representations for End-to-End Autonomous Driving

    Oct 8, 2026Jinchang Xu, Hongda Yu, Fengwei Dong +9End-to-End Autonomous DrivingWorld Model-Based Planning

  2. DA-WAM: Decision-Aligned Future Latents for Driving World Models

    Aug 19, 2026Ruiguo Zhong, Xiaolong Chen, Benshan Ma +5World Models for Autonomous DrivingAutonomous Driving Planning

  3. Drive-HWM: Hierarchical World Models for Dynamic-Latent Guided Autonomous Driving

    Sep 3, 2026Zhaoxin Fan, Tianbao Zhang, Wenjun Wu +5World ModelsHierarchical World Models