Discovering and Achieving Goals via World Models
On the map: World models → Embodied and robot world models
Abstract (arXiv)
How can artificial agents learn to solve many diverse tasks in complex visual environments in the absence of any supervision? We decompose this question into two problems: discovering new goals and learning to reliably achieve them. We introduce Latent Explorer Achiever (LEXA), a unified solution to these that learns a world model from image inputs and uses it to train an explorer and an achiever policy from imagined rollouts. Unlike prior methods that explore by reaching previously visited states, the explorer plans to discover unseen surprising states through foresight, which are then used as diverse targets for the achiever to practice. After the unsupervised phase, LEXA solves tasks specified as goal images zero-shot without any additional learning. LEXA substantially outperforms previous approaches to unsupervised goal-reaching, both on prior benchmarks and on a new challenging benchmark with a total of 40 test tasks spanning across four standard robotic manipulation and locomotion domains. LEXA further achieves goals that require interacting with multiple objects in sequence. Finally, to demonstrate the scalability and generality of LEXA, we train a single general agent across four distinct environments. Code and videos at https://orybkin.github.io/lexa/
Key facts. Cited by 131 papers in the map (data through 7 Oct 2026). Placed in Embodied and robot world models, one of 11 directions in World models. Builds on 46 of its 58 references that are in the map. First posted on arXiv 18 Oct 2021.
Builds on
46 of 58 references.
References · 46
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,168
- 2017Hindsight Experience Replaycited by 1,096
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2018Exploration by Random Network Distillationcited by 966
34 more
- 2018DeepMind Control Suitecited by 935
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
- 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 632
- 2019Relay Policy Learning: Solving Long-Horizon Tasks via Imitation and Reinforcement Learningcited by 380
- 2020Planning to Explore via Self-Supervised World Modelscited by 365
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2016Deep Visual Foresight for Planning Robot Motioncited by 276
- 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 266
- 2019Self-Supervised Exploration via Disagreementcited by 263
- 2016Variational Intrinsic Controlcited by 262
- 2017Automatic Goal Generation for Reinforcement Learning Agentscited by 252
- 2018Visual Reinforcement Learning with Imagined Goalscited by 248
- 2018Visual Foresight: Model-Based Deep Reinforcement Learning for Vision-Based Robotic Controlcited by 220
- 2019Go-Explore: a New Approach for Hard-Exploration Problemscited by 207
- 2019Search on the Replay Buffer: Bridging Planning and Reinforcement Learningcited by 188
- 2019Skew-Fit: State-Covering Self-Supervised Reinforcement Learningcited by 171
- 2017Intrinsic Motivation and Automatic Curricula via Asymmetric Self-Playcited by 139
- 2021Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skillscited by 125
- 2018Variational Option Discovery Algorithmscited by 121
- 2020Explore, Discover and Learn: Unsupervised Discovery of State-Covering Skillscited by 120
- 2018Zero-Shot Visual Imitationcited by 114
- 2018Model-Based Active Explorationcited by 105
- 2018Unsupervised Control Through Non-Parametric Discriminative Rewardscited by 102
- 2020Automatic Curriculum Learning through Value Disagreementcited by 86
- 2021Asymmetric self-play for automatic goal discovery in robotic manipulationcited by 71
- 2019Dynamical Distance Learning for Semi-Supervised and Unsupervised Skill Discoverycited by 65
- 2018Learning and Querying Fast Generative Models for Reinforcement Learningcited by 48
- 2020Model-Based Visual Planning with Self-Supervised Functional Distancescited by 41
- 2021Clockwork Variational Autoencoderscited by 40
- 2020Ready Policy One: World Building Through Active Learningcited by 33
- 2019Learning to Reach Goals via Iterated Supervised Learningcited by 26
- 2019Self-supervised Learning of Image Embedding for Continuous Controlcited by 16
- 2021Variational Empowerment as Representation Learning for Goal-Based Reinforcement Learningcited by 15
- 2019MULEX: Disentangling Exploitation from Exploration in Deep RLcited by 4
Built on by
Citing papers per year: 2022: 23; 2023: 29; 2024: 34; 2025: 32; 2026: 13.
131 papers.
Sort:
By area:
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 357
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2022Multi-Game Decision Transformerscited by 212
- 2022Evolving Curricula with Regret-Based Environment Designcited by 166
- 2022Goal-Conditioned Reinforcement Learning: Problems and Solutionscited by 145
- 2023Structured World Models from Human Videoscited by 144
- 2022Deep Hierarchical Planning from Pixelscited by 96
- 2022Don't Change the Algorithm, Change the Data: Exploratory Data for Offline Reinforcement Learningcited by 93
- 2024Foundation Policies with Hilbert Representationscited by 89
- 2023Bootstrap Your Own Skills: Learning to Solve New Tasks with Large Language Model Guidancecited by 85
- 2023METRA: Scalable Unsupervised RL with Metric-Aware Abstractioncited by 80
- 2024World Models for Autonomous Driving: An Initial Surveycited by 79
38 more
- 2024Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Makingcited by 53
- 2025Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Modelscited by 51
- 2022Skill-based Model-based Reinforcement Learningcited by 50
- 2025A Survey: Learning Embodied Intelligence from Physical Simulators and World Modelscited by 50
- 2023Controllability-Aware Unsupervised Skill Discoverycited by 49
- 2024Learning Multimodal Behaviors from Scratch with Diffusion Policy Gradientcited by 49
- 2026World Action Models: The Next Frontier in Embodied AIcited by 46
- 2023SMART: Self-supervised Multi-task pretrAining with contRol Transformerscited by 44
- 2025What Drives Success in Physical Planning with Joint-Embedding Predictive World Models?cited by 37
- 2022Choreographer: Learning and Adapting Skills in Imaginationcited by 31
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximizationcited by 30
- 2025Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failurescited by 30
- 2022Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulationcited by 29
- 2024Entity-Centric Reinforcement Learning for Object Manipulation from Pixelscited by 27
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worldscited by 26
- 2022An information-theoretic perspective on intrinsic motivation in reinforcement learning: a surveycited by 25
- 2023Learning Goal-Conditioned Policies Offline with Self-Supervised Reward Shapingcited by 25
- 2022The Free Energy Principle for Perception and Action: A Deep Learning Perspectivecited by 24
- 2023SPRINT: Scalable Policy Pre-Training via Language Instruction Relabelingcited by 24
- 2025Reward Models in Deep Reinforcement Learning: A Surveycited by 24
- 2023ALAN: Autonomously Exploring Robotic Agents in the Real Worldcited by 23
- 2023Accelerating Exploration with Unlabeled Prior Datacited by 21
- 2025Dream2Flow: Bridging Video Generation and Open-World Manipulation with 3D Object Flowcited by 21
- 2023Reparameterized Policy Learning for Multimodal Trajectory Optimizationcited by 20
- 2023Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Datacited by 19
- 2024Can a MISL Fly? Analysis and Ingredients for Mutual Information Skill Learningcited by 18
- 2024TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representationscited by 17
- 2022Understanding the Complexity Gains of Single-Task RL with a Curriculumcited by 16
- 2023Curious Replay for Model-based Adaptationcited by 16
- 2025A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Searchcited by 16
- 2023COPlanner: Plan to Roll Out Conservatively but to Explore Optimistically for Model-Based RLcited by 14
- 2024Solving Minimum-Cost Reach Avoid using Reinforcement Learningcited by 14
- 2022Latent State Marginalization as a Low-cost Approach for Improving Explorationcited by 13
- 2023Imitating Graph-Based Planning with Goal-Conditioned Policiescited by 13
- 2023Reward-Free Curricula for Training Robust World Modelscited by 13
- 2023Breadcrumbs to the Goal: Goal-Conditioned Exploration from Human-in-the-Loop Feedbackcited by 13
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learningcited by 13
- 2024CarDreamer: Open-Source Learning Platform for World Model based Autonomous Drivingcited by 13
50 most cited of 131 All 131 (JSON)
- 2026Going Beyond State-Reaching: Learning Abstractions for Intrinsically Motivated Option Discovery29 Sep 2026
- 2026ChronoSRL: Temporal Geometry for Self-Supervised Reinforcement Learning28 Sep 2026
- 2026Autonomy, Social Norms, and Alignment: Towards a Developmental Framework for Autonomous Artificial Agents10 Sep 2026
- 2026SeerGuard: A Safety Framework for Mobile GUI Agents via World Model Prediction17 Jul 2026
- 2026Learning Task-Sufficient World Models by Synergizing Agentic Exploration and Structured Modeling5 Jul 2026
- 2026Learning Generalizable Skill Policy with Data-Efficient Unsupervised RL1 Jul 2026
- 2026World Action Models: The Next Frontier in Embodied AI12 May 2026
- 2026Learning What Matters: Adaptive Information-Theoretic Objectives for Robot Exploration12 May 2026
- 2026Unifying Goal-Conditioned RL and Unsupervised Skill Learning via Control-Maximization7 May 2026
- 2026SVL: Goal-Conditioned Reinforcement Learning as Survival Learning19 Apr 2026
- 2026Maximum Entropy Behavior Exploration for Sim2Real Zero-Shot Reinforcement Learning26 Mar 2026
- 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Exploration8 Mar 2026
38 more
- 2026Learning from Demonstrations via Capability-Aware Goal Sampling13 Jan 2026
- 2025Dream2Flow: Bridging Video Generation and Open-World Manipulation with 3D Object Flow31 Dec 2025
- 2025What Drives Success in Physical Planning with Joint-Embedding Predictive World Models?30 Dec 2025
- 2025TEACH: Temporal Variance-Driven Curriculum for Reinforcement Learning28 Dec 2025
- 2025SIMA 2: A Generalist Embodied Agent for Virtual Worlds4 Dec 2025
- 2025SOMBRL: Scalable and Optimistic Model-Based RL25 Nov 2025
- 2025Periodic Skill Discovery5 Nov 2025
- 2025Information-Theoretic Policy Pre-Training with Empowerment7 Oct 2025
- 2025MultiModal Action Conditioned Video Generation2 Oct 2025
- 2025Context and Diversity Matter: The Emergence of In-Context Learning in World Models26 Sep 2025
- 2025AnySafe: Adapting Latent Safety Filters at Runtime via Safety Constraint Parameterization in the Latent Space23 Sep 2025
- 2025Categorical Policies: Multimodal Policy Learning and Exploration in Continuous Control19 Aug 2025
- 2025FOUNDER: Grounding Foundation Models in World Models for Open-Ended Embodied Decision Making15 Jul 2025
- 2025Preemptive Solving of Future Problems: Multitask Preplay in Humans and Machines8 Jul 2025
- 2025Epistemically-guided forward-backward exploration7 Jul 2025
- 2025A Survey: Learning Embodied Intelligence from Physical Simulators and World Models1 Jul 2025
- 2025Strict Subgoal Execution: Reliable Long-Horizon Planning in Hierarchical Reinforcement Learning26 Jun 2025
- 2025Reward Models in Deep Reinforcement Learning: A Survey18 Jun 2025
- 2025Intention-Conditioned Flow Occupancy Models10 Jun 2025
- 2025A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Search5 Jun 2025
- 2025Intrinsic Goals for Autonomous Agents: Model-Based Exploration in Virtual Zebrafish Predicts Ethological Behavior and Whole-Brain Dynamics30 May 2025
- 2025OSVI-WM: One-Shot Visual Imitation for Unseen Tasks using World-Model-Guided Trajectory Generation26 May 2025
- 2025DISCOVER: Automated Curricula for Sparse-Reward Reinforcement Learning26 May 2025
- 2025Imagine Beyond! Distributionally Robust Auto-Encoding for State Space Coverage in Online Reinforcement Learning23 May 2025
- 2025Temporal Distance-aware Transition Augmentation for Offline Model-based Reinforcement Learning19 May 2025
- 2025Imagine, Verify, Execute: Memory-guided Agentic Exploration with Vision-Language Models12 May 2025
- 2025Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failures1 May 2025
- 2025Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Models15 Apr 2025
- 2025Autonomous state-space segmentation for Deep-RL sparse reward scenarios4 Apr 2025
- 2025Focusing Robot Open-Ended Reinforcement Learning Through Users' Purposes16 Mar 2025
- 2025SENSEI: Semantic Exploration Guided by Foundation Models to Learn Versatile World Models3 Mar 2025
- 2025Simulus: Combining Improvements in Sample-Efficient World Model Agents17 Feb 2025
- 2025DHP: Discrete Hierarchical Planning for Hierarchical Reinforcement Learning Agents4 Feb 2025
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximization16 Dec 2024
- 2024Can a MISL Fly? Analysis and Ingredients for Mutual Information Skill Learning11 Dec 2024
- 2024Free Energy Projective Simulation (FEPS): Active inference with interpretability22 Nov 2024
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning7 Nov 2024
- 2024Pre-trained Visual Dynamics Representations for Efficient Policy Learning5 Nov 2024
50 newest of 131 All 131 (JSON)
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 357
- 2023Structured World Models from Human Videoscited by 144
- 2022Deep Hierarchical Planning from Pixelscited by 96
- 2024World Models for Autonomous Driving: An Initial Surveycited by 79
- 2025A Survey: Learning Embodied Intelligence from Physical Simulators and World Modelscited by 50
- 2026World Action Models: The Next Frontier in Embodied AIcited by 46
- 2025What Drives Success in Physical Planning with Joint-Embedding Predictive World Models?cited by 37
- 2025Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failurescited by 30
- 2022Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulationcited by 29
- 2023Reward-Free Curricula for Training Robust World Modelscited by 13
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learningcited by 13
- 2024CarDreamer: Open-Source Learning Platform for World Model based Autonomous Drivingcited by 13
12 most cited of 30 All 30 (JSON)
- 2026SeerGuard: A Safety Framework for Mobile GUI Agents via World Model Prediction17 Jul 2026
- 2026Learning Task-Sufficient World Models by Synergizing Agentic Exploration and Structured Modeling5 Jul 2026
- 2026World Action Models: The Next Frontier in Embodied AI12 May 2026
- 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Exploration8 Mar 2026
- 2025What Drives Success in Physical Planning with Joint-Embedding Predictive World Models?30 Dec 2025
- 2025MultiModal Action Conditioned Video Generation2 Oct 2025
- 2025Context and Diversity Matter: The Emergence of In-Context Learning in World Models26 Sep 2025
- 2025AnySafe: Adapting Latent Safety Filters at Runtime via Safety Constraint Parameterization in the Latent Space23 Sep 2025
- 2025FOUNDER: Grounding Foundation Models in World Models for Open-Ended Embodied Decision Making15 Jul 2025
- 2025A Survey: Learning Embodied Intelligence from Physical Simulators and World Models1 Jul 2025
- 2025OSVI-WM: One-Shot Visual Imitation for Unseen Tasks using World-Model-Guided Trajectory Generation26 May 2025
- 2025Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failures1 May 2025
12 newest of 30 All 30 (JSON)
- 2022Multi-Game Decision Transformerscited by 212
- 2022Don't Change the Algorithm, Change the Data: Exploratory Data for Offline Reinforcement Learningcited by 93
- 2025A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Searchcited by 16
- 2024D5RL: Diverse Datasets for Data-Driven Deep Reinforcement Learningcited by 11
- 2024Unsupervised-to-Online Reinforcement Learningcited by 11
- 2025Temporal Distance-aware Transition Augmentation for Offline Model-based Reinforcement Learningcited by 5
- 2024Planning Transformer: Long-Horizon Offline Reinforcement Learning with Planning Tokenscited by 4
- 2024Zero-Shot Offline Imitation Learning via Optimal Transportcited by 3
- 2026Learning from Demonstrations via Capability-Aware Goal Samplingcited by 1
- 2022Task-Agnostic Learning to Accomplish New Taskscited by 0
- 2024Offline Policy Learning via Skill-step Abstraction for Long-horizon Goal-Conditioned Taskscited by 0
- 2026Learning from Demonstrations via Capability-Aware Goal Sampling13 Jan 2026
- 2025A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Search5 Jun 2025
- 2025Temporal Distance-aware Transition Augmentation for Offline Model-based Reinforcement Learning19 May 2025
- 2024Zero-Shot Offline Imitation Learning via Optimal Transport11 Oct 2024
- 2024Planning Transformer: Long-Horizon Offline Reinforcement Learning with Planning Tokens14 Sep 2024
- 2024Unsupervised-to-Online Reinforcement Learning27 Aug 2024
- 2024Offline Policy Learning via Skill-step Abstraction for Long-horizon Goal-Conditioned Tasks21 Aug 2024
- 2024D5RL: Diverse Datasets for Data-Driven Deep Reinforcement Learning15 Aug 2024
- 2022Task-Agnostic Learning to Accomplish New Tasks9 Sep 2022
- 2022Multi-Game Decision Transformers30 May 2022
- 2022Don't Change the Algorithm, Change the Data: Exploratory Data for Offline Reinforcement Learning31 Jan 2022
- 2022Contrastive Learning as Goal-Conditioned Reinforcement Learningcited by 241
- 2022Evolving Curricula with Regret-Based Environment Designcited by 166
- 2022Goal-Conditioned Reinforcement Learning: Problems and Solutionscited by 145
- 2024Foundation Policies with Hilbert Representationscited by 89
- 2023METRA: Scalable Unsupervised RL with Metric-Aware Abstractioncited by 80
- 2024Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Makingcited by 53
- 2025Zero-Shot Whole-Body Humanoid Control via Behavioral Foundation Modelscited by 51
- 2022Skill-based Model-based Reinforcement Learningcited by 50
- 2023Controllability-Aware Unsupervised Skill Discoverycited by 49
- 2023SMART: Self-supervised Multi-task pretrAining with contRol Transformerscited by 44
- 2022Choreographer: Learning and Adapting Skills in Imaginationcited by 31
- 2024MaxInfoRL: Boosting exploration in reinforcement learning through information gain maximizationcited by 30
12 most cited of 83 All 83 (JSON)
- 2026Going Beyond State-Reaching: Learning Abstractions for Intrinsically Motivated Option Discovery29 Sep 2026
- 2026ChronoSRL: Temporal Geometry for Self-Supervised Reinforcement Learning28 Sep 2026
- 2026Autonomy, Social Norms, and Alignment: Towards a Developmental Framework for Autonomous Artificial Agents10 Sep 2026
- 2026Learning Generalizable Skill Policy with Data-Efficient Unsupervised RL1 Jul 2026
- 2026Learning What Matters: Adaptive Information-Theoretic Objectives for Robot Exploration12 May 2026
- 2026Unifying Goal-Conditioned RL and Unsupervised Skill Learning via Control-Maximization7 May 2026
- 2026SVL: Goal-Conditioned Reinforcement Learning as Survival Learning19 Apr 2026
- 2026Maximum Entropy Behavior Exploration for Sim2Real Zero-Shot Reinforcement Learning26 Mar 2026
- 2025Dream2Flow: Bridging Video Generation and Open-World Manipulation with 3D Object Flow31 Dec 2025
- 2025TEACH: Temporal Variance-Driven Curriculum for Reinforcement Learning28 Dec 2025
- 2025SOMBRL: Scalable and Optimistic Model-Based RL25 Nov 2025
- 2025Periodic Skill Discovery5 Nov 2025
12 newest of 83 All 83 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026The Planning Limits of Latent World Models
- 2025GoalLadder: Incremental Goal Discovery with Vision-Language Models
- 2026One from Infinity: Actualizing Futures from Pretrained World Models into Robot Actions
- 2026Latent Action as Intention Enables Efficient Future Imagination for World Action Models
- 2023NoMaD: Goal Masked Diffusion Policies for Navigation and Exploration
- 2025Act2Goal: From World Model To General Goal-conditioned Policy
- 2026OpenFrontier: General Navigation with Visual-Language Grounded Frontiers
- 2024RoboDreamer: Learning Compositional World Models for Robot Imagination