Search papers, labs, and topics across Lattice.
This paper investigates the controllability of driving world models built on large video-diffusion backbones without the need for retraining. By employing differentiable energy functions to steer planned trajectories at sampling time, the authors demonstrate that a rectified-flow driving world model can effectively adjust its behavior in real-time. However, they identify a limitation where the generated video does not align with the steered trajectory due to insufficient cross-stream coupling in the model's architecture.
Steering driving world models in real-time without retraining could revolutionize how we control AI-generated simulations.
Driving world models built on large video-diffusion backbones generate realistic scenes but are hard to control: enforcing a traffic norm typically means retraining the backbone or conditioning it on hand-built layouts. We ask whether controllability requires training at all. Our experiment shows that a rectified-flow driving world model, which jointly generates future video and a planned ego trajectory, can have its planned trajectory steered entirely at sampling time by differentiable energy functions that encode driving norms, without knowledge-specific retraining of the diffusion backbone. Concretely, we demonstrate that a world model built on Open-Sora 2.0 MM-DiT backbone can be steered to brake at a counterfactual target by injecting energy guidance at sampling time. However, we find that the generated video does not yet follow the steered trajectory through the backbone's joint self-attention and identify the cross-stream coupling as a crucial requirement for end-to-end-controllable rollouts.