Resetting the Optimizer in Deep RL: An Empirical Study
Not on the map. Why?
Abstract (arXiv)
We focus on the task of approximating the optimal value function in deep reinforcement learning. This iterative process is comprised of solving a sequence of optimization problems where the loss function changes per iteration. The common approach to solving this sequence of problems is to employ modern variants of the stochastic gradient descent algorithm such as Adam. These optimizers maintain their own internal parameters such as estimates of the first-order and the second-order moments of the gradient, and update them over time. Therefore, information obtained in previous iterations is used to solve the optimization problem in the current iteration. We demonstrate that this can contaminate the moment estimates because the optimization landscape can change arbitrarily from one iteration to the next one. To hedge against this negative effect, a simple idea is to reset the internal parameters of the optimizer when starting a new iteration. We empirically investigate this resetting idea by employing various optimizers in conjunction with the Rainbow algorithm. We demonstrate that this simple modification significantly improves the performance of deep RL on the Atari benchmark.
Key facts. Cited by 26 papers in the map (data through 14 Oct 2026). Builds on 28 of its 53 references that are in the map. First posted on arXiv 30 Jun 2023.
Builds on
28 of 53 references.
References · 28
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,745
- 2015Prioritized Experience Replaycited by 1,099
- 2016An overview of gradient descent optimization algorithmscited by 1,033
- 2017Rainbow: Combining Improvements in Deep Reinforcement Learningcited by 786
- 2015Dueling Network Architectures for Deep Reinforcement Learningcited by 785
- 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 772
- 2017A Distributional Perspective on Reinforcement Learningcited by 765
- 2020Towards Continual Reinforcement Learning: A Review and Perspectivescited by 274
16 more
- 2022The Primacy Bias in Deep Reinforcement Learningcited by 247
- 2019A Theoretical Analysis of Deep Q-Learningcited by 215
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 125
- 2018Dopamine: A Research Framework for Deep Reinforcement Learningcited by 122
- 2023Loss of Plasticity in Continual Deep Reinforcement Learningcited by 121
- 2018A Closer Look at Deep Learning Heuristics: Learning rate restarts, Warmup and Distillationcited by 110
- 2023Deep Reinforcement Learning with Plasticity Injectioncited by 74
- 2020The Value-Improvement Path: Towards Better Representations for Reinforcement Learningcited by 60
- 2017Multi-step Reinforcement Learning: A Unifying Algorithmcited by 42
- 2020Scheduled Restart Momentum for Accelerated Stochastic Gradient Descentcited by 24
- 2020Lipschitz Lifelong Reinforcement Learningcited by 23
- 2021Correcting Momentum in Temporal Difference Learningcited by 12
- 2022Meta-Gradients in Non-Stationary Environmentscited by 11
- 2022The Nature of Temporal Difference Errors in Multi-step Distributional Reinforcement Learningcited by 10
- 2021Faster Deep Reinforcement Learning with Slower Online Networkcited by 7
- 2020Restarting Algorithms: Sometimes there is Free Lunchcited by 5
Built on by
Citing papers per year: 2024: 11; 2025: 5; 2026: 10.
26 papers.
Sort:
By area:
- 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 86
- 2024No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPOcited by 27
- 2024Streaming Deep Reinforcement Learning Finally Workscited by 25
- 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learningcited by 23
- 2025Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churncited by 22
- 2024Dissecting Deep RL with High Update Ratios: Combatting Value Divergencecited by 17
- 2024Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement Learningcited by 17
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?cited by 14
- 2024Adam on Local Time: Addressing Nonstationarity in RL with Relative Adam Timestepscited by 11
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learningcited by 11
- 2025Activation by Interval-wise Dropout: A Simple Way to Prevent Neural Networks from Plasticity Losscited by 6
- 2025Eau De Q-Network: Adaptive Distillation of Neural Networks in Deep Reinforcement Learningcited by 6
14 more
- 2026Understanding Quantization of Optimizer States in LLM Pre-training: Dynamics of State Staleness and Effectiveness of State Resetscited by 6
- 2024Reinforcement Learning From Imperfect Corrective Actions And Proxy Rewardscited by 3
- 2024Beyond the Boundaries of Proximal Policy Optimizationcited by 3
- 2026Do Neural Networks Lose Plasticity in a Gradually Changing World?cited by 2
- 2026The Rank and Gradient Lost in Non-stationarity: Sample Weight Decay for Mitigating Plasticity Loss in Reinforcement Learningcited by 2
- 2024Learning the Target Network in Function Spacecited by 1
- 2024Uncertainty Quantification with Bayesian Higher Order ReLU KANscited by 1
- 2026Rethinking Adam for Time Series Forecasting: A Simple Heuristic to Improve Optimization under Distribution Shiftscited by 1
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RLcited by 1
- 2026Can Scale Save Us From Plasticity Loss in Large Language Models?cited by 1
- 2026The Role of Target Update Frequencies in Q-Learningcited by 0
- 2026Rethinking Plasticity in Deep Reinforcement Learningcited by 0
- 2026On the Stability of Growth in Structural Plasticity: Forward-Active yet Backward-Starvedcited by 0
- 2026SingularClip: Preventing Spectral Collapse to Maintain Plasticity in Continual and Reinforcement Learningcited by 0
- 2026SingularClip: Preventing Spectral Collapse to Maintain Plasticity in Continual and Reinforcement Learning18 Aug 2026
- 2026Can Scale Save Us From Plasticity Loss in Large Language Models?23 Jun 2026
- 2026On the Stability of Growth in Structural Plasticity: Forward-Active yet Backward-Starved14 May 2026
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RL3 May 2026
- 2026The Rank and Gradient Lost in Non-stationarity: Sample Weight Decay for Mitigating Plasticity Loss in Reinforcement Learning2 Apr 2026
- 2026Rethinking Plasticity in Deep Reinforcement Learning22 Mar 2026
- 2026Understanding Quantization of Optimizer States in LLM Pre-training: Dynamics of State Staleness and Effectiveness of State Resets17 Mar 2026
- 2026Rethinking Adam for Time Series Forecasting: A Simple Heuristic to Improve Optimization under Distribution Shifts10 Mar 2026
- 2026Do Neural Networks Lose Plasticity in a Gradually Changing World?9 Feb 2026
- 2026The Role of Target Update Frequencies in Q-Learning3 Feb 2026
- 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning18 Jun 2025
- 2025Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churn31 May 2025
14 more
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learning25 May 2025
- 2025Eau De Q-Network: Adaptive Distillation of Neural Networks in Deep Reinforcement Learning3 Mar 2025
- 2025Activation by Interval-wise Dropout: A Simple Way to Prevent Neural Networks from Plasticity Loss3 Feb 2025
- 2024Adam on Local Time: Addressing Nonstationarity in RL with Relative Adam Timesteps22 Dec 2024
- 2024Beyond the Boundaries of Proximal Policy Optimization1 Nov 2024
- 2024Streaming Deep Reinforcement Learning Finally Works18 Oct 2024
- 2024Reinforcement Learning From Imperfect Corrective Actions And Proxy Rewards8 Oct 2024
- 2024Uncertainty Quantification with Bayesian Higher Order ReLU KANs2 Oct 2024
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?9 Jul 2024
- 2024Learning the Target Network in Function Space3 Jun 2024
- 2024Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement Learning26 May 2024
- 2024No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO1 May 2024
- 2024Dissecting Deep RL with High Update Ratios: Combatting Value Divergence9 Mar 2024
- 2024Disentangling the Causes of Plasticity Loss in Neural Networks29 Feb 2024
- 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 86
- 2024No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPOcited by 27
- 2024Streaming Deep Reinforcement Learning Finally Workscited by 25
- 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learningcited by 23
- 2025Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churncited by 22
- 2024Dissecting Deep RL with High Update Ratios: Combatting Value Divergencecited by 17
- 2024Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement Learningcited by 17
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?cited by 14
- 2024Adam on Local Time: Addressing Nonstationarity in RL with Relative Adam Timestepscited by 11
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learningcited by 11
- 2025Eau De Q-Network: Adaptive Distillation of Neural Networks in Deep Reinforcement Learningcited by 6
- 2024Beyond the Boundaries of Proximal Policy Optimizationcited by 3
12 most cited of 17 All 17 (JSON)
- 2026SingularClip: Preventing Spectral Collapse to Maintain Plasticity in Continual and Reinforcement Learning18 Aug 2026
- 2026The Rank and Gradient Lost in Non-stationarity: Sample Weight Decay for Mitigating Plasticity Loss in Reinforcement Learning2 Apr 2026
- 2026Rethinking Plasticity in Deep Reinforcement Learning22 Mar 2026
- 2026The Role of Target Update Frequencies in Q-Learning3 Feb 2026
- 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning18 Jun 2025
- 2025Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churn31 May 2025
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learning25 May 2025
- 2025Eau De Q-Network: Adaptive Distillation of Neural Networks in Deep Reinforcement Learning3 Mar 2025
- 2024Adam on Local Time: Addressing Nonstationarity in RL with Relative Adam Timesteps22 Dec 2024
- 2024Beyond the Boundaries of Proximal Policy Optimization1 Nov 2024
- 2024Streaming Deep Reinforcement Learning Finally Works18 Oct 2024
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?9 Jul 2024
12 newest of 17 All 17 (JSON)
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RLcited by 1
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RL3 May 2026
- 2025Activation by Interval-wise Dropout: A Simple Way to Prevent Neural Networks from Plasticity Losscited by 6
- 2026Understanding Quantization of Optimizer States in LLM Pre-training: Dynamics of State Staleness and Effectiveness of State Resetscited by 6
- 2026Do Neural Networks Lose Plasticity in a Gradually Changing World?cited by 2
- 2024Uncertainty Quantification with Bayesian Higher Order ReLU KANscited by 1
- 2026Rethinking Adam for Time Series Forecasting: A Simple Heuristic to Improve Optimization under Distribution Shiftscited by 1
- 2026Can Scale Save Us From Plasticity Loss in Large Language Models?cited by 1
- 2026On the Stability of Growth in Structural Plasticity: Forward-Active yet Backward-Starvedcited by 0
- 2026Can Scale Save Us From Plasticity Loss in Large Language Models?23 Jun 2026
- 2026On the Stability of Growth in Structural Plasticity: Forward-Active yet Backward-Starved14 May 2026
- 2026Understanding Quantization of Optimizer States in LLM Pre-training: Dynamics of State Staleness and Effectiveness of State Resets17 Mar 2026
- 2026Rethinking Adam for Time Series Forecasting: A Simple Heuristic to Improve Optimization under Distribution Shifts10 Mar 2026
- 2026Do Neural Networks Lose Plasticity in a Gradually Changing World?9 Feb 2026
- 2025Activation by Interval-wise Dropout: A Simple Way to Prevent Neural Networks from Plasticity Loss3 Feb 2025
- 2024Uncertainty Quantification with Bayesian Higher Order ReLU KANs2 Oct 2024