Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation
Not on the map. Why?
Abstract (arXiv)
In model-free deep reinforcement learning (RL) algorithms, using noisy value estimates to supervise policy evaluation and optimization is detrimental to the sample efficiency. As this noise is heteroscedastic, its effects can be mitigated using uncertainty-based weights in the optimization process. Previous methods rely on sampled ensembles, which do not capture all aspects of uncertainty. We provide a systematic analysis of the sources of uncertainty in the noisy supervision that occurs in RL, and introduce inverse-variance RL, a Bayesian framework which combines probabilistic ensembles and Batch Inverse Variance weighting. We propose a method whereby two complementary uncertainty estimation methods account for both the Q-value and the environment stochasticity to better mitigate the negative impacts of noisy supervision. Our results show significant improvement in terms of sample efficiency on discrete and continuous control tasks.
Key facts. Cited by 26 papers in the map (data through 7 Oct 2026). Builds on 37 of its 52 references that are in the map. First posted on arXiv 5 Jan 2022.
Builds on
37 of 52 references.
References · 37
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,211
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,274
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2019Dota 2 with Large Scale Deep Reinforcement Learningcited by 1,084
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 833
25 more
- 2017A Distributional Perspective on Reinforcement Learningcited by 743
- 2017Deep Reinforcement Learning that Matterscited by 727
- 2016Deep Exploration via Bootstrapped DQNcited by 518
- 2019Quantifying the Carbon Emissions of Machine Learningcited by 516
- 2018Accurate Uncertainties for Deep Learning Using Calibrated Regressioncited by 417
- 2019Challenges of Real-World Reinforcement Learningcited by 282
- 2017Noisy Networks for Explorationcited by 262
- 2018Randomized Prior Functions for Deep Reinforcement Learningcited by 218
- 2019Benchmarking Model-Based Reinforcement Learningcited by 158
- 2017Deep Exploration via Randomized Value Functionscited by 155
- 2020SUNRISE: A Simple Unified Framework for Ensemble Learning in Deep Reinforcement Learningcited by 150
- 2019Behaviour Suite for Reinforcement Learningcited by 129
- 2021Uncertainty Weighted Actor-Critic for Offline Reinforcement Learningcited by 125
- 2018The Limits and Potentials of Deep Learning for Roboticscited by 104
- 2021DEUP: Direct Epistemic Uncertainty Predictioncited by 102
- 2019Evaluating and Calibrating Uncertainty Prediction in Regression Taskscited by 85
- 2020DisCor: Corrective Feedback in Reinforcement Learning via Distribution Correctioncited by 71
- 2019Estimating Risk and Uncertainty in Deep Reinforcement Learningcited by 68
- 2021Deep Reinforcement Learning for the Control of Robotic Manipulation: A Focussed Mini-Reviewcited by 34
- 2021DriverGym: Democratising Reinforcement Learning for Autonomous Drivingcited by 23
- 2018Reward Estimation for Variance Reduction in Deep Reinforcement Learningcited by 17
- 2020Temporal Difference Uncertainties as a Signal for Explorationcited by 15
- 2021State-Aware Variational Thompson Sampling for Deep Q-Networkscited by 5
- 2021Batch Inverse-Variance Weighting: Deep Heteroscedastic Regressioncited by 3
- 2021f-Cal: Calibrated aleatoric uncertainty estimation from neural networks for robot perceptioncited by 3
Built on by
Citing papers per year: 2022: 7; 2023: 3; 2024: 8; 2025: 5; 2026: 3.
26 papers.
Sort:
By area:
- 2022A Review of Uncertainty for Deep Reinforcement Learningcited by 36
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
- 2022DNS: Determinantal Point Process Based Neural Network Sampler for Ensemble Reinforcement Learningcited by 8
- 2023Uncertainty-Aware Decision Transformer for Stochastic Driving Environmentscited by 8
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPscited by 7
- 2024Learning to Play Atari in a World of Tokenscited by 6
- 2025Guaranteeing Out-Of-Distribution Detection in Deep RL via Transition Estimationcited by 6
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximationcited by 5
- 2025Uncertainty Prioritized Experience Replaycited by 5
- 2025Analytical Survey of Learning with Low-Resource Data: From Analysis to Investigationcited by 5
- 2022Nuclear Norm Maximization Based Curiosity-Driven Learningcited by 4
- 2024Contextualized Hybrid Ensemble Q-learning: Learning Fast with Control Priorscited by 4
14 more
- 2025SimBank: from Simulation to Solution in Prescriptive Process Monitoringcited by 4
- 2024A unified uncertainty-aware exploration: Combining epistemic and aleatory uncertaintycited by 2
- 2024Symmetric Reinforcement Learning Loss for Robust Learning on Diverse Tasks and Model Scalescited by 2
- 2024Active Reinforcement Learning Strategies for Offline Policy Improvementcited by 2
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Explorationcited by 2
- 2022Normality-Guided Distributional Reinforcement Learning for Continuous Controlcited by 1
- 2024Grid-Mapping Pseudo-Count Constraint for Offline Reinforcement Learningcited by 1
- 2026CRAD: Class-wise Reliability-Aware Distillation for Decentralized Heterogeneous Federated Learningcited by 1
- 2022Dynamic Memory-based Curiosity: A Bootstrap Approach for Explorationcited by 0
- 2022MEET: A Monte Carlo Exploration-Exploitation Trade-off for Buffer Samplingcited by 0
- 2023An Architecture for Deploying Reinforcement Learning in Industrial Environmentscited by 0
- 2024Generalized Gaussian Temporal Difference Error for Uncertainty-aware Reinforcement Learningcited by 0
- 2025SrSv: Integrating Sequential Rollouts with Sequential Value Estimation for Multi-agent Reinforcement Learningcited by 0
- 2026EasyPPO: Stabilizing the Critic Is Keycited by 0
- 2026EasyPPO: Stabilizing the Critic Is Key29 Sep 2026
- 2026CRAD: Class-wise Reliability-Aware Distillation for Decentralized Heterogeneous Federated Learning31 Aug 2026
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Exploration8 Jun 2026
- 2025Analytical Survey of Learning with Low-Resource Data: From Analysis to Investigation10 Oct 2025
- 2025Uncertainty Prioritized Experience Replay10 Jun 2025
- 2025SimBank: from Simulation to Solution in Prescriptive Process Monitoring28 Mar 2025
- 2025Guaranteeing Out-Of-Distribution Detection in Deep RL via Transition Estimation7 Mar 2025
- 2025SrSv: Integrating Sequential Rollouts with Sequential Value Estimation for Multi-agent Reinforcement Learning3 Mar 2025
- 2024Active Reinforcement Learning Strategies for Offline Policy Improvement17 Dec 2024
- 2024Generalized Gaussian Temporal Difference Error for Uncertainty-aware Reinforcement Learning5 Aug 2024
- 2024Contextualized Hybrid Ensemble Q-learning: Learning Fast with Control Priors28 Jun 2024
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximation24 Jun 2024
14 more
- 2024Learning to Play Atari in a World of Tokens3 Jun 2024
- 2024Symmetric Reinforcement Learning Loss for Robust Learning on Diverse Tasks and Model Scales27 May 2024
- 2024Grid-Mapping Pseudo-Count Constraint for Offline Reinforcement Learning3 Apr 2024
- 2024A unified uncertainty-aware exploration: Combining epistemic and aleatory uncertainty5 Jan 2024
- 2023Uncertainty-Aware Decision Transformer for Stochastic Driving Environments28 Sep 2023
- 2023An Architecture for Deploying Reinforcement Learning in Industrial Environments2 Jun 2023
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPs17 Mar 2023
- 2022MEET: A Monte Carlo Exploration-Exploitation Trade-off for Buffer Sampling24 Oct 2022
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient3 Oct 2022
- 2022Normality-Guided Distributional Reinforcement Learning for Continuous Control28 Aug 2022
- 2022Dynamic Memory-based Curiosity: A Bootstrap Approach for Exploration24 Aug 2022
- 2022A Review of Uncertainty for Deep Reinforcement Learning18 Aug 2022
- 2022Nuclear Norm Maximization Based Curiosity-Driven Learning21 May 2022
- 2022DNS: Determinantal Point Process Based Neural Network Sampler for Ensemble Reinforcement Learning31 Jan 2022
- 2024Learning to Play Atari in a World of Tokenscited by 6
- 2024Learning to Play Atari in a World of Tokens3 Jun 2024
- 2022A Review of Uncertainty for Deep Reinforcement Learningcited by 36
- 2022DNS: Determinantal Point Process Based Neural Network Sampler for Ensemble Reinforcement Learningcited by 8
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPscited by 7
- 2025Guaranteeing Out-Of-Distribution Detection in Deep RL via Transition Estimationcited by 6
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximationcited by 5
- 2025Uncertainty Prioritized Experience Replaycited by 5
- 2025Analytical Survey of Learning with Low-Resource Data: From Analysis to Investigationcited by 5
- 2022Nuclear Norm Maximization Based Curiosity-Driven Learningcited by 4
- 2024Contextualized Hybrid Ensemble Q-learning: Learning Fast with Control Priorscited by 4
- 2025SimBank: from Simulation to Solution in Prescriptive Process Monitoringcited by 4
- 2024A unified uncertainty-aware exploration: Combining epistemic and aleatory uncertaintycited by 2
- 2024Active Reinforcement Learning Strategies for Offline Policy Improvementcited by 2
12 most cited of 21 All 21 (JSON)
- 2026EasyPPO: Stabilizing the Critic Is Key29 Sep 2026
- 2026CRAD: Class-wise Reliability-Aware Distillation for Decentralized Heterogeneous Federated Learning31 Aug 2026
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Exploration8 Jun 2026
- 2025Analytical Survey of Learning with Low-Resource Data: From Analysis to Investigation10 Oct 2025
- 2025Uncertainty Prioritized Experience Replay10 Jun 2025
- 2025SimBank: from Simulation to Solution in Prescriptive Process Monitoring28 Mar 2025
- 2025Guaranteeing Out-Of-Distribution Detection in Deep RL via Transition Estimation7 Mar 2025
- 2025SrSv: Integrating Sequential Rollouts with Sequential Value Estimation for Multi-agent Reinforcement Learning3 Mar 2025
- 2024Active Reinforcement Learning Strategies for Offline Policy Improvement17 Dec 2024
- 2024Generalized Gaussian Temporal Difference Error for Uncertainty-aware Reinforcement Learning5 Aug 2024
- 2024Contextualized Hybrid Ensemble Q-learning: Learning Fast with Control Priors28 Jun 2024
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximation24 Jun 2024
12 newest of 21 All 21 (JSON)