Nonparametric Bayesian Optimization for General Rewards
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
This work focuses on Bayesian optimization (BO) under reward model uncertainty. We propose the first BO algorithm that achieves no-regret guarantee in a general reward setting, requiring only Lipschitz continuity of the objective function and accommodating a broad class of measurement noise. The core of our approach is a novel surrogate model, termed as infinite Gaussian process (∞-GP). It is a Bayesian nonparametric model that places a prior on the space of reward distributions, enabling it to represent a substantially broader class of reward models than classical Gaussian process (GP). The ∞-GP is used in combination with Thompson Sampling (TS) to enable effective exploration and exploitation. Correspondingly, we develop a new TS regret analysis framework for general rewards, which relates the regret to the total variation distance between the surrogate model and the true reward distribution. Furthermore, with a truncated Gibbs sampling procedure, our method is computationally scalable, incurring minimal additional memory and computational complexities compared to classical GP. Empirical results demonstrate state-of-the-art performance, particularly in settings with non-stationary, heavy-tailed, or other ill-conditioned rewards.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 10 of its 73 references that are in the map. First posted on arXiv 7 Feb 2026.
Builds on
10 of 73 references.
References · 10
- 2018A Tutorial on Bayesian Optimizationcited by 734
- 2020NAS-Bench-201: Extending the Scope of Reproducible Neural Architecture Searchcited by 412
- 2018Gaussian Processes and Kernel Methods: A Review on Connections and Equivalencescited by 188
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2023A Study of Bayesian Neural Network Surrogates for Bayesian Optimizationcited by 50
- 2022Non-Stationary Spatial Modelingcited by 13
- 2024Bayesian Optimization of Antibodies Informed by a Generative Model of Evolving Sequencescited by 10
- 2017Adaptive Rate of Convergence of Thompson Sampling for Gaussian Process Optimizationcited by 9
- 2023Pseudo-Bayesian Optimizationcited by 4
- 2025On the convergence rate of noisy Bayesian Optimization with Expected Improvementcited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2018Nonparametric Gaussian Mixture Models for the Multi-Armed Bandit
- 2025Direct Regret Optimization in Bayesian Optimization
- 2025Order Optimal Regret Bounds for Sharpe Ratio Optimization under Thompson Sampling
- 2023The Choice of Noninformative Priors for Thompson Sampling in Multiparameter Bandit Models
- 2023Thompson Sampling for Linear Bandit Problems with Normal-Gamma Priors
- 2023Optimality of Thompson Sampling with Noninformative Priors for Pareto Bandits
- 2024Thompson Sampling for Stochastic Bandits with Noisy Contexts: An Information-Theoretic Regret Analysis
- 2023Reward Learning as Doubly Nonparametric Bandits: Optimal Design and Scaling Laws