Bayesian Optimization of Function Networks with Partial Evaluations

Read original: arXiv:2311.02146 - Published 6/17/2024 by Poompol Buathong, Jiayue Wan, Raul Astudillo, Samuel Daulton, Maximilian Balandat, Peter I. Frazier
Total Score

0

🛠️

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • This paper introduces a novel Bayesian optimization technique for efficiently optimizing complex function networks.
  • Existing methods for Bayesian optimization of function networks (BOFN) evaluate the full network at each iteration, which can be computationally expensive.
  • The authors propose a new knowledge gradient acquisition function that selects which node and corresponding inputs to evaluate, reducing the number of function evaluations required.
  • The proposed method outperforms existing BOFN approaches and other benchmarks across synthetic and real-world problems.

Plain English Explanation

Bayesian optimization is a powerful tool for finding the best settings or inputs to maximize an expensive or time-consuming function. This paper looks at a specific type of Bayesian optimization where the function being optimized is actually a network of functions, with each function taking the output of previous functions as input.

Optimizing these function networks can be very computationally intensive, as the entire network needs to be evaluated at each step. The authors of this paper propose a new way to perform this optimization that is more efficient. Instead of evaluating the full network every time, their method selects which individual function(s) to evaluate based on an acquisition function that considers the costs of evaluating each part of the network.

By only evaluating a subset of the network at each step, the authors show that their method can find the optimal settings more quickly than existing approaches, without sacrificing performance. This could be very useful for real-world problems where evaluating the entire function network is expensive, such as in machine learning model design or engineering design optimization.

Technical Explanation

The paper introduces a novel Bayesian optimization of function networks (BOFN) algorithm that leverages the underlying network structure to reduce the number of function evaluations required. Existing BOFN approaches evaluate the full network at each iteration, which can be computationally expensive, especially for large or complex networks.

To address this, the authors propose a knowledge gradient acquisition function that selects which node and corresponding inputs to evaluate in a cost-aware manner. This allows the method to focus evaluations on the most promising parts of the network, thereby reducing the overall query cost.

The authors provide an efficient approach to optimizing their acquisition function using a combination of techniques, including Thompson sampling and batch evaluation. They demonstrate that their method outperforms existing BOFN algorithms and other benchmarks across a range of synthetic and real-world problems, including hyperparameter tuning for machine learning models and engineering design optimization.

Critical Analysis

The paper presents a compelling approach to improving the efficiency of Bayesian optimization for function networks. By selectively evaluating only the most promising parts of the network, the authors are able to achieve significant performance gains without sacrificing the quality of the final solution.

One potential limitation is that the method assumes the ability to evaluate nodes individually, which may not always be the case in real-world applications. The authors acknowledge this and suggest that future work could explore ways to relax this assumption.

Additionally, the paper does not provide a detailed analysis of the computational complexity of the proposed algorithm, which would be useful for understanding its scalability to large-scale problems. It would also be interesting to see how the method compares to other recent advances in Bayesian optimization, such as preferential Bayesian optimization or Bayesian neural network surrogates.

Overall, the paper presents a novel and promising approach to Bayesian optimization that could have significant practical implications in a variety of domains. The authors have made a valuable contribution to the field, and their work warrants further exploration and development.

Conclusion

This paper introduces a novel Bayesian optimization technique for efficiently optimizing complex function networks. By leveraging the underlying network structure and using a cost-aware acquisition function, the proposed method is able to reduce the number of function evaluations required while maintaining high performance.

The authors have demonstrated the effectiveness of their approach across a range of synthetic and real-world problems, including hyperparameter tuning and engineering design optimization. This work represents an important advancement in the field of Bayesian optimization and could have significant implications for a variety of applications where efficient optimization of complex, expensive-to-evaluate functions is crucial.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

🛠️

Total Score

0

Bayesian Optimization of Function Networks with Partial Evaluations

Poompol Buathong, Jiayue Wan, Raul Astudillo, Samuel Daulton, Maximilian Balandat, Peter I. Frazier

