Policy Search via Bayesian Optimization with Temporal Difference Gaussian Processes
Armin Lederer, Anuj Srivastava, Marco Bagatella, Andreas Krause
Abstract
Bayesian optimization (BO) is a method commonly used for policy search in problems with low-dimensional policy parameterizations. While it is generally considered data-efficient, existing BO approaches are agnostic to the sequential structure of the optimization objective induced by policy roll-outs. Thereby, valuable information is discarded that could improve the convergence of BO. We address this inefficiency by developing and rigorously analyzing a novel approach for BO that relies on a temporal difference learning formulation for discounted infinite-horizon value functions based on Gaussian process (GP) regression. We derive learning error bounds for the proposed temporal difference GPs, such that we can exploit upper confidence bounds to analyze the cumulative regret of our BO approach. This analysis is further refined by bounding the maximal information gain for our temporal difference GP model. In a comparison with relevant baseline methods, we demonstrate the practical advantages of our method.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on15
- Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret BoundLin Yang, Mengdi WangICML 2020 · 308 citations
- Provably Efficient Reinforcement Learning for Discounted MDPs with Feature MappingDongruo Zhou, Jiafan He, Quanquan GuICML 2021 · 143 citations
- Information Theoretic Regret Bounds for Online Nonlinear ControlSham M. Kakade, Akshay Krishnamurthy, Kendall Lowrey, Motoya Ohnishi et al.NeurIPS 2020 · 137 citations
- Efficient Model-Based Reinforcement Learning through Optimistic Policy Search and PlanningSebastian Curi, Felix Berkenkamp, Andreas KrauseNeurIPS 2020 · 120 citations
- Practical and Rigorous Uncertainty Bounds for Gaussian Process RegressionChristian Fiedler, Carsten W. Scherer, Sebastian TrimpeAAAI 2021 · 92 citations
Related papers
- On Regret Bounds of Thompson Sampling for Bayesian OptimizationShion Takeno, Shogo IwazakiICML 2026 · 3 citations
- Objective Bound Conditional Gaussian Process for Bayesian OptimizationTaewon Jeong, Heeyoung KimICML 2021 · 3 citations
- Exploring and Exploiting Model Uncertainty in Bayesian OptimizationZishi Zhang, Tao Ren, Yijie PengNeurIPS 2025 · 1 citation
- Bayesian Optimization under Stochastic Delayed FeedbackArun Verma, Zhongxiang Dai, Bryan Kian Hsiang LowICML 2022 · 15 citations
- Trading Convergence Rate with Computational Budget in High Dimensional Bayesian OptimizationHung Tran-The, Sunil Gupta, Santu Rana, Svetha VenkateshAAAI 2020 · 14 citations
