ChronoEdit: Towards Temporal Reasoning for Image Editing and World Simulation
Jay Zhangjie Wu, Xuanchi Ren, Tianchang Shen, Tianshi Cao, Kai He, Yifan Lu, Ruiyuan Gao, Enze Xie, Shiyi Lan, Jose M. Alvarez, Jun Gao, Sanja Fidler, Zian Wang, Huan Ling
TL;DR
This work introduces ChronoEdit, a foundation model for physically consistent image editing by reframing edits as a short video generation task using pretrained image-to-video diffusion models. A temporal reasoning inference stage inserts intermediate reasoning tokens to imagine plausible, physically viable edit trajectories, then discards them to refine the final frame efficiently. The authors curate a large synthetic video dataset and propose PBench-Edit to evaluate physical consistency in world-simulation contexts. Empirical results show state-of-the-art open-source performance and competitive results with leading proprietary systems, with fast variants like ChronoEdit-Turbo and decisions about reasoning horizon that balance quality and efficiency. Overall, ChronoEdit provides a scalable approach for temporally coherent, physically grounded image edits applicable to autonomous driving, robotics, and other simulation tasks.
Abstract
Recent advances in large generative models have greatly enhanced both image editing and in-context image generation, yet a critical gap remains in ensuring physical consistency, where edited objects must remain coherent. This capability is especially vital for world simulation related tasks. In this paper, we present ChronoEdit, a framework that reframes image editing as a video generation problem. First, ChronoEdit treats the input and edited images as the first and last frames of a video, allowing it to leverage large pretrained video generative models that capture not only object appearance but also the implicit physics of motion and interaction through learned temporal consistency. Second, ChronoEdit introduces a temporal reasoning stage that explicitly performs editing at inference time. Under this setting, target frame is jointly denoised with reasoning tokens to imagine a plausible editing trajectory that constrains the solution space to physically viable transformations. The reasoning tokens are then dropped after a few steps to avoid the high computational cost of rendering a full video. To validate ChronoEdit, we introduce PBench-Edit, a new benchmark of image-prompt pairs for contexts that require physical consistency, and demonstrate that ChronoEdit surpasses state-of-the-art baselines in both visual fidelity and physical plausibility. Project page for code and models: https://research.nvidia.com/labs/toronto-ai/chronoedit
