mlsanityData to 7 Oct 2026

On Efficient Bayesian Exploration in Model-Based Reinforcement Learning

arXiv 2507.02639 ↗First posted 3 Jul 2025cs.LGcited by 2

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

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2017On Calibration of Modern Neural Networkscited by 4,860
  3. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
  4. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  5. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
  6. 2015Continuous control with deep reinforcement learningcited by 3,279
  7. 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
  8. 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
  9. 2018Exploration by Random Network Distillationcited by 966
  10. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
  11. 2017Deep Bayesian Active Learning with Image Datacited by 812
  12. 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
27 more
  1. 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 748
  2. 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 632
  3. 2016Deep Exploration via Bootstrapped DQNcited by 518
  4. 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
  5. 2015Deep Kernel Learningcited by 440
  6. 2013Gaussian Processes for Big Datacited by 377
  7. 2020Planning to Explore via Self-Supervised World Modelscited by 365
  8. 2016#Exploration: A Study of Count-Based Exploration for Deep Reinforcement Learningcited by 307
  9. 2019Self-Supervised Exploration via Disagreementcited by 263
  10. 2017Count-Based Exploration with Neural Density Modelscited by 255
  11. 2018Randomized Prior Functions for Deep Reinforcement Learningcited by 218
  12. 2022Exploration in Deep Reinforcement Learning: A Surveycited by 158
  13. 2015Incentivizing Exploration In Reinforcement Learning With Deep Predictive Modelscited by 154
  14. 2020Uncertainty Quantification and Deep Ensemblescited by 138
  15. 2022Quantifying Aleatoric and Epistemic Uncertainty in Machine Learning: Are Conditional Entropy and Mutual Information Appropriate Measures?cited by 124
  16. 2023Modern Bayesian Experimental Designcited by 121
  17. 2019Variational Bayesian Optimal Experimental Designcited by 116
  18. 2018Model-Based Active Explorationcited by 105
  19. 2021Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Designcited by 103
  20. 2016Stochastic Variational Deep Kernel Learningcited by 92
  21. 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
  22. 2018Efficient Exploration through Bayesian Deep Q-Networkscited by 67
  23. 2013Empowerment -- an Introductioncited by 65
  24. 2014Minimax Analysis of Active Learningcited by 61
  25. 2017Data-Efficient Reinforcement Learning with Probabilistic Model Predictive Controlcited by 50
  26. 2021An Experimental Design Perspective on Model-Based Reinforcement Learningcited by 16
  27. 2018Optimization of Smooth Functions with Noisy Observations: Local Minimax Ratescited by 14

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.