Bayesian optimization is a powerful framework for optimizing functions that are expensive or time-consuming to evaluate. Recent work has considered Bayesian optimization of function networks (BOFN), where the objective function is given by a network of functions, each taking as input the output of previous nodes in the network as well as additional parameters. Leveraging this network structure has been shown to yield significant performance improvements. Existing BOFN algorithms for general-purpose networks evaluate the full network at each iteration. However, many real-world applications allow for evaluating nodes individually. To exploit this, we propose a novel knowledge gradient acquisition function that chooses which node and corresponding inputs to evaluate in a cost-aware manner, thereby reducing query costs by evaluating only on a part of the network at each step. We provide an efficient approach to optimizing our acquisition function and show that it outperforms existing BOFN methods and other benchmarks across several synthetic and real-world problems. Our acquisition function is the first to enable cost-aware optimization of a broad class of function networks.

Read more

6/17/2024

Bayesian Optimization of Functions over Node Subsets in Graphs
Total Score

0

Bayesian Optimization of Functions over Node Subsets in Graphs

Huidong Liang, Xingchen Wan, Xiaowen Dong

We address the problem of optimizing over functions defined on node subsets in a graph. The optimization of such functions is often a non-trivial task given their combinatorial, black-box and expensive-to-evaluate nature. Although various algorithms have been introduced in the literature, most are either task-specific or computationally inefficient and only utilize information about the graph structure without considering the characteristics of the function. To address these limitations, we utilize Bayesian Optimization (BO), a sample-efficient black-box solver, and propose a novel framework for combinatorial optimization on graphs. More specifically, we map each $k$-node subset in the original graph to a node in a new combinatorial graph and adopt a local modeling approach to efficiently traverse the latter graph by progressively sampling its subgraphs using a recursive algorithm. Extensive experiments under both synthetic and real-world setups demonstrate the effectiveness of the proposed BO framework on various types of graphs and optimization tasks, where its behavior is analyzed in detail with ablation studies.

Read more

5/27/2024

FunBO: Discovering Acquisition Functions for Bayesian Optimization with FunSearch
Total Score

0

FunBO: Discovering Acquisition Functions for Bayesian Optimization with FunSearch

Virginia Aglietti, Ira Ktena, Jessica Schrouff, Eleni Sgouritsa, Francisco J. R. Ruiz, Alan Malek, Alexis Bellot, Silvia Chiappa

The sample efficiency of Bayesian optimization algorithms depends on carefully crafted acquisition functions (AFs) guiding the sequential collection of function evaluations. The best-performing AF can vary significantly across optimization problems, often requiring ad-hoc and problem-specific choices. This work tackles the challenge of designing novel AFs that perform well across a variety of experimental settings. Based on FunSearch, a recent work using Large Language Models (LLMs) for discovery in mathematical sciences, we propose FunBO, an LLM-based method that can be used to learn new AFs written in computer code by leveraging access to a limited number of evaluations for a set of objective functions. We provide the analytic expression of all discovered AFs and evaluate them on various global optimization benchmarks and hyperparameter optimization tasks. We show how FunBO identifies AFs that generalize well in and out of the training distribution of functions, thus outperforming established general-purpose AFs and achieving competitive performance against AFs that are customized to specific function types and are learned via transfer-learning algorithms.

Read more

7/2/2024

🧠

Total Score

0

Large-Batch, Iteration-Efficient Neural Bayesian Design Optimization

Navid Ansari, Alireza Javanmardi, Eyke Hullermeier, Hans-Peter Seidel, Vahid Babaei

Bayesian optimization (BO) provides a powerful framework for optimizing black-box, expensive-to-evaluate functions. It is therefore an attractive tool for engineering design problems, typically involving multiple objectives. Thanks to the rapid advances in fabrication and measurement methods as well as parallel computing infrastructure, querying many design problems can be heavily parallelized. This class of problems challenges BO with an unprecedented setup where it has to deal with very large batches, shifting its focus from sample efficiency to iteration efficiency. We present a novel Bayesian optimization framework specifically tailored to address these limitations. Our key contribution is a highly scalable, sample-based acquisition function that performs a non-dominated sorting of not only the objectives but also their associated uncertainty. We show that our acquisition function in combination with different Bayesian neural network surrogates is effective in data-intensive environments with a minimal number of iterations. We demonstrate the superiority of our method by comparing it with state-of-the-art multi-objective optimizations. We perform our evaluation on two real-world problems -- airfoil design and 3D printing -- showcasing the applicability and efficiency of our approach. Our code is available at: https://github.com/an-on-ym-ous/lbn_mobo

Read more

9/6/2024