Network Randomization: A Simple Technique for Generalization in Deep Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
Deep reinforcement learning (RL) agents often fail to generalize to unseen environments (yet semantically similar to trained agents), particularly when they are trained on high-dimensional state spaces, such as images. In this paper, we propose a simple technique to improve a generalization ability of deep RL agents by introducing a randomized (convolutional) neural network that randomly perturbs input observations. It enables trained agents to adapt to new domains by learning robust features invariant across varied and randomized environments. Furthermore, we consider an inference method based on the Monte Carlo approximation to reduce the variance induced by this randomization. We demonstrate the superiority of our method across 2D CoinRun, 3D DeepMind Lab exploration and 3D robotics control tasks: it significantly outperforms various regularization and data augmentation methods for the same purpose.
Key facts. Cited by 102 papers in the map (data through 7 Oct 2026). First posted on arXiv 11 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 30; 2022: 23; 2023: 18; 2024: 14; 2025: 14; 2026: 3.
102 papers.
Sort:
By area:
- 2021Domain Generalization: A Surveycited by 760
- 2021Domain Generalization with MixStylecited by 563
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2021Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversitycited by 127
- 2021Contrastive Behavioral Similarity Embeddings for Generalization in Reinforcement Learningcited by 123
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2021Broaden Your Views for Self-Supervised Video Learningcited by 108
- 2021State Entropy Maximization with Random Encoders for Efficient Explorationcited by 107
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2022SURF: Semi-supervised Reward Learning with Data Augmentation for Feedback-efficient Preference-based Reinforcement Learningcited by 96
- 2021Decoupling Value and Policy for Generalization in Reinforcement Learningcited by 81
- 2021Vision-Language Navigation with Random Environmental Mixupcited by 70
38 more
- 2021MixStyle Neural Networks for Domain Generalization and Adaptationcited by 64
- 2021Cross-Domain Few-Shot Classification via Adversarial Task Augmentationcited by 63
- 2022Pre-Trained Image Encoder for Generalizable Visual Reinforcement Learningcited by 58
- 2021RobustNav: Towards Benchmarking Robustness in Embodied Navigationcited by 52
- 2021Adversarially Guided Actor-Criticcited by 47
- 2021SECANT: Self-Expert Cloning for Zero-Shot Generalization of Visual Policiescited by 42
- 2021MADE: Exploration via Maximizing Deviation from Explored Regionscited by 40
- 2022Generalizing Goal-Conditioned Reinforcement Learning with Variational Causal Reasoningcited by 33
- 2022Look where you look! Saliency-guided Q-networks for generalization in visual Reinforcement Learningcited by 32
- 2022Chain of Thought Imitation with Procedure Cloningcited by 30
- 2023On the Importance of Exploration for Generalization in Reinforcement Learningcited by 30
- 2023The Generalization Gap in Offline Reinforcement Learningcited by 25
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2022A Comprehensive Survey of Data Augmentation in Visual Reinforcement Learningcited by 23
- 2023GuidedMixup: An Efficient Mixup Strategy Guided by Saliency Mapscited by 23
- 2021Taxonomy of Machine Learning Safety: A Survey and Primercited by 22
- 2023Causal Reinforcement Learning: A Surveycited by 22
- 2022Mitigating Dataset Bias by Using Per-sample Gradientcited by 21
- 2023Explore to Generalize in Zero-Shot RLcited by 20
- 2021InsertionNet -- A Scalable Solution for Insertioncited by 15
- 2021Generalization of Reinforcement Learning with Policy-Aware Adversarial Data Augmentationcited by 15
- 2022Learning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributionscited by 15
- 2023MoVie: Visual Model-Based Policy Adaptation for View Generalizationcited by 15
- 2022Learning Deep Sensorimotor Policies for Vision-based Autonomous Drone Racingcited by 13
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractionscited by 13
- 2021Improving Generalization in Meta-RL with Imaginary Tasks from Latent Dynamics Mixturecited by 12
- 2024Causal Action Influence Aware Counterfactual Data Augmentationcited by 12
- 2023Guide Your Agent with Adaptive Multimodal Rewardscited by 10
- 2022Local Feature Swapping for Generalization in Reinforcement Learningcited by 9
- 2024Exploring Cross-Domain Few-Shot Classification via Frequency-Aware Promptingcited by 9
- 2025Gradient-Guided Annealing for Domain Generalizationcited by 9
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 8
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learningcited by 8
- 2021Scenic4RL: Programmatic Modeling and Generation of Reinforcement Learning Environmentscited by 6
- 2025Impoola: The Power of Average Pooling for Image-Based Deep Reinforcement Learningcited by 6
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learningcited by 5
- 2024Understanding What Affects the Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidencecited by 5
- 2024Focus On What Matters: Separated Models For Visual-Based RL Generalizationcited by 5
50 most cited of 102 All 102 (JSON)
- 2026Enhancing VLM Reward Models Through Structure-Aware Fine-Tuning4 Aug 2026
- 2026CAD-Free Learning of Spacecraft Pose Estimators via NeRF-Based Augmentations19 May 2026
- 2026Sample-Efficient Neurosymbolic Deep Reinforcement Learning6 Jan 2026
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulation15 Aug 2025
- 2025Self-Predictive Dynamics for Generalization of Vision-based Reinforcement Learning5 Jun 2025
- 2025Optimizing Sensory Neurons: Nonlinear Attention Mechanisms for Accelerated Convergence in Permutation-Invariant Neural Networks for Reinforcement Learning31 May 2025
- 2025How Ensembles of Distilled Policies Improve Generalisation in Reinforcement Learning22 May 2025
- 2025Bidirectional Distillation: A Mixed-Play Framework for Multi-Agent Generalizable Behaviors16 May 2025
- 2025Modeling Unseen Environments with Language-guided Composable Causal Components in Reinforcement Learning13 May 2025
- 2025Towards Fully Automated Decision-Making Systems for Greenhouse Control: Challenges and Opportunities27 Mar 2025
- 2025Provable Zero-Shot Generalization in Offline Reinforcement Learning11 Mar 2025
38 more
- 2025Impoola: The Power of Average Pooling for Image-Based Deep Reinforcement Learning7 Mar 2025
- 2025Gradient-Guided Annealing for Domain Generalization27 Feb 2025
- 2025Salience-Invariant Consistent Policy Learning for Generalization in Visual Reinforcement Learning12 Feb 2025
- 2025A Dual-Agent Adversarial Framework for Robust Generalization in Deep Reinforcement Learning29 Jan 2025
- 2025Representation Convergence: Mutual Distillation is Secretly a Form of Regularization5 Jan 2025
- 2024Reinforcement Learning from Wild Animal Videos5 Dec 2024
- 2024Reinforcement Learning with Euclidean Data Augmentation for State-Based Continuous Control16 Oct 2024
- 2024Training on Irrelevant States Implies Data Augmentation: Generalization in Contextual MDPs4 Oct 2024
- 2024Mitigating Adversarial Perturbations for Deep Reinforcement Learning via Vector Quantization4 Oct 2024
- 2024Focus On What Matters: Separated Models For Visual-Based RL Generalization29 Sep 2024
- 2024Exploring Cross-Domain Few-Shot Classification via Frequency-Aware Prompting24 Jun 2024
- 2024Explore-Go: Leveraging Exploration for Generalisation in Deep Reinforcement Learning12 Jun 2024
- 2024Improving Generalization in Aerial and Terrestrial Mobile Robots Control Through Delayed Policy Learning4 Jun 2024
- 2024Causal Action Influence Aware Counterfactual Data Augmentation29 May 2024
- 2024Do Agents Dream of Electric Sheep?: Improving Generalization in Reinforcement Learning through Generative Learning12 Mar 2024
- 2024Advancing Investment Frontiers: Industry-grade Deep Reinforcement Learning for Portfolio Optimization27 Feb 2024
- 2024Understanding What Affects the Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidence5 Feb 2024
- 2024Efficient Training of Generalizable Visuomotor Policies via Control-Aware Augmentation17 Jan 2024
- 2024A Survey Analyzing Generalization in Deep Reinforcement Learning4 Jan 2024
- 2023The Generalization Gap in Offline Reinforcement Learning10 Dec 2023
- 2023Guide Your Agent with Adaptive Multimodal Rewards19 Sep 2023
- 2023Adversarial Style Transfer for Robust Policy Optimization in Deep Reinforcement Learning29 Aug 2023
- 2023Causal Reinforcement Learning: A Survey4 Jul 2023
- 2023MoVie: Visual Model-Based Policy Adaptation for View Generalization3 Jul 2023
- 2023GuidedMixup: An Efficient Mixup Strategy Guided by Saliency Maps29 Jun 2023
- 2023VIBR: Learning View-Invariant Value Functions for Robust Visual Control14 Jun 2023
- 2023The Role of Diverse Replay for Generalisation in Reinforcement Learning9 Jun 2023
- 2023On the Importance of Exploration for Generalization in Reinforcement Learning8 Jun 2023
- 2023Explore to Generalize in Zero-Shot RL5 Jun 2023
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractions9 May 2023
- 2023Train a Real-world Local Path Planner in One Hour via Partially Decoupled Reinforcement Learning and Vectorized Diversity7 May 2023
- 2023Simple Noisy Environment Augmentation for Reinforcement Learning4 May 2023
- 2023Adversarial Policy Optimization in Deep Reinforcement Learning27 Apr 2023
- 2023Task Aware Dreamer for Task Generalization in Reinforcement Learning9 Mar 2023
- 2023Generalization in Visual Reinforcement Learning with the Reward Sequence Distribution19 Feb 2023
- 2023Robust Representation Learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractions12 Feb 2023
- 2023PAC-Bayesian Soft Actor-Critic Learning30 Jan 2023
- 2022Learning Generalizable Representations for Reinforcement Learning via Adaptive Meta-learner of Behavioral Similarities26 Dec 2022
50 newest of 102 All 102 (JSON)
- 2023The Generalization Gap in Offline Reinforcement Learningcited by 25
- 2023Guide Your Agent with Adaptive Multimodal Rewardscited by 10
- 2025Provable Zero-Shot Generalization in Offline Reinforcement Learningcited by 3
- 2022Model Generation with Provable Coverability for Offline Reinforcement Learningcited by 0
- 2025Provable Zero-Shot Generalization in Offline Reinforcement Learning11 Mar 2025
- 2023The Generalization Gap in Offline Reinforcement Learning10 Dec 2023
- 2023Guide Your Agent with Adaptive Multimodal Rewards19 Sep 2023
- 2022Model Generation with Provable Coverability for Offline Reinforcement Learning1 Jun 2022
- 2021Domain Generalization: A Surveycited by 760
- 2021Domain Generalization with MixStylecited by 563
- 2021A Survey of Zero-shot Generalisation in Deep Reinforcement Learningcited by 164
- 2021Co-Mixup: Saliency Guided Joint Mixup with Supermodular Diversitycited by 127
- 2021Contrastive Behavioral Similarity Embeddings for Generalization in Reinforcement Learningcited by 123
- 2021Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPscited by 115
- 2021Broaden Your Views for Self-Supervised Video Learningcited by 108
- 2021State Entropy Maximization with Random Encoders for Efficient Explorationcited by 107
- 2021Why Generalization in RL is Difficult: Epistemic POMDPs and Implicit Partial Observabilitycited by 102
- 2021Decoupling Value and Policy for Generalization in Reinforcement Learningcited by 81
- 2021Vision-Language Navigation with Random Environmental Mixupcited by 70
- 2021MixStyle Neural Networks for Domain Generalization and Adaptationcited by 64
12 most cited of 93 All 93 (JSON)
- 2026CAD-Free Learning of Spacecraft Pose Estimators via NeRF-Based Augmentations19 May 2026
- 2026Sample-Efficient Neurosymbolic Deep Reinforcement Learning6 Jan 2026
- 2025Reinforcement Learning Using known Invariances5 Nov 2025
- 2025Multi-Group Equivariant Augmentation for Reinforcement Learning in Robot Manipulation15 Aug 2025
- 2025Self-Predictive Dynamics for Generalization of Vision-based Reinforcement Learning5 Jun 2025
- 2025Optimizing Sensory Neurons: Nonlinear Attention Mechanisms for Accelerated Convergence in Permutation-Invariant Neural Networks for Reinforcement Learning31 May 2025
- 2025How Ensembles of Distilled Policies Improve Generalisation in Reinforcement Learning22 May 2025
- 2025Bidirectional Distillation: A Mixed-Play Framework for Multi-Agent Generalizable Behaviors16 May 2025
- 2025Modeling Unseen Environments with Language-guided Composable Causal Components in Reinforcement Learning13 May 2025
- 2025Towards Fully Automated Decision-Making Systems for Greenhouse Control: Challenges and Opportunities27 Mar 2025
- 2025Impoola: The Power of Average Pooling for Image-Based Deep Reinforcement Learning7 Mar 2025
- 2025Gradient-Guided Annealing for Domain Generalization27 Feb 2025
12 newest of 93 All 93 (JSON)