Abstract
Value Diffusion World Models combine end-to-end online training with latent diffusion dynamics to enable fast, uncertain dynamics prediction for Model Predictive Control in reinforcement learning environments.
World models can enable Model Predictive Control (MPC), but this requires dynamics prediction that is both fast enough for online use and expressive enough to represent uncertain futures. Diffusion models offer a natural mechanism for modeling uncertain dynamics, yet their iterative inference procedure makes them difficult to use for low-latency latent planning. We bridge this gap with Value Diffusion World Models (Valdi), combining end-to-end online training for MPC with a latent diffusion dynamics model. In preliminary experiments on the CarRacing environment, we show that Valdi, using a single diffusion step at both training and inference, matches a deterministic MLP baseline. Our experiments expose a trade-off between predictive multimodality and control performance in this setup. Code is available at https://github.com/Kit115/ValueDiffusionWorldModels.
Community
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- JEDI: Joint Embedding Diffusion World Model for Online Model-Based Reinforcement Learning (2026)
- Ada-Diffuser: Latent-Aware Adaptive Diffusion for Decision-Making (2026)
- MODIP: Efficient Model-Based Optimization for Diffusion Policies (2026)
- Beyond the Next Step: Variable-Length Latent World Models for Long-Horizon Planning (2026)
- IDOL: Inverse-Dynamics-Guided Future Prediction for End-to-End Autonomous Driving (2026)
- RS-Diffuser: Risk-Sensitive Diffusion Planning with Distributional Value Guidance (2026)
- AdaJEPA: An Adaptive Latent World Model (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2607.00917 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper