Dreaming: Model-based Reinforcement Learning by Latent Imagination without Reconstruction
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
In the present paper, we propose a decoder-free extension of Dreamer, a leading model-based reinforcement learning (MBRL) method from pixels. Dreamer is a sample- and cost-efficient solution to robot learning, as it is used to train latent state-space models based on a variational autoencoder and to conduct policy optimization by latent trajectory imagination. However, this autoencoding based approach often causes object vanishing, in which the autoencoder fails to perceives key objects for solving control tasks, and thus significantly limiting Dreamer's potential. This work aims to relieve this Dreamer's bottleneck and enhance its performance by means of removing the decoder. For this purpose, we firstly derive a likelihood-free and InfoMax objective of contrastive learning from the evidence lower bound of Dreamer. Secondly, we incorporate two components, (i) independent linear dynamics and (ii) the random crop data augmentation, to the learning scheme so as to improve the training performance. In comparison to Dreamer and other recent model-free reinforcement learning methods, our newly devised Dreamer with InfoMax and without generative decoder (Dreaming) achieves the best scores on 5 difficult simulated robotics tasks, in which Dreamer suffers from object vanishing.
Key facts. Cited by 66 papers in the map (data through 14 Oct 2026). Placed in Latent dynamics models and model-based RL, one of 11 directions in World models. Builds on 5 of its 37 references that are in the map. First posted on arXiv 29 Jul 2020.
Builds on
5 of 37 references.
Closest ancestors · 4
Same direction, newest first
- 2020Planning to Explore via Self-Supervised World Models12 May
- 2019Dream to Control: Learning Behaviors by Latent Imagination3 Dec
- 2018Recurrent World Models Facilitate Policy Evolution4 Sep
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Models30 May
Other references · 1
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,264
Built on by
Citing papers per year: 2021: 8; 2022: 17; 2023: 14; 2024: 10; 2025: 8; 2026: 9.
66 papers.
Sort:
By area:
- 2022DayDreamer: World Models for Physical Robot Learningcited by 400
- 2022Masked World Models for Visual Controlcited by 170
- 2024Is Sora a World Simulator? A Comprehensive Survey on General World Models and Beyondcited by 111
- 2021DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representationscited by 74
- 2025A Comprehensive Survey on World Models for Embodied AIcited by 68
- 2022Challenges and Opportunities in Offline Reinforcement Learning from Visual Observationscited by 63
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 54
- 2026World Action Models: The Next Frontier in Embodied AIcited by 50
- 2021Towards Robust Bisimulation Metric Learningcited by 49
- 2022Visuo-Tactile Transformers for Manipulationcited by 47
- 2023Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learningcited by 45
- 2025Latent Action Learning Requires Supervision in the Presence of Distractorscited by 44
38 more
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challengescited by 36
- 2021INVIGORATE: Interactive Visual Grounding and Grasping in Cluttercited by 32
- 2023Facing Off World Model Backbones: RNNs, Transformers, and S4cited by 31
- 2022Joint Embedding Predictive Architectures Focus on Slow Featurescited by 30
- 2022DreamingV2: Reinforcement Learning with Discrete World Models without Reconstructioncited by 29
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2022Augmenting Reinforcement Learning with Transformer-based Scene Representation Learning for Decision-making of Autonomous Drivingcited by 24
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 21
- 2023Simplified Temporal Consistency Reinforcement Learningcited by 17
- 2022Flow-based Recurrent Belief State Learning for POMDPscited by 16
- 2022Time to augment self-supervised visual representation learningcited by 15
- 2024Generative Emergent Communication: Large Language Model is a Collective World Modelcited by 15
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractionscited by 14
- 2024MuDreamer: Learning Predictive World Models without Reconstructioncited by 13
- 2023Mastering Strategy Card Game (Legends of Code and Magic) via End-to-End Policy and Optimistic Smooth Fictitious Playcited by 10
- 2024R-AIF: Solving Sparse-Reward Robotic Tasks from Pixels with Active Inference and World Modelscited by 10
- 2022S2P: State-conditioned Image Synthesis for Data Augmentation in Offline Reinforcement Learningcited by 9
- 2023Combining Reconstruction and Contrastive Methods for Multimodal Representations in RLcited by 9
- 2021Measuring Progress in Deep Reinforcement Learning Sample Efficiencycited by 8
- 2023Hieros: Hierarchical Imagination on Structured State Space Sequence World Modelscited by 8
- 2023DreamSmooth: Improving Model-based Reinforcement Learning via Reward Smoothingcited by 8
- 2022Multi-View Dreaming: Multi-View World Model with Contrastive Learningcited by 6
- 2022Transformer in Transformer as Backbone for Deep Reinforcement Learningcited by 6
- 2024Learning to Play Atari in a World of Tokenscited by 6
- 2021Planning from Pixels in Environments with Combinatorially Hard Search Spacescited by 5
- 2021Maximum Entropy Model-based Reinforcement Learningcited by 5
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learningcited by 5
- 2025Towards Empowerment Gain through Causal Structure Learning in Model-Based RLcited by 5
- 2021A Whole Brain Probabilistic Generative Model: Toward Realizing Cognitive Architectures for Developmental Robotscited by 4
- 2026Next Embedding Prediction Makes World Models Strongercited by 4
- 2022Tactile-Sensitive NewtonianVAE for High-Accuracy Industrial Connector Insertioncited by 3
- 2022Cost-Driven Representation Learning for Linear Quadratic Gaussian Control: Part Icited by 3
- 2023Sample-efficient Real-time Planning with Curiosity Cross-Entropy Method and Contrastive Learningcited by 3
- 2023ConBaT: Control Barrier Transformer for Safe Policy Learningcited by 3
- 2025Edge General Intelligence Through World Models and Agentic AI: Fundamentals, Solutions, and Challengescited by 3
- 2025Dream to Recall: Imagination-Guided Experience Retrieval for Memory-Persistent Vision-and-Language Navigationcited by 3
- 2026World Models as Group Actionscited by 3
50 most cited of 66 All 66 (JSON)
- 2026CF-JEPA: Improving Robustness of JEPA World Models via Controllability Factorization30 Sep 2026
- 2026Flowing Through States: Neural ODE Regularization for Reinforcement Learning6 Aug 2026
- 2026From Pixels to Temporal Correlations: Learning Informative Representations for Reinforcement Learning Pre-training1 Jul 2026
- 2026Task-Relevant Representation Decoupling for Visual Reinforcement Learning Generalization1 Jul 2026
- 2026World Models as Group Actions23 May 2026
- 2026World Action Models: The Next Frontier in Embodied AI12 May 2026
- 2026HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning7 May 2026
- 2026Next Embedding Prediction Makes World Models Stronger3 Mar 2026
- 2026Discrete World Models via Regularization2 Mar 2026
- 2025IPR-1: Interactive Physical Reasoner19 Nov 2025
- 2025A Comprehensive Survey on World Models for Embodied AI19 Oct 2025
- 2025Dream to Recall: Imagination-Guided Experience Retrieval for Memory-Persistent Vision-and-Language Navigation9 Oct 2025
38 more
- 2025Learning State-Space Models of Dynamic Systems from Arbitrary Data using Joint Embedding Predictive Architectures14 Aug 2025
- 2025Edge General Intelligence Through World Models and Agentic AI: Fundamentals, Solutions, and Challenges13 Aug 2025
- 2025MInCo: Mitigating Information Conflicts in Distracted Visual Model-based Reinforcement Learning5 Apr 2025
- 2025Towards Empowerment Gain through Causal Structure Learning in Model-Based RL14 Feb 2025
- 2025Latent Action Learning Requires Supervision in the Presence of Distractors1 Feb 2025
- 2024Generative Emergent Communication: Large Language Model is a Collective World Model31 Dec 2024
- 2024MEATRD: Multimodal Anomalous Tissue Region Detection Enhanced with Spatial Transcriptomics14 Dec 2024
- 2024R-AIF: Solving Sparse-Reward Robotic Tasks from Pixels with Active Inference and World Models21 Sep 2024
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learning23 Jun 2024
- 2024Learning to Play Atari in a World of Tokens3 Jun 2024
- 2024MuDreamer: Learning Predictive World Models without Reconstruction23 May 2024
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2024Is Sora a World Simulator? A Comprehensive Survey on General World Models and Beyond6 May 2024
- 2024Learning Interpretable Policies in Hindsight-Observable POMDPs through Partially Supervised Reinforcement Learning14 Feb 2024
- 2024Bridging State and History Representations: Understanding Self-Predictive RL17 Jan 2024
- 2023DreamSmooth: Improving Model-based Reinforcement Learning via Reward Smoothing2 Nov 2023
- 2023Hieros: Hierarchical Imagination on Structured State Space Sequence World Models8 Oct 2023
- 2023HarmonyDream: Task Harmonization Inside World Models30 Sep 2023
- 2023World-Model-Based Control for Industrial box-packing of Multiple Objects using NewtonianVAE4 Aug 2023
- 2023Facing Off World Model Backbones: RNNs, Transformers, and S45 Jul 2023
- 2023Simplified Temporal Consistency Reinforcement Learning15 Jun 2023
- 2023Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning29 May 2023
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learning9 Mar 2023
- 2023ConBaT: Control Barrier Transformer for Safe Policy Learning7 Mar 2023
- 2023Mastering Strategy Card Game (Legends of Code and Magic) via End-to-End Policy and Optimistic Smooth Fictitious Play7 Mar 2023
- 2023Sample-efficient Real-time Planning with Curiosity Cross-Entropy Method and Contrastive Learning7 Mar 2023
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractions12 Feb 2023
- 2023Combining Reconstruction and Contrastive Methods for Multimodal Representations in RL10 Feb 2023
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challenges14 Jan 2023
- 2022Transformer in Transformer as Backbone for Deep Reinforcement Learning30 Dec 2022
- 2022Cost-Driven Representation Learning for Linear Quadratic Gaussian Control: Part I30 Dec 2022
- 2022Joint Embedding Predictive Architectures Focus on Slow Features20 Nov 2022
- 2022Visuo-Tactile Transformers for Manipulation30 Sep 2022
- 2022S2P: State-conditioned Image Synthesis for Data Augmentation in Offline Reinforcement Learning30 Sep 2022
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objective18 Sep 2022
- 2022Augmenting Reinforcement Learning with Transformer-based Scene Representation Learning for Decision-making of Autonomous Driving24 Aug 2022
- 2022Sparse Representation Learning with Modified q-VAE towards Minimal Realization of World Model8 Aug 2022
- 2022Time to augment self-supervised visual representation learning27 Jul 2022
50 newest of 66 All 66 (JSON)
- 2022DayDreamer: World Models for Physical Robot Learningcited by 400
- 2022Masked World Models for Visual Controlcited by 170
- 2024Is Sora a World Simulator? A Comprehensive Survey on General World Models and Beyondcited by 111
- 2021DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representationscited by 74
- 2025A Comprehensive Survey on World Models for Embodied AIcited by 68
- 2026World Action Models: The Next Frontier in Embodied AIcited by 50
- 2023Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learningcited by 45
- 2023World Models and Predictive Coding for Cognitive and Developmental Robotics: Frontiers and Challengescited by 36
- 2023Facing Off World Model Backbones: RNNs, Transformers, and S4cited by 31
- 2022Joint Embedding Predictive Architectures Focus on Slow Featurescited by 30
- 2022DreamingV2: Reinforcement Learning with Discrete World Models without Reconstructioncited by 29
- 2023HarmonyDream: Task Harmonization Inside World Modelscited by 22
12 most cited of 31 All 31 (JSON)
- 2026CF-JEPA: Improving Robustness of JEPA World Models via Controllability Factorization30 Sep 2026
- 2026World Models as Group Actions23 May 2026
- 2026World Action Models: The Next Frontier in Embodied AI12 May 2026
- 2026HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning7 May 2026
- 2026Next Embedding Prediction Makes World Models Stronger3 Mar 2026
- 2026Discrete World Models via Regularization2 Mar 2026
- 2025IPR-1: Interactive Physical Reasoner19 Nov 2025
- 2025A Comprehensive Survey on World Models for Embodied AI19 Oct 2025
- 2025Dream to Recall: Imagination-Guided Experience Retrieval for Memory-Persistent Vision-and-Language Navigation9 Oct 2025
- 2025Learning State-Space Models of Dynamic Systems from Arbitrary Data using Joint Embedding Predictive Architectures14 Aug 2025
- 2025Edge General Intelligence Through World Models and Agentic AI: Fundamentals, Solutions, and Challenges13 Aug 2025
- 2024Generative Emergent Communication: Large Language Model is a Collective World Model31 Dec 2024
12 newest of 31 All 31 (JSON)
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 54
- 2025Towards Empowerment Gain through Causal Structure Learning in Model-Based RLcited by 5
- 2024Learning Interpretable Policies in Hindsight-Observable POMDPs through Partially Supervised Reinforcement Learningcited by 2
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learningcited by 1
- 2025MInCo: Mitigating Information Conflicts in Distracted Visual Model-based Reinforcement Learningcited by 0
- 2026Task-Relevant Representation Decoupling for Visual Reinforcement Learning Generalizationcited by 0
- 2026From Pixels to Temporal Correlations: Learning Informative Representations for Reinforcement Learning Pre-trainingcited by 0
- 2026Flowing Through States: Neural ODE Regularization for Reinforcement Learningcited by 0
- 2026Flowing Through States: Neural ODE Regularization for Reinforcement Learning6 Aug 2026
- 2026From Pixels to Temporal Correlations: Learning Informative Representations for Reinforcement Learning Pre-training1 Jul 2026
- 2026Task-Relevant Representation Decoupling for Visual Reinforcement Learning Generalization1 Jul 2026
- 2025MInCo: Mitigating Information Conflicts in Distracted Visual Model-based Reinforcement Learning5 Apr 2025
- 2025Towards Empowerment Gain through Causal Structure Learning in Model-Based RL14 Feb 2025
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learning23 Jun 2024
- 2024Learning Interpretable Policies in Hindsight-Observable POMDPs through Partially Supervised Reinforcement Learning14 Feb 2024
- 2024Bridging State and History Representations: Understanding Self-Predictive RL17 Jan 2024
- 2021Towards Robust Bisimulation Metric Learningcited by 49
- 2022Visuo-Tactile Transformers for Manipulationcited by 47
- 2025Latent Action Learning Requires Supervision in the Presence of Distractorscited by 44
- 2021INVIGORATE: Interactive Visual Grounding and Grasping in Cluttercited by 32
- 2022Simplifying Model-based RL: Learning Representations, Latent-space Models, and Policies with One Objectivecited by 28
- 2023Simplified Temporal Consistency Reinforcement Learningcited by 17
- 2022Flow-based Recurrent Belief State Learning for POMDPscited by 16
- 2022Time to augment self-supervised visual representation learningcited by 15
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractionscited by 14
- 2023Mastering Strategy Card Game (Legends of Code and Magic) via End-to-End Policy and Optimistic Smooth Fictitious Playcited by 10
- 2023Combining Reconstruction and Contrastive Methods for Multimodal Representations in RLcited by 9
- 2021Measuring Progress in Deep Reinforcement Learning Sample Efficiencycited by 8
12 most cited of 24 All 24 (JSON)
- 2025Latent Action Learning Requires Supervision in the Presence of Distractors1 Feb 2025
- 2024MEATRD: Multimodal Anomalous Tissue Region Detection Enhanced with Spatial Transcriptomics14 Dec 2024
- 2023DreamSmooth: Improving Model-based Reinforcement Learning via Reward Smoothing2 Nov 2023
- 2023Simplified Temporal Consistency Reinforcement Learning15 Jun 2023
- 2023ConBaT: Control Barrier Transformer for Safe Policy Learning7 Mar 2023
- 2023Mastering Strategy Card Game (Legends of Code and Magic) via End-to-End Policy and Optimistic Smooth Fictitious Play7 Mar 2023
- 2023Sample-efficient Real-time Planning with Curiosity Cross-Entropy Method and Contrastive Learning7 Mar 2023
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractions12 Feb 2023
- 2023Combining Reconstruction and Contrastive Methods for Multimodal Representations in RL10 Feb 2023
- 2022Transformer in Transformer as Backbone for Deep Reinforcement Learning30 Dec 2022
- 2022Cost-Driven Representation Learning for Linear Quadratic Gaussian Control: Part I30 Dec 2022
- 2022Visuo-Tactile Transformers for Manipulation30 Sep 2022
12 newest of 24 All 24 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2018Learning Real-World Robot Policies by Dreaming
- 2026R2-Dreamer: Redundancy-Reduced World Models without Decoders or Augmentation
- 2023Model-Based Reinforcement Learning with Isolated Imaginations
- 2026Dreamer-CDP: Improving Reconstruction-free World Models Via Continuous Deterministic Representation Prediction
- 2020Bridging Imagination and Reality for Model-Based Deep Reinforcement Learning
- 2022Iso-Dream: Isolating and Leveraging Noncontrollable Visual Dynamics in World Models
- 2026Dream-MPC: Gradient-Based Model Predictive Control with Latent Imagination
- 2026DreamFormer: Dream Imitation with a Transformer World Model for Language-Conditioned Robotic Manipulation