When Is Generalizable Reinforcement Learning Tractable?
Not on the map. Why?
Abstract (arXiv)
Agents trained by reinforcement learning (RL) often fail to generalize beyond the environment they were trained in, even when presented with new scenarios that seem similar to the training environment. We study the query complexity required to train RL agents that generalize to multiple environments. Intuitively, tractable generalization is only possible when the environments are similar or close in some sense. To capture this, we introduce Weak Proximity, a natural structural condition that requires the environments to have highly similar transition and reward functions and share a policy providing optimal value. Despite such shared structure, we prove that tractable generalization is impossible in the worst case. This holds even when each individual environment can be efficiently solved to obtain an optimal linear policy, and when the agent possesses a generative model. Our lower bound applies to the more complex task of representation learning for the purpose of efficient generalization to multiple environments. On the positive side, we introduce Strong Proximity, a strengthened condition which we prove is sufficient for efficient generalization.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 31 of its 61 references that are in the map. First posted on arXiv 1 Jan 2021.
Builds on
31 of 61 references.
References · 31
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2019Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learningcited by 1,168
- 2017Deep Reinforcement Learning that Matterscited by 727
- 2019Leveraging Procedural Generation to Benchmark Reinforcement Learningcited by 478
- 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 424
- 2019Efficient Off-Policy Meta-Reinforcement Learning via Probabilistic Context Variablescited by 372
- 2018Quantifying Generalization in Reinforcement Learningcited by 345
- 2016Deep Reinforcement Learning for Robotic Manipulation with Asynchronous Off-Policy Updatescited by 342
- 2018Learning to Adapt in Dynamic, Real-World Environments Through Meta-Reinforcement Learningcited by 270
- 2015Actor-Mimic: Deep Multitask and Transfer Reinforcement Learningcited by 176
- 2019Is a Good Representation Sufficient for Sample Efficient Reinforcement Learning?cited by 140
- 2019Scalable methods for computing state similarity in deterministic Markov Decision Processescited by 133
19 more
- 2018Assessing Generalization in Deep Reinforcement Learningcited by 132
- 2019Model-Based Reinforcement Learning with a Generative Model is Minimax Optimalcited by 124
- 2019Learning with Good Feature Representations in Bandits and in RL with a Generative Modelcited by 124
- 2018Generalization and Regularization in DQNcited by 117
- 2019Non-Asymptotic Gap-Dependent Regret Bounds for Tabular MDPscited by 104
- 2019Observational Overfitting in Reinforcement Learningcited by 90
- 2020On Reward-Free Reinforcement Learning with Linear Function Approximationcited by 89
- 2018A Dissection of Overfitting and Generalization in Continuous Reinforcement Learningcited by 86
- 2018Composable Deep Reinforcement Learning for Robotic Manipulationcited by 70
- 2017Towards Generalization and Simplicity in Continuous Controlcited by 65
- 2018Gotta Learn Fast: A New Benchmark for Generalization in RLcited by 59
- 2013Sample Complexity of Multi-task Reinforcement Learningcited by 58
- 2020Invariant Policy Optimization: Towards Stronger Generalization in Reinforcement Learningcited by 37
- 2019Guided Meta-Policy Searchcited by 26
- 2019On the Generalization Gap in Reparameterizable Reinforcement Learningcited by 21
- 2020On the Global Optimality of Model-Agnostic Meta-Learningcited by 21
- 2020Instance based Generalization in Reinforcement Learningcited by 17
- 2019Comments on the Du-Kakade-Wang-Yang Lower Boundscited by 16
- 2019How Does an Approximate Model Help in Reinforcement Learning?cited by 4
Built on by
Citing papers per year: 2021: 4; 2022: 5; 2023: 1; 2024: 3; 2025: 3; 2026: 3.
19 papers.
Sort:
By area:
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 2023On the Importance of Exploration for Generalization in Reinforcement Learningcited by 30
- 2022Generalization in Cooperative Multi-Agent Systemscited by 18
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLcited by 17
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learningcited by 8
- 2022On the Power of Pre-training for Generalization in RL: Provable Benefits and Hardnesscited by 7
- 2021Sample Efficient Reinforcement Learning In Continuous State Spaces: A Perspective Beyond Linearitycited by 6
- 2022Meta Learning MDPs with Linear Transition Modelscited by 4
- 2022Deep Reinforcement Learning for Multi-Agent Interactioncited by 3
- 2025Provable Zero-Shot Generalization in Offline Reinforcement Learningcited by 3
- 2021Towards robust and domain agnostic reinforcement learning competitionscited by 2
7 more
- 2021Active Reinforcement Learning over MDPscited by 0
- 2024A Research Agenda for Usability and Generalisation in Reinforcement Learningcited by 0
- 2025Inductive Biases for Zero-shot Systematic Generalization in Language-informed Reinforcement Learningcited by 0
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contextscited by 0
- 2026Certificate-Guided Evaluation of Reinforcement Learning Generalizationcited by 0
- 2026A Model-Driven Approach for Developing Families of Reinforcement Learning Environmentscited by 0
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalizationcited by 0
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026A Model-Driven Approach for Developing Families of Reinforcement Learning Environments18 Jun 2026
- 2026Certificate-Guided Evaluation of Reinforcement Learning Generalization30 May 2026
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contexts10 Jul 2025
- 2025Provable Zero-Shot Generalization in Offline Reinforcement Learning11 Mar 2025
- 2025Inductive Biases for Zero-shot Systematic Generalization in Language-informed Reinforcement Learning25 Jan 2025
- 2024A Research Agenda for Usability and Generalisation in Reinforcement Learning22 Dec 2024
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RL26 Oct 2024
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learning4 Jan 2024
- 2023On the Importance of Exploration for Generalization in Reinforcement Learning8 Jun 2023
- 2022On the Power of Pre-training for Generalization in RL: Provable Benefits and Hardness19 Oct 2022
- 2022Deep Reinforcement Learning for Multi-Agent Interaction2 Aug 2022
7 more
- 2022Contextualize Me -- The Case for Context in Reinforcement Learning9 Feb 2022
- 2022Generalization in Cooperative Multi-Agent Systems31 Jan 2022
- 2022Meta Learning MDPs with Linear Transition Models21 Jan 2022
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learning18 Nov 2021
- 2021Active Reinforcement Learning over MDPs5 Aug 2021
- 2021Sample Efficient Reinforcement Learning In Continuous State Spaces: A Perspective Beyond Linearity15 Jun 2021
- 2021Towards robust and domain agnostic reinforcement learning competitions7 Jun 2021
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2022Contextualize Me -- The Case for Context in Reinforcement Learningcited by 41
- 2023On the Importance of Exploration for Generalization in Reinforcement Learningcited by 30
- 2022Generalization in Cooperative Multi-Agent Systemscited by 18
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLcited by 17
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learningcited by 8
- 2022On the Power of Pre-training for Generalization in RL: Provable Benefits and Hardnesscited by 7
- 2021Sample Efficient Reinforcement Learning In Continuous State Spaces: A Perspective Beyond Linearitycited by 6
- 2022Meta Learning MDPs with Linear Transition Modelscited by 4
- 2022Deep Reinforcement Learning for Multi-Agent Interactioncited by 3
- 2021Towards robust and domain agnostic reinforcement learning competitionscited by 2
- 2021Active Reinforcement Learning over MDPscited by 0
12 most cited of 18 All 18 (JSON)
- 2026Formalizing Task-Space Complexity for Zero-Shot Generalization18 Jun 2026
- 2026A Model-Driven Approach for Developing Families of Reinforcement Learning Environments18 Jun 2026
- 2026Certificate-Guided Evaluation of Reinforcement Learning Generalization30 May 2026
- 2025Zero-Shot Context Generalization in Reinforcement Learning from Few Training Contexts10 Jul 2025
- 2025Inductive Biases for Zero-shot Systematic Generalization in Language-informed Reinforcement Learning25 Jan 2025
- 2024A Research Agenda for Usability and Generalisation in Reinforcement Learning22 Dec 2024
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RL26 Oct 2024
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learning4 Jan 2024
- 2023On the Importance of Exploration for Generalization in Reinforcement Learning8 Jun 2023
- 2022On the Power of Pre-training for Generalization in RL: Provable Benefits and Hardness19 Oct 2022
- 2022Deep Reinforcement Learning for Multi-Agent Interaction2 Aug 2022
- 2022Contextualize Me -- The Case for Context in Reinforcement Learning9 Feb 2022
12 newest of 18 All 18 (JSON)