Meta-Learning Acquisition Functions for Transfer Learning in Bayesian Optimization
About
Transferring knowledge across tasks to improve data-efficiency is one of the open key challenges in the field of global black-box optimization. Readily available algorithms are typically designed to be universal optimizers and, therefore, often suboptimal for specific tasks. We propose a novel transfer learning method to obtain customized optimizers within the well-established framework of Bayesian optimization, allowing our algorithm to utilize the proven generalization capabilities of Gaussian processes. Using reinforcement learning to meta-train an acquisition function (AF) on a set of related tasks, the proposed method learns to extract implicit structural information and to exploit it for improved data-efficiency. We present experiments on a simulation-to-real transfer task as well as on several synthetic functions and on two hyperparameter search problems. The results show that our algorithm (1) automatically identifies structural properties of objective functions from available source tasks or simulations, (2) performs favourably in settings with both scarse and abundant source data, and (3) falls back to the performance level of general AFs if no particular structure is present.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Simple Regret | Synthetic GP SM periods=[0.2,0.4,0.8] 3D | Simple Regret0.00e+0 | 16 | |
| Simple Regret Minimization | Electrical Grid Stability Step 120 | Simple Regret (25th Pctl)0.00e+0 | 7 | |
| Simple Regret Minimization | HPOBench XGB Step 30 | 25th Percentile Simple Regret0.00e+0 | 7 | |
| Simple Regret Minimization | HPOBench XGB Step 60 | 25th Percentile Simple Regret0.00e+0 | 7 | |
| Simple Regret Minimization | Oil Step 30 | 25th Percentile Simple Regret0.00e+0 | 7 | |
| Simple Regret Minimization | Oil Step 60 | 25th Percentile Simple Regret0.00e+0 | 7 | |
| Simple Regret Minimization | Asteroid Step 30 | 25th Percentile Regret0.00e+0 | 7 | |
| Simple Regret Minimization | Asteroid Step 60 | Simple Regret (25th Pctl)0.00e+0 | 7 | |
| Simple Regret Minimization | Electrical Grid Stability Step 60 | 25th Percentile Simple Regret2.42 | 7 | |
| Simple Regret Minimization | PM2.5 Step 60 | 25th Percentile Simple Regret20 | 7 |