mlsanityData to 7 Oct 2026

Nonparametric Bayesian Optimization for General Rewards

arXiv 2602.07411 ↗First posted 7 Feb 2026cs.LGcited by 0

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

  1. 2018A Tutorial on Bayesian Optimizationcited by 734
  2. 2020NAS-Bench-201: Extending the Scope of Reproducible Neural Architecture Searchcited by 412
  3. 2018Gaussian Processes and Kernel Methods: A Review on Connections and Equivalencescited by 188
  4. 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
  5. 2023A Study of Bayesian Neural Network Surrogates for Bayesian Optimizationcited by 50
  6. 2022Non-Stationary Spatial Modelingcited by 13
  7. 2024Bayesian Optimization of Antibodies Informed by a Generative Model of Evolving Sequencescited by 10
  8. 2017Adaptive Rate of Convergence of Thompson Sampling for Gaussian Process Optimizationcited by 9
  9. 2023Pseudo-Bayesian Optimizationcited by 4
  10. 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

  1. 2018Nonparametric Gaussian Mixture Models for the Multi-Armed Bandit
  2. 2025Direct Regret Optimization in Bayesian Optimization
  3. 2025Order Optimal Regret Bounds for Sharpe Ratio Optimization under Thompson Sampling
  4. 2023The Choice of Noninformative Priors for Thompson Sampling in Multiparameter Bandit Models
  5. 2023Thompson Sampling for Linear Bandit Problems with Normal-Gamma Priors
  6. 2023Optimality of Thompson Sampling with Noninformative Priors for Pareto Bandits
  7. 2024Thompson Sampling for Stochastic Bandits with Noisy Contexts: An Information-Theoretic Regret Analysis
  8. 2023Reward Learning as Doubly Nonparametric Bandits: Optimal Design and Scaling Laws