Beyond Single-Speed Reasoning: Coordinating Fast and Slow Dynamics for Efficient World Modeling
Hongwei Wang, Yangru Huang, Guangyao Chen, Xu Wang, Yi Jin
Abstract
Model-based reinforcement learning (MBRL) enables efficient decision-making by learning predictive world models of environment dynamics. Despite recent advances, existing models often struggle to reconcile accurate short-term transitions with coherent long-term planning, especially in partially observable or long-horizon settings. We argue that this limitation often stems from modeling all transitions at a single temporal resolution, which makes it challenging to simultaneously capture fine-grained local dynamics and abstract global structures. To this end, we propose SF-RSSM (Slow-Fast Recurrent State-Space Model), a novel method that decouples short-term and long-term dynamics via a dual-branch design. The fast branch captures short-horizon transitions using residual prediction, while the slow branch models longrange dependencies with a GRU-based recurrent pathway. A distillation mechanism is developed to enable cooperation across timescales, with the slow model providing soft targets to guide the fast model. Additionally, a curiosity module encourages exploration by promoting learning in regions where the fast and slow branches exhibit divergent dynamics. Experiments on CARLA, DMControl and Atari benchmarks show that SF-RSSM outperforms strong baselines in policy performance.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 6ede2fd8-0284-46c1-9c71-8c935ade1298Builds on11
- Dream to Control: Learning Behaviors by Latent ImaginationDanijar Hafner, Timothy P. Lillicrap, Jimmy Ba, Mohammad NorouziICLR 2020 · 1,852 citations
- CURL: Contrastive Unsupervised Representations for Reinforcement LearningMichael Laskin, Aravind Srinivas, Pieter AbbeelICML 2020 · 1,261 citations
- Mastering Atari with Discrete World ModelsDanijar Hafner, Timothy P. Lillicrap, Mohammad Norouzi, Jimmy BaICLR 2021 · 1,170 citations
- Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from PixelsDenis Yarats, Ilya Kostrikov, Rob FergusICLR 2021 · 911 citations
- Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement LearningDenis Yarats, Rob Fergus, Alessandro Lazaric, Lerrel PintoICLR 2022 · 457 citations
Related papers
- DyMoDreamer: World Modeling with Dynamic ModulationBoxuan Zhang, Runqing Wang, Wei Xiao, Weipu Zhang et al.NeurIPS 2025 · 2 citations
- Mastering Memory Tasks with World ModelsMohammad Reza Samsami, Artem Zholus, Janarthanan Rajendran, Sarath ChandarICLR 2024 · 42 citations
- Learning Latent Dynamic Robust Representations for World ModelsRuixiang Sun, Hongyu Zang, Xin Li, Riashat IslamICML 2024 · 15 citations
- Facing Off World Model Backbones: RNNs, Transformers, and S4Fei Deng, Junyeong Park, Sungjin AhnNeurIPS 2023 · 53 citations
- DreamSmooth: Improving Model-based Reinforcement Learning via Reward SmoothingVint Lee, Pieter Abbeel, Youngwoon LeeICLR 2024 · 10 citations
