COME: Adding Scene-Centric Forecasting Control to Occupancy World Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shi, Yining, Jiang, Kun, Meng, Qiang, Wang, Ke, Wang, Jiabao, Sun, Wenchao, Wen, Tuopu, Yang, Mengmeng, Yang, Diange
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916795592474624
author Shi, Yining
Jiang, Kun
Meng, Qiang
Wang, Ke
Wang, Jiabao
Sun, Wenchao
Wen, Tuopu
Yang, Mengmeng
Yang, Diange
author_facet Shi, Yining
Jiang, Kun
Meng, Qiang
Wang, Ke
Wang, Jiabao
Sun, Wenchao
Wen, Tuopu
Yang, Mengmeng
Yang, Diange
contents World models are critical for autonomous driving to simulate environmental dynamics and generate synthetic data. Existing methods struggle to disentangle ego-vehicle motion (perspective shifts) from scene evolvement (agent interactions), leading to suboptimal predictions. Instead, we propose to separate environmental changes from ego-motion by leveraging the scene-centric coordinate systems. In this paper, we introduce COME: a framework that integrates scene-centric forecasting Control into the Occupancy world ModEl. Specifically, COME first generates ego-irrelevant, spatially consistent future features through a scene-centric prediction branch, which are then converted into scene condition using a tailored ControlNet. These condition features are subsequently injected into the occupancy world model, enabling more accurate and controllable future occupancy predictions. Experimental results on the nuScenes-Occ3D dataset show that COME achieves consistent and significant improvements over state-of-the-art (SOTA) methods across diverse configurations, including different input sources (ground-truth, camera-based, fusion-based occupancy) and prediction horizons (3s and 8s). For example, under the same settings, COME achieves 26.3% better mIoU metric than DOME and 23.7% better mIoU metric than UniScene. These results highlight the efficacy of disentangled representation learning in enhancing spatio-temporal prediction fidelity for world models. Code and videos will be available at https://github.com/synsin0/COME.
format Preprint
id arxiv_https___arxiv_org_abs_2506_13260
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle COME: Adding Scene-Centric Forecasting Control to Occupancy World Model
Shi, Yining
Jiang, Kun
Meng, Qiang
Wang, Ke
Wang, Jiabao
Sun, Wenchao
Wen, Tuopu
Yang, Mengmeng
Yang, Diange
Computer Vision and Pattern Recognition
World models are critical for autonomous driving to simulate environmental dynamics and generate synthetic data. Existing methods struggle to disentangle ego-vehicle motion (perspective shifts) from scene evolvement (agent interactions), leading to suboptimal predictions. Instead, we propose to separate environmental changes from ego-motion by leveraging the scene-centric coordinate systems. In this paper, we introduce COME: a framework that integrates scene-centric forecasting Control into the Occupancy world ModEl. Specifically, COME first generates ego-irrelevant, spatially consistent future features through a scene-centric prediction branch, which are then converted into scene condition using a tailored ControlNet. These condition features are subsequently injected into the occupancy world model, enabling more accurate and controllable future occupancy predictions. Experimental results on the nuScenes-Occ3D dataset show that COME achieves consistent and significant improvements over state-of-the-art (SOTA) methods across diverse configurations, including different input sources (ground-truth, camera-based, fusion-based occupancy) and prediction horizons (3s and 8s). For example, under the same settings, COME achieves 26.3% better mIoU metric than DOME and 23.7% better mIoU metric than UniScene. These results highlight the efficacy of disentangled representation learning in enhancing spatio-temporal prediction fidelity for world models. Code and videos will be available at https://github.com/synsin0/COME.
title COME: Adding Scene-Centric Forecasting Control to Occupancy World Model
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2506.13260