NeST-BO: Fast Local Bayesian Optimization via Newton-Step Targeting of Gradient and Hessian Information
2510.05516v1
cs.LG, math.OC
2025-10-12
Авторы:
Wei-Ting Tang, Akshay Kudva, Joel A. Paulson
Abstract
Bayesian optimization (BO) is effective for expensive black-box problems but
remains challenging in high dimensions. We propose NeST-BO, a local BO method
that targets the Newton step by jointly learning gradient and Hessian
information with Gaussian process surrogates, and selecting evaluations via a
one-step lookahead bound on Newton-step error. We show that this bound (and
hence the step error) contracts with batch size, so NeST-BO directly inherits
inexact-Newton convergence: global progress under mild stability assumptions
and quadratic local rates once steps are sufficiently accurate. To scale, we
optimize the acquisition in low-dimensional subspaces (e.g., random embeddings
or learned sparse subspaces), reducing the dominant cost of learning curvature
from $O(d^2)$ to $O(m^2)$ with $m \ll d$ while preserving step targeting.
Across high-dimensional synthetic and real-world problems, including cases with
thousands of variables and unknown active subspaces, NeST-BO consistently
yields faster convergence and lower regret than state-of-the-art local and
high-dimensional BO baselines.
Ссылки и действия
Дополнительные ресурсы: