Generalization and Regularization in DQN
Not on the map. Why?
Abstract (arXiv)
Deep reinforcement learning algorithms have shown an impressive ability to learn complex control policies in high-dimensional tasks. However, despite the ever-increasing performance on popular benchmarks, policies learned by deep reinforcement learning algorithms can struggle to generalize when evaluated in remarkably similar environments. In this paper we propose a protocol to evaluate generalization in reinforcement learning through different modes of Atari 2600 games. With that protocol we assess the generalization capabilities of DQN, one of the most traditional deep reinforcement learning algorithms, and we provide evidence suggesting that DQN overspecializes to the training environment. We then comprehensively evaluate the impact of dropout and ℓ2 regularization, as well as the impact of reusing learned representations to improve the generalization capabilities of DQN. Despite regularization being largely underutilized in deep reinforcement learning, we show that it can, in fact, help DQN learn more general features. These features can be reused and fine-tuned on similar tasks, considerably improving DQN's sample efficiency.
Key facts. Cited by 117 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Sep 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 28; 2022: 28; 2023: 19; 2024: 25; 2025: 13; 2026: 4.
117 papers.
Sort:
By area:
- 2023The Rise and Potential of Large Language Model Based Agents: A Surveycited by 1,422
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2021Domain Generalization with MixStylecited by 563
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithmscited by 139
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 136
- 2021Contrastive Behavioral Similarity Embeddings for Generalization in Reinforcement Learningcited by 123
- 2023Loss of Plasticity in Continual Deep Reinforcement Learningcited by 118
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2021Scalable Evaluation of Multi-Agent Reinforcement Learning with Melting Potcited by 97
38 more
- 2021Decoupling Value and Policy for Generalization in Reinforcement Learningcited by 81
- 2022Offline Q-Learning on Diverse Multi-Task Data Both Scales And Generalizescited by 72
- 2021MixStyle Neural Networks for Domain Generalization and Adaptationcited by 64
- 2021Adversarially Guided Actor-Criticcited by 47
- 2021SECANT: Self-Expert Cloning for Zero-Shot Generalization of Visual Policiescited by 42
- 2023What Makes Pre-Trained Visual Representations Successful for Robust Manipulation?cited by 38
- 2022Melting Pot 2.0cited by 37
- 2021CORA: Benchmarks, Baselines, and Metrics as a Platform for Continual Reinforcement Learning Agentscited by 36
- 2025Hyperspherical Normalization for Scalable Deep Reinforcement Learningcited by 35
- 2022Look where you look! Saliency-guided Q-networks for generalization in visual Reinforcement Learningcited by 32
- 2021DRIBO: Robust Deep Reinforcement Learning via Multi-View Information Bottleneckcited by 31
- 2022Chain of Thought Imitation with Procedure Cloningcited by 30
- 2023On the Importance of Exploration for Generalization in Reinforcement Learningcited by 30
- 2023OCAtari: Object-Centric Atari 2600 Reinforcement Learning Environmentscited by 30
- 2021Robust Predictable Controlcited by 29
- 2023The Generalization Gap in Offline Reinforcement Learningcited by 25
- 2024Distributionally Robust Off-Dynamics Reinforcement Learning: Provable Efficiency with Linear Function Approximationcited by 25
- 2022Investigating the Properties of Neural Network Representations in Reinforcement Learningcited by 24
- 2021Adaptive Rational Activations to Boost Deep Reinforcement Learningcited by 23
- 2022A Comprehensive Survey of Data Augmentation in Visual Reinforcement Learningcited by 23
- 2021Return-based Scaling: Yet Another Normalisation Trick for Deep RLcited by 22
- 2022Mastering the Unsupervised Reinforcement Learning Benchmark from Pixelscited by 20
- 2023Explore to Generalize in Zero-Shot RLcited by 20
- 2021When Is Generalizable Reinforcement Learning Tractable?cited by 19
- 2021Towards Deeper Deep Reinforcement Learning with Spectral Normalizationcited by 19
- 2021Sequoia: A Software Framework to Unify Continual Learning Researchcited by 19
- 2022Online Policy Optimization for Robust MDPcited by 19
- 2023A Study of Global and Episodic Bonuses for Exploration in Contextual MDPscited by 19
- 2023Gradient-based Planning with World Modelscited by 18
- 2022Avalon: A Benchmark for RL Generalization Using Procedurally Generated Worldscited by 17
- 2024Dissecting Deep RL with High Update Ratios: Combatting Value Divergencecited by 17
- 2024Minimax Optimal and Computationally Efficient Algorithms for Distributionally Robust Offline Reinforcement Learningcited by 17
- 2022Regularized Q-learningcited by 16
- 2022Learning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributionscited by 15
- 2024Upper and Lower Bounds for Distributionally Robust Off-Dynamics Reinforcement Learningcited by 15
- 2025Horizon Generalization in Reinforcement Learningcited by 15
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learningcited by 14
- 2023Normalization Enhances Generalization in Visual Reinforcement Learningcited by 13
50 most cited of 117 All 117 (JSON)
- 2026AI for Games in the Foundation Model Era15 Sep 2026
- 2026Boosting deep Reinforcement Learning using pretraining with Logical Options6 Mar 2026
- 2026Hierarchical Successor Representation for Robust Transfer13 Feb 2026
- 2026Temporal Difference Learning with Constrained Initial Representations12 Feb 2026
- 2025On The Presence of Double-Descent in Deep Reinforcement Learning10 Nov 2025
- 2025The Formalism-Implementation Gap in Reinforcement Learning Research17 Oct 2025
- 2025Learning Distinguishable Representations in Deep Q-Networks for Linear Transfer29 Sep 2025
- 2025Sparsity-Driven Plasticity in Multi-Task Reinforcement Learning9 Aug 2025
- 2025Generalization in Reinforcement Learning for Radio Access Networks9 Jul 2025
- 2025Zero-Shot Reinforcement Learning Under Partial Observability18 Jun 2025
- 2025MCA-Bench: A Multimodal Benchmark for Evaluating CAPTCHA Robustness Against VLM-based Attacks6 Jun 2025
- 2025Bidirectional Distillation: A Mixed-Play Framework for Multi-Agent Generalizable Behaviors16 May 2025
38 more
- 2025Understanding the theoretical properties of projected Bellman equation, linear Q-learning, and approximate value iteration15 Apr 2025
- 2025Deep Reinforcement Learning via Object-Centric Attention3 Apr 2025
- 2025Hyperspherical Normalization for Scalable Deep Reinforcement Learning21 Feb 2025
- 2025Horizon Generalization in Reinforcement Learning6 Jan 2025
- 2025Representation Convergence: Mutual Distillation is Secretly a Form of Regularization5 Jan 2025
- 2024A Research Agenda for Usability and Generalisation in Reinforcement Learning22 Dec 2024
- 2024SNN-Based Online Learning of Concepts and Action Laws in an Open World19 Nov 2024
- 2024CALE: Continuous Arcade Learning Environment31 Oct 2024
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2024Upper and Lower Bounds for Distributionally Robust Off-Dynamics Reinforcement Learning30 Sep 2024
- 2024Curricula for Learning Robust Policies with Factored State Representations in Changing Environments13 Sep 2024
- 2024Mitigating Information Loss in Tree-Based Reinforcement Learning via Direct Optimization16 Aug 2024
- 2024SigmaRL: A Sample-Efficient and Generalizable Multi-Agent Reinforcement Learning Framework for Motion Planning14 Aug 2024
- 2024Towards Generalizable Reinforcement Learning via Causality-Guided Self-Adaptive Representations30 Jul 2024
- 2024Evaluation of Reinforcement Learning for Autonomous Penetration Testing using A3C, Q-learning and DQN22 Jul 2024
- 2024The Overcooked Generalisation Challenge: Evaluating Cooperation with Novel Partners in Unknown Environments Using Unsupervised Environment Design25 Jun 2024
- 2024The Perils of Optimizing Learned Reward Functions: Low Training Error Does Not Guarantee Low Regret22 Jun 2024
- 2024Investigating Pre-Training Objectives for Generalization in Vision-Based Reinforcement Learning10 Jun 2024
- 2024Massively Multiagent Minigames for Training Generalist Agents7 Jun 2024
- 2024HackAtari: Atari Learning Environments for Robust and Continual Reinforcement Learning6 Jun 2024
- 2024Scaling Laws for Reward Model Overoptimization in Direct Alignment Algorithms5 Jun 2024
- 2024Improving Generalization in Aerial and Terrestrial Mobile Robots Control Through Delayed Policy Learning4 Jun 2024
- 2024Intervention-Assisted Policy Gradient Methods for Online Stochastic Queuing Network Optimization: Technical Report5 Apr 2024
- 2024Minimax Optimal and Computationally Efficient Algorithms for Distributionally Robust Offline Reinforcement Learning14 Mar 2024
- 2024Do Agents Dream of Electric Sheep?: Improving Generalization in Reinforcement Learning through Generative Learning12 Mar 2024
- 2024Dissecting Deep RL with High Update Ratios: Combatting Value Divergence9 Mar 2024
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RL6 Mar 2024
- 2024Distributionally Robust Off-Dynamics Reinforcement Learning: Provable Efficiency with Linear Function Approximation23 Feb 2024
- 2024Learning Interpretable Policies in Hindsight-Observable POMDPs through Partially Supervised Reinforcement Learning14 Feb 2024
- 2024Simplifying Latent Dynamics with Softly State-Invariant World Models31 Jan 2024
- 2023Gradient-based Planning with World Models28 Dec 2023
- 2023MaDi: Learning to Mask Distractions for Generalization in Visual Deep Reinforcement Learning23 Dec 2023
- 2023The Generalization Gap in Offline Reinforcement Learning10 Dec 2023
- 2023What Makes Pre-Trained Visual Representations Successful for Robust Manipulation?3 Nov 2023
- 2023The Rise and Potential of Large Language Model Based Agents: A Survey14 Sep 2023
- 2023Measuring and Mitigating Interference in Reinforcement Learning10 Jul 2023
- 2023OCAtari: Object-Centric Atari 2600 Reinforcement Learning Environments14 Jun 2023
- 2023Design Principles for Model Generalization and Scalable AI Integration in Radio Access Networks9 Jun 2023
50 newest of 117 All 117 (JSON)
- 2023Gradient-based Planning with World Modelscited by 18
- 2024Simplifying Latent Dynamics with Softly State-Invariant World Modelscited by 10
- 2026AI for Games in the Foundation Model Eracited by 2
- 2026AI for Games in the Foundation Model Era15 Sep 2026
- 2024Simplifying Latent Dynamics with Softly State-Invariant World Models31 Jan 2024
- 2023Gradient-based Planning with World Models28 Dec 2023
- 2022Chain of Thought Imitation with Procedure Cloningcited by 30
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learningcited by 14
- 2024BlendRL: A Framework for Merging Symbolic and Neural Policy Learning15 Oct 2024
- 2022Chain of Thought Imitation with Procedure Cloning22 May 2022
- 2023The Rise and Potential of Large Language Model Based Agents: A Surveycited by 1,422
- 2021Learning robust driving policies without online exploration15 Mar 2021
- 2021Domain Generalization with MixStylecited by 563
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 136
- 2021Contrastive Behavioral Similarity Embeddings for Generalization in Reinforcement Learningcited by 123
- 2023Loss of Plasticity in Continual Deep Reinforcement Learningcited by 118
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2021Scalable Evaluation of Multi-Agent Reinforcement Learning with Melting Potcited by 97
- 2021Decoupling Value and Policy for Generalization in Reinforcement Learningcited by 81
- 2021MixStyle Neural Networks for Domain Generalization and Adaptationcited by 64
- 2021Adversarially Guided Actor-Criticcited by 47
12 most cited of 104 All 104 (JSON)
- 2026Boosting deep Reinforcement Learning using pretraining with Logical Options6 Mar 2026
- 2026Hierarchical Successor Representation for Robust Transfer13 Feb 2026
- 2026Temporal Difference Learning with Constrained Initial Representations12 Feb 2026
- 2025On The Presence of Double-Descent in Deep Reinforcement Learning10 Nov 2025
- 2025The Formalism-Implementation Gap in Reinforcement Learning Research17 Oct 2025
- 2025Learning Distinguishable Representations in Deep Q-Networks for Linear Transfer29 Sep 2025
- 2025Sparsity-Driven Plasticity in Multi-Task Reinforcement Learning9 Aug 2025
- 2025Generalization in Reinforcement Learning for Radio Access Networks9 Jul 2025
- 2025Zero-Shot Reinforcement Learning Under Partial Observability18 Jun 2025
- 2025MCA-Bench: A Multimodal Benchmark for Evaluating CAPTCHA Robustness Against VLM-based Attacks6 Jun 2025
- 2025Bidirectional Distillation: A Mixed-Play Framework for Multi-Agent Generalizable Behaviors16 May 2025
- 2025Understanding the theoretical properties of projected Bellman equation, linear Q-learning, and approximate value iteration15 Apr 2025
12 newest of 104 All 104 (JSON)