Efficient Off-Policy Meta-Reinforcement Learning via Probabilistic Context Variables
Not on the map. Why?
Cited by 47 in Offline RL and imitation learning from data, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Deep reinforcement learning algorithms require large amounts of experience to learn an individual task. While in principle meta-reinforcement learning (meta-RL) algorithms enable agents to learn new skills from small amounts of experience, several major challenges preclude their practicality. Current methods rely heavily on on-policy experience, limiting their sample efficiency. The also lack mechanisms to reason about task uncertainty when adapting to new tasks, limiting their effectiveness in sparse reward problems. In this paper, we address these challenges by developing an off-policy meta-RL algorithm that disentangles task inference and control. In our approach, we perform online probabilistic filtering of latent task variables to infer how to solve a new task from small amounts of experience. This probabilistic interpretation enables posterior sampling for structured and efficient exploration. We demonstrate how to integrate these task variables with off-policy RL algorithms to achieve both meta-training and adaptation efficiency. Our method outperforms prior algorithms in sample efficiency by 20-100X as well as in asymptotic performance on several meta-RL benchmarks.
Key facts. Cited by 372 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Mar 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 3; 2020: 3; 2021: 74; 2022: 77; 2023: 60; 2024: 54; 2025: 50; 2026: 51.
372 papers.
Sort:
By area:
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
- 2023Human-Timescale Adaptation in an Open-Ended Task Spacecited by 136
- 2023Large Language Models for Robotics: A Surveycited by 132
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2025Towards System 2 Reasoning in LLMs: Learning How to Think With Meta Chain-of-Thoughtcited by 108
- 2021Generalized Decision Transformer for Offline Hindsight Information Matchingcited by 105
- 2020Offline Meta-Reinforcement Learning with Advantage Weightingcited by 92
- 2022Few-Shot Preference Learning for Human-in-the-Loop RLcited by 85
- 2023Transformers as Decision Makers: Provable In-Context Reinforcement Learning via Supervised Pretrainingcited by 82
38 more
- 2022A Walk in the Park: Learning to Walk in 20 Minutes With Model-Free Reinforcement Learningcited by 76
- 2020Fighting Copycat Agents in Behavioral Cloning from Observation Historiescited by 70
- 2022CACTI: A Framework for Scalable Multi-Task Multi-Scene Visual Imitation Learningcited by 70
- 2022Automated Reinforcement Learning (AutoRL): A Survey and Open Problemscited by 69
- 2021A Survey of Exploration Methods in Reinforcement Learningcited by 68
- 2023MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RLcited by 65
- 2022Challenges and Opportunities in Offline Reinforcement Learning from Visual Observationscited by 63
- 2024Efficient Online Reinforcement Learning Fine-Tuning Need Not Retain Offline Datacited by 58
- 2021AdaAfford: Learning to Adapt Manipulation Affordance for 3D Articulated Objects via Few-shot Interactionscited by 56
- 2021Offline Meta-Reinforcement Learning with Online Self-Supervisioncited by 52
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2021Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learningcited by 49
- 2021AdaRL: What, Where, and How to Adapt in Transfer Reinforcement Learningcited by 46
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 41
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 2021Baby Intuitions Benchmark (BIB): Discerning the goals, preferences, and actions of otherscited by 39
- 2022Transformers are Meta-Reinforcement Learnerscited by 39
- 2022Hypernetworks in Meta-Reinforcement Learningcited by 39
- 2021Interesting Object, Curious Agent: Learning Task-Agnostic Explorationcited by 38
- 2021Offline Meta-Reinforcement Learning for Industrial Insertioncited by 37
- 2021Tell me why! Explanations support learning relational and causal structurecited by 36
- 2022Skill-based Meta-Reinforcement Learningcited by 36
- 2023AMAGO: Scalable In-Context Reinforcement Learning for Adaptive Agentscited by 36
- 2021Alchemy: A benchmark and analysis toolkit for meta-reinforcement learning agentscited by 35
- 2026MetaClaw: Just Talk -- An Agent That Meta-Learns and Evolves in the Wildcited by 35
- 2021Deep reinforcement learning in medical imaging: A literature reviewcited by 34
- 2021Recomposing the Reinforcement Learning Building Blocks with Hypernetworkscited by 33
- 2022Robust Task Representations for Offline Meta-Reinforcement Learning via Contrastive Learningcited by 33
- 2023End-to-End Meta-Bayesian Optimisation with Transformer Neural Processescited by 31
- 2024Unsupervised Zero-Shot Reinforcement Learning via Functional Reward Encodingscited by 31
- 2021Procedural Generalization by Planning with Self-Supervised World Modelscited by 30
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitationcited by 30
- 2023Cross-Domain Policy Adaptation via Value-Guided Data Filteringcited by 30
- 2021Coach-Player Multi-Agent Reinforcement Learning for Dynamic Team Compositioncited by 28
- 2022Don't Start From Scratch: Leveraging Prior Data to Automate Robotic Reinforcement Learningcited by 28
- 2019Watch, Try, Learn: Meta-Learning from Demonstrations and Rewardcited by 26
- 2022Context is Everything: Implicit Identification for Dynamics Adaptationcited by 26
- 2024FuRL: Visual-Language Models as Fuzzy Rewards for Reinforcement Learningcited by 26
50 most cited of 372 All 372 (JSON)
- 2026Meta-Multi-Agent Reinforcement Learning for Fast Adaptation of Interactive Policies with Applications to Autonomous Driving30 Sep 2026
- 2026Scalable In-Context Reinforcement Learning with Recurrent Algorithm Distillation28 Sep 2026
- 2026Nociception as a Control Primitive: Afferent Channels and Nociceptive Memory for Agents Deployed in One Body28 Sep 2026
- 2026From Weak Data to Strong Policy: Q-Targets Enable Provable In-Context Reinforcement Learning24 Sep 2026
- 2026Characterizing Replay Retention Under Dynamics Shift in Model-Based Reinforcement Learning16 Sep 2026
- 2026The Free Inference Dimension: Complexity Measure for Zero-Collision Navigation under Hypothesis Mixtures15 Sep 2026
- 2026Safe Meta-Reinforcement Learning via Information Space Reachability14 Sep 2026
- 2026Amortized Low-Rank Adaptation for Model-Based Reinforcement Learning10 Sep 2026
- 2026Proactive Context-Forecasted Safety Constraints for Nonstationary Reinforcement Learning8 Sep 2026
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning27 Aug 2026
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
38 more
- 2026SpikeWorld: Fast-State Adaptation for Frozen Spiking World Models7 Aug 2026
- 2026Does Latent Context Help? A Controlled Evaluation of Inverse Reinforcement Learning in Arctic Shipping6 Aug 2026
- 2026Diagnose, Recover, Certify: Task Readiness under Hidden Dynamics Changes30 Jul 2026
- 2026MetaKoopman: Bayesian Meta-Learning of Koopman Operators for Modeling Structured Dynamics under Distribution Shifts28 Jul 2026
- 2026Model of Models: When Does Emitting a Specialist Beat Attending, Adapting, or Tuning?20 Jul 2026
- 2026Building2Building: A Large Scale Benchmark for Generalizable Real-World Reinforcement Learning17 Jul 2026
- 2026Reflective VLA: In-Context Action Consequences Make VLAs Generalize23 Jun 2026
- 2026What Must Generalist Agents Remember?17 Jun 2026
- 2026Do as the Romans Do: Learning Universal Behaviors from Heterogeneous Agents16 Jun 2026
- 2026Knowledge Reutilization in Meta-Reinforcement Learning16 Jun 2026
- 2026VisualClaw: A Real-Time, Personalized Agent for the Physical World15 Jun 2026
- 2026FlowMo-WM: A World Model with Object Momentum and Hidden Ambient Drift11 Jun 2026
- 2026Learning to Route LLMs from Implicit Cost-Performance Preferences via Meta-Learning4 Jun 2026
- 2026Return-to-Go Is More Than a Number: Q-Guided Alignment for Return-Conditioned Supervised Learning27 May 2026
- 2026Transferable Reinforcement Learning via Probabilistic Latent Embeddings and Dynamic Policy Adaptation for Sim-to-Real Deployment26 May 2026
- 2026Curriculum reinforcement learning with measurable task representation learning22 May 2026
- 2026CoRMA: Contrastive RMA for Contact-Rich Meta-Adaptation21 May 2026
- 2026MATE: Solving Contextual Markov Decision Processes with Memory of Accumulated Transition Embeddings17 May 2026
- 2026BAPR: Bayesian amnesic piecewise-robust reinforcement learning for non-stationary continuous control15 May 2026
- 2026Zero-shot Imitation Learning by Latent Topology Mapping8 May 2026
- 2026LANTERN: LLM-Augmented Neurosymbolic Transfer with Experience-Gated Reasoning Networks6 May 2026
- 2026Neural Operators for Multi-Task Control and Adaptation3 Apr 2026
- 2026Factorization Regret mediates compositional generalization in latent space28 Mar 2026
- 2026Self Paced Gaussian Contextual Reinforcement Learning24 Mar 2026
- 2026MetaClaw: Just Talk -- An Agent That Meta-Learns and Evolves in the Wild17 Mar 2026
- 2026Few-Shot Adaptation to Non-Stationary Environments via Latent Trend Embedding for Robotics11 Mar 2026
- 2026Meta-RL with Shared Representations Enables Fast Adaptation in Energy Systems9 Mar 2026
- 2026On the Structural Non-Preservation of Epistemic Behaviour under Policy Transformation24 Feb 2026
- 2026Meta-Learning and Meta-Reinforcement Learning -- Tracing the Path towards DeepMind's Adaptive Agent23 Feb 2026
- 2026DCoPilot: Generative AI-Empowered Policy Adaptation for Dynamic Data Center Operations2 Feb 2026
- 2026Probabilistic Performance Guarantees for Multi-Task Reinforcement Learning2 Feb 2026
- 2026Unraveling the Hidden Dynamical Structure in Recurrent Neural Policies1 Feb 2026
- 2026Learning Policy Representations for Steerable Behavior Synthesis29 Jan 2026
- 2026Constrained Meta Reinforcement Learning with Provable Test-Time Safety29 Jan 2026
- 2026Few-Shot Learning for Dynamic Operations of Automated Electric Taxi Fleets under Evolving Charging Infrastructure: A Meta-Deep Reinforcement Learning Approach29 Jan 2026
- 2026CADENT: Gated Hybrid Distillation for Sample-Efficient Transfer in Reinforcement Learning28 Jan 2026
- 2026Safe Continual Reinforcement Learning Methods for Nonstationary Environments. Towards a Survey of the State of the Art8 Jan 2026
- 2026Tape: A Cellular Automata Benchmark for Evaluating Rule-Shift Generalization in Reinforcement Learning8 Jan 2026
50 newest of 372 All 372 (JSON)
- 2021Generalized Decision Transformer for Offline Hindsight Information Matchingcited by 105
- 2020Offline Meta-Reinforcement Learning with Advantage Weightingcited by 92
- 2023Transformers as Decision Makers: Provable In-Context Reinforcement Learning via Supervised Pretrainingcited by 82
- 2020Fighting Copycat Agents in Behavioral Cloning from Observation Historiescited by 70
- 2022CACTI: A Framework for Scalable Multi-Task Multi-Scene Visual Imitation Learningcited by 70
- 2022Challenges and Opportunities in Offline Reinforcement Learning from Visual Observationscited by 63
- 2024Efficient Online Reinforcement Learning Fine-Tuning Need Not Retain Offline Datacited by 58
- 2021Offline Meta-Reinforcement Learning with Online Self-Supervisioncited by 52
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2021Offline Meta-Reinforcement Learning for Industrial Insertioncited by 37
- 2022Robust Task Representations for Offline Meta-Reinforcement Learning via Contrastive Learningcited by 33
- 2019Watch, Try, Learn: Meta-Learning from Demonstrations and Rewardcited by 26
12 most cited of 45 All 45 (JSON)
- 2026Do as the Romans Do: Learning Universal Behaviors from Heterogeneous Agents16 Jun 2026
- 2026Zero-shot Imitation Learning by Latent Topology Mapping8 May 2026
- 2025Meta-Inverse Reinforcement Learning for Mean Field Games via Probabilistic Context Variables4 Sep 2025
- 2025Text-to-Decision Agent: Offline Meta-Reinforcement Learning from Natural Language Supervision21 Apr 2025
- 2025Yes, Q-learning Helps Offline In-Context RL24 Feb 2025
- 2024Entropy Regularized Task Representation Learning for Offline Meta-Reinforcement Learning19 Dec 2024
- 2024Efficient Online Reinforcement Learning Fine-Tuning Need Not Retain Offline Data10 Dec 2024
- 2024TEA: Trajectory Encoding Augmentation for Robust and Transferable Policies in Offline Reinforcement Learning28 Nov 2024
- 2024Bridging the Resource Gap: Deploying Advanced Imitation Learning Models onto Affordable Embedded Platforms18 Nov 2024
- 2024Continual Diffuser (CoD): Mastering Continual Offline Reinforcement Learning with Experience Rehearsal4 Sep 2024
- 2024Skills Regularized Task Decomposition for Multi-task Offline Reinforcement Learning28 Aug 2024
- 2024Residual Learning and Context Encoding for Adaptive Offline-to-Online Reinforcement Learning12 Jun 2024
12 newest of 45 All 45 (JSON)
- 2022A Walk in the Park: Learning to Walk in 20 Minutes With Model-Free Reinforcement Learningcited by 76
- 2021Augmented World Models Facilitate Zero-Shot Dynamics Generalization From a Single Offline Environmentcited by 41
- 2021Procedural Generalization by Planning with Self-Supervised World Modelscited by 30
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalizationcited by 16
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglementcited by 16
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learningcited by 13
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learningcited by 5
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalizationcited by 5
- 2022Prototypical context-aware dynamics generalization for high-dimensional model-based reinforcement learningcited by 3
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planningcited by 3
- 2023Meta-World Conditional Neural Processescited by 1
- 2026FlowMo-WM: A World Model with Object Momentum and Hidden Ambient Driftcited by 0
12 most cited of 14 All 14 (JSON)
- 2026Amortized Low-Rank Adaptation for Model-Based Reinforcement Learning10 Sep 2026
- 2026SpikeWorld: Fast-State Adaptation for Frozen Spiking World Models7 Aug 2026
- 2026FlowMo-WM: A World Model with Object Momentum and Hidden Ambient Drift11 Jun 2026
- 2025One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planning9 Sep 2025
- 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalization27 Aug 2025
- 2024Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement15 Oct 2024
- 2024Dreaming of Many Worlds: Learning Contextual World Models Aids Zero-Shot Generalization16 Mar 2024
- 2024MAMBA: an Effective World Model Approach for Meta-Reinforcement Learning14 Mar 2024
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learning9 Mar 2023
- 2023Meta-World Conditional Neural Processes20 Feb 2023
- 2022Prototypical context-aware dynamics generalization for high-dimensional model-based reinforcement learning23 Nov 2022
- 2022A Walk in the Park: Learning to Walk in 20 Minutes With Model-Free Reinforcement Learning16 Aug 2022
12 newest of 14 All 14 (JSON)
- 2022Self-Adaptive Driving in Nonstationary Environments through Conjectural Online Lookahead Adaptationcited by 12
- 2021Quick Learner Automated Vehicle Adapting its Roadmanship to Varying Traffic Cultures with Meta Reinforcement Learningcited by 3
- 2023Improving Deep Dynamics Models for Autonomous Vehicles with Multimodal Latent Mapping of Surfacescited by 1
- 2023Neurosymbolic Meta-Reinforcement Lookahead Learning Achieves Safe Self-Driving in Non-Stationary Environmentscited by 1
- 2022Cognitive Level-k Meta-Learning for Safe and Pedestrian-Aware Autonomous Drivingcited by 0
- 2026Meta-Multi-Agent Reinforcement Learning for Fast Adaptation of Interactive Policies with Applications to Autonomous Drivingcited by 0
- 2026Meta-Multi-Agent Reinforcement Learning for Fast Adaptation of Interactive Policies with Applications to Autonomous Driving30 Sep 2026
- 2023Neurosymbolic Meta-Reinforcement Lookahead Learning Achieves Safe Self-Driving in Non-Stationary Environments5 Sep 2023
- 2023Improving Deep Dynamics Models for Autonomous Vehicles with Multimodal Latent Mapping of Surfaces21 Mar 2023
- 2022Cognitive Level-k Meta-Learning for Safe and Pedestrian-Aware Autonomous Driving17 Dec 2022
- 2022Self-Adaptive Driving in Nonstationary Environments through Conjectural Online Lookahead Adaptation6 Oct 2022
- 2021Quick Learner Automated Vehicle Adapting its Roadmanship to Varying Traffic Cultures with Meta Reinforcement Learning18 Apr 2021
- 2026MetaClaw: Just Talk -- An Agent That Meta-Learns and Evolves in the Wildcited by 35
- 2024Large Language Models as Agents in Two-Player Gamescited by 10
- 2025Policy Search, Retrieval, and Composition via Task Similarity in Collaborative Agentic Systemscited by 0
- 2025A-LAMP: Agentic LLM-Based Framework for Automated MDP Modeling and Policy Generationcited by 0
- 2026MetaClaw: Just Talk -- An Agent That Meta-Learns and Evolves in the Wild17 Mar 2026
- 2025A-LAMP: Agentic LLM-Based Framework for Automated MDP Modeling and Policy Generation12 Dec 2025
- 2025Policy Search, Retrieval, and Composition via Task Similarity in Collaborative Agentic Systems5 Jun 2025
- 2024Large Language Models as Agents in Two-Player Games12 Feb 2024
- 2021MT-Opt: Continuous Multi-Task Robotic Reinforcement Learning at Scalecited by 226
- 2021How to Train Your Robot with Deep Reinforcement Learning; Lessons We've Learnedcited by 201
- 2022In-context Reinforcement Learning with Algorithm Distillationcited by 186
- 2023Supervised Pretraining Can Learn In-Context Reinforcement Learningcited by 137
- 2023Human-Timescale Adaptation in an Open-Ended Task Spacecited by 136
- 2023Large Language Models for Robotics: A Surveycited by 132
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2022Few-Shot Preference Learning for Human-in-the-Loop RLcited by 85
- 2022Automated Reinforcement Learning (AutoRL): A Survey and Open Problemscited by 69
- 2021A Survey of Exploration Methods in Reinforcement Learningcited by 68
- 2021AdaAfford: Learning to Adapt Manipulation Affordance for 3D Articulated Objects via Few-shot Interactionscited by 56
- 2021Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learningcited by 49
12 most cited of 290 All 290 (JSON)
- 2026Scalable In-Context Reinforcement Learning with Recurrent Algorithm Distillation28 Sep 2026
- 2026Nociception as a Control Primitive: Afferent Channels and Nociceptive Memory for Agents Deployed in One Body28 Sep 2026
- 2026From Weak Data to Strong Policy: Q-Targets Enable Provable In-Context Reinforcement Learning24 Sep 2026
- 2026Characterizing Replay Retention Under Dynamics Shift in Model-Based Reinforcement Learning16 Sep 2026
- 2026The Free Inference Dimension: Complexity Measure for Zero-Collision Navigation under Hypothesis Mixtures15 Sep 2026
- 2026Safe Meta-Reinforcement Learning via Information Space Reachability14 Sep 2026
- 2026Proactive Context-Forecasted Safety Constraints for Nonstationary Reinforcement Learning8 Sep 2026
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning27 Aug 2026
- 2026Dynamic Context Scheduling: Learning Beyond the Static Universe21 Aug 2026
- 2026Task Specialization Fine-Tuning for Contextual Reinforcement Learning17 Aug 2026
- 2026Diagnose, Recover, Certify: Task Readiness under Hidden Dynamics Changes30 Jul 2026
- 2026MetaKoopman: Bayesian Meta-Learning of Koopman Operators for Modeling Structured Dynamics under Distribution Shifts28 Jul 2026
12 newest of 290 All 290 (JSON)