BoGrape: Bayesian optimization over graphs with shortest-path encoded
Yilin Xie, Shiqiang Zhang, Jixiang Qing, Ruth Misener, Calvin Tsay
Abstract
Graph-structured data are central to many scientific and industrial applications where the goal is to optimize expensive black-box objectives defined over graph structures or node configurations -- as seen in molecular design, supply chains, and sensor placement. Bayesian optimization offers a principled approach for such settings, but existing methods largely focus on functions defined over nodes of a fixed graph. Moreover, graph optimization is often approached heuristically, and it remains unclear how to systematically incorporate structural constraints into BO. To address these gaps, we build on shortest-path graph kernels to develop a principled framework for acquisition optimization over unseen graph structures and associated node attributes. Through a novel formulation based on mixed-integer programming, we enable global exploration of the combinatorial domain over graph structures and explicit embedding of problem-specific constraints. We demonstrate that our method, BoGrape, is competitive both on general synthetic benchmarks and representative molecular design case studies with application-specific constraints.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c6034293-0fc0-4a55-bebc-c5b42ecc3408Builds on10
- BANANAS: Bayesian Optimization with Neural Architectures for Neural Architecture SearchColin White, Willie Neiswanger, Yash SavaniAAAI 2021 · 401 citations
- Interpretable Neural Architecture Search via Bayesian Optimisation with Weisfeiler-Lehman KernelsBin Xin Ru, Xingchen Wan, Xiaowen Dong, Michael A. OsborneICLR 2021 · 116 citations
- Partition-Based Formulations for Mixed-Integer Optimization of Trained ReLU Neural NetworksCalvin Tsay, Jan Kronqvist, Alexander Thebelt, Ruth MisenerNeurIPS 2021 · 93 citations
- Adversarial Attacks on Graph Classifiers via Bayesian OptimisationXingchen Wan, Henry Kenlay, Robin Ru, Arno Blaas et al.NeurIPS 2021 · 27 citations
- On Redundancy and Diversity in Cell-based Neural Architecture SearchXingchen Wan, Binxin Ru, Pedro M. Esperança, Zhenguo LiICLR 2022 · 27 citations
Related papers
- Mercer Features for Efficient Combinatorial Bayesian OptimizationAryan Deshwal, Syrine Belakaria, Janardhan Rao DoppaAAAI 2021 · 39 citations
- Bayesian Optimisation of Functions on GraphsXingchen Wan, Pierre Osselin, Henry Kenlay, Binxin Ru et al.NeurIPS 2023 · 9 citations
- Bayesian Optimization of Functions over Node Subsets in GraphsHuidong Liang, Xingchen Wan, Xiaowen DongNeurIPS 2024 · 3 citations
- Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial SpacesAryan Deshwal, Janardhan Rao DoppaNeurIPS 2021 · 65 citations
- Batched Energy-Entropy acquisition for Bayesian OptimizationFelix Teufel, Carsten Stahlhut, Jesper Ferkinghoff-BorgNeurIPS 2024 · 3 citations
