BARK: A Fully Bayesian Tree Kernel for Black-box Optimization

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Boyne, Toby, Folch, Jose Pablo, Lee, Robert M, Shafei, Behrang, Misener, Ruth
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918047227314176
author Boyne, Toby
Folch, Jose Pablo
Lee, Robert M
Shafei, Behrang
Misener, Ruth
author_facet Boyne, Toby
Folch, Jose Pablo
Lee, Robert M
Shafei, Behrang
Misener, Ruth
contents We perform Bayesian optimization using a Gaussian process perspective on Bayesian Additive Regression Trees (BART). Our BART Kernel (BARK) uses tree agreement to define a posterior over piecewise-constant functions, and we explore the space of tree kernels using a Markov chain Monte Carlo approach. Where BART only samples functions, the resulting BARK model obtains samples of Gaussian processes defining distributions over functions, which allow us to build acquisition functions for Bayesian optimization. Our tree-based approach enables global optimization over the surrogate, even for mixed-feature spaces. Moreover, where many previous tree-based kernels provide uncertainty quantification over function values, our sampling scheme captures uncertainty over the tree structure itself. Our experiments show the strong performance of BARK on both synthetic and applied benchmarks, due to the combination of our fully Bayesian surrogate and the optimization procedure.
format Preprint
id arxiv_https___arxiv_org_abs_2503_05574
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BARK: A Fully Bayesian Tree Kernel for Black-box Optimization
Boyne, Toby
Folch, Jose Pablo
Lee, Robert M
Shafei, Behrang
Misener, Ruth
Machine Learning
Optimization and Control
We perform Bayesian optimization using a Gaussian process perspective on Bayesian Additive Regression Trees (BART). Our BART Kernel (BARK) uses tree agreement to define a posterior over piecewise-constant functions, and we explore the space of tree kernels using a Markov chain Monte Carlo approach. Where BART only samples functions, the resulting BARK model obtains samples of Gaussian processes defining distributions over functions, which allow us to build acquisition functions for Bayesian optimization. Our tree-based approach enables global optimization over the surrogate, even for mixed-feature spaces. Moreover, where many previous tree-based kernels provide uncertainty quantification over function values, our sampling scheme captures uncertainty over the tree structure itself. Our experiments show the strong performance of BARK on both synthetic and applied benchmarks, due to the combination of our fully Bayesian surrogate and the optimization procedure.
title BARK: A Fully Bayesian Tree Kernel for Black-box Optimization
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2503.05574