arXiv · 2506.08441
Time-Aware World Model for Adaptive Prediction and Control
Abstract
In this work, we introduce the Time-Aware World Model (TAWM), a model-based approach that explicitly incorporates temporal dynamics. By conditioning on the time-step size, {\Delta}t, and training over a diverse range of {\Delta}t values -- rather than sampling at a fixed time-step -- TAWM learns both high- and low-frequency task dynamics across diverse control problems. Grounded in the information-theoretic insight that the optimal sampling rate depends on a system's underlying dynamics, this time-aware formulation improves both performance and data efficiency. Empirical evaluations show that TAWM consistently outperforms conventional models across varying observation rates in a variety of control tasks, using the same number of training samples and iterations. Our code can be found online at: github.com/anh-nn01/Time-Aware-World-Model.
Explore related subjects
Keep this discovery
Anh N. Nhu, Sanghyun Son, Ming Lin. 2025-06-10. Time-Aware World Model for Adaptive Prediction and Control. https://arxiv.org/abs/2506.08441
Cite the original work for its findings. Save a collection to share your selection of sources.