On Efficient Bayesian Exploration in Model-Based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
In this work, we address the challenge of data-efficient exploration in reinforcement learning by examining existing principled, information-theoretic approaches to intrinsic motivation. Specifically, we focus on a class of exploration bonuses that targets epistemic uncertainty rather than the aleatoric noise inherent in the environment. We prove that these bonuses naturally signal epistemic information gains and converge to zero once the agent becomes sufficiently certain about the environment's dynamics and rewards, thereby aligning exploration with genuine knowledge gaps. Our analysis provides formal guarantees for IG-based approaches, which previously lacked theoretical grounding. To enable practical use, we also discuss tractable approximations via sparse variational Gaussian Processes, Deep Kernels and Deep Ensemble models. We then outline a general framework - Predictive Trajectory Sampling with Bayesian Exploration (PTS-BE) - which integrates model-based planning with information-theoretic bonuses to achieve sample-efficient deep exploration. We empirically demonstrate that PTS-BE substantially outperforms other baselines across a variety of environments characterized by sparse rewards and/or purely exploratory tasks.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 39 of its 82 references that are in the map. First posted on arXiv 3 Jul 2025.
Builds on
39 of 82 references.
References · 39
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 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
- 2015Continuous control with deep reinforcement learningcited by 3,279
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2018Exploration by Random Network Distillationcited by 966
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2017Deep Bayesian Active Learning with Image Datacited by 812
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
27 more
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
- 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 632
- 2016Deep Exploration via Bootstrapped DQNcited by 518
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2015Deep Kernel Learningcited by 440
- 2013Gaussian Processes for Big Datacited by 377
- 2020Planning to Explore via Self-Supervised World Modelscited by 365
- 2016#Exploration: A Study of Count-Based Exploration for Deep Reinforcement Learningcited by 307
- 2019Self-Supervised Exploration via Disagreementcited by 263
- 2017Count-Based Exploration with Neural Density Modelscited by 255
- 2018Randomized Prior Functions for Deep Reinforcement Learningcited by 218
- 2022Exploration in Deep Reinforcement Learning: A Surveycited by 158
- 2015Incentivizing Exploration In Reinforcement Learning With Deep Predictive Modelscited by 154
- 2020Uncertainty Quantification and Deep Ensemblescited by 138
- 2022Quantifying Aleatoric and Epistemic Uncertainty in Machine Learning: Are Conditional Entropy and Mutual Information Appropriate Measures?cited by 124
- 2023Modern Bayesian Experimental Designcited by 121
- 2019Variational Bayesian Optimal Experimental Designcited by 116
- 2018Model-Based Active Explorationcited by 105
- 2021Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Designcited by 103
- 2016Stochastic Variational Deep Kernel Learningcited by 92
- 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
- 2018Efficient Exploration through Bayesian Deep Q-Networkscited by 67
- 2013Empowerment -- an Introductioncited by 65
- 2014Minimax Analysis of Active Learningcited by 61
- 2017Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Controlcited by 50
- 2021An Experimental Design Perspective on Model-Based Reinforcement Learningcited by 16
- 2018Optimization of Smooth Functions with Noisy Observations: Local Minimax Ratescited by 14
Built on by
Citing papers per year: 2026: 2.
2 papers.