Cost-Aware Stopping for Freeze-Thaw Bayesian Optimization: When Does One-Step Lookahead Suffice?
Organizations: State University of New York Korea · Virginia Tech · Stony Brook University
Abstract
Freeze-thaw Bayesian optimization (BO) evaluates candidates in resumable increments but usually stops at a budget or convergence test. We maximize a budget-free utility, the best observed score minus a cost per increment, with a one-step lookahead: advance the candidate with the largest expected improvement (EI) under a fixed surrogate, and stop once no EI exceeds the cost. This rule, EI stopping, never falls below stopping immediately in expected utility, yet it can stop too early. We construct a certificate, a lower bound on the value of two-step lookahead, that proves such a stop premature, and repeating this correction never lowers expected utility under the surrogate. Under the surrogate, EI stops are premature on average at low costs and lose little at high costs. On three hyperparameter optimization benchmarks, a pre-registered test finds EI stopping noninferior to the correction in most high-cost settings, while the correction gains at low costs on two of the three; more information sharing in the surrogate raises EI stopping's utility. On the surrogates of two published methods, EI stopping trades lower scores for fewer training increments. EI stopping is a simple cost-aware reference whose utility depends on the surrogate.
Figures & tables
| LCBench | PD1 | ODBench | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Gate | H2-S EI-S | UCL | Verdict | Gate | H2-S EI-S | UCL | Verdict | Gate | H2-S EI-S | UCL | Verdict | |
| 0.0001 | 35% | open | 40% | open | 23% | open | ||||||
| 0.0002 | 27% | NI | 43% | open | 19% | open | ||||||
| 0.0005 | 22% | NI | 31% | open | 19% | open | ||||||
| 0.001 | 17% | NI | 20% | open | 29% | open | ||||||
| 0.002 | 13% | NI | 10% | open | 24% | open | ||||||
| FT-PFN | Deep-kernel GP | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| CFBO-NT, native stop, | ifBO, | DyHPO, | ||||||||||
| Bench. | ||||||||||||
| LCBench | 0.0001 | |||||||||||
| 0.0002 | ||||||||||||
| 0.0005 | ||||||||||||
| 0.001 | ||||||||||||
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Bellman | Two-step certificate and policies | |||
| history state after advances | , | optimal value and action value | largest available EI, augmented by | ||
| incumbent score, | centered advantage of arm | pool-aware two-step advantage | |||
| stop-now payoff | downstream option value | same-arm two-step advantage | |||
| available arms | , | Bellman and one-step stop sets | switching premium | ||
| headroom | EI stopping (EI-S) | certified lower enclosure | |||
| one-step expected improvement | H2 stopping (H2-S), the certified policy | episode utility | |||
| : 3 arms, 2 increments | : 2 arms, 3 increments | ||||||
| Instances | Optimal [95%] | Within [95%] | Recovered [95%] | Optimal [95%] | Within [95%] | Recovered [95%] | |
| LCBench | 280 | 74.6% [69, 79] | 95.4% [92, 97] | 99.8% [99.7, 100.0] | 60.4% [55, 66] | 77.1% [72, 82] | 85.0% [77, 91] |
| PD1 | 280 | 71.8% [66, 77] | 88.9% [85, 92] | 99.6% [99, 100] | 66.8% [61, 72] | 79.3% [74, 84] | 92.5% [87, 97] |
| ODBench | 280 | 71.1% [66, 76] | 85.0% [80, 89] | 98.5% [98, 99] | 55.4% [50, 61] | 69.3% [64, 74] | 81.4% [74, 87] |
| All | 840 | 72.5% [69, 75] | 89.8% [88, 92] | 99.2% [98.9, 99.5] | 60.8% [57, 64] | 75.2% [72, 78] | 85.5% [81, 89] |
| 120 | 74.2% [66, 81] | 83.3% [76, 89] | 99.3% [99, 100] | 51.7% [43, 60] | 56.7% [48, 65] | 74.3% [60, 84] | |
| Bounds under the planning law | Seeds 1–50 | Seeds 51–100 | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Bench. | Upper | Gaussian part | Lower | [limit] | Verdict | Realized | [limits] | Planning law | Realized | [limits] | |
| LCBench | 0.0001 | 1,509 | 36.9 | 18.4 | 17.9 | premature | +5.72 | [-0.460, +14.9] | 9.26 | +0.232 | [-0.657, +1.50] |
| 0.0002 | 2,708 | 30.5 | 11.7 | 11.2 | premature | +0.613 | [-0.971, +2.64] | 13.4 | +8.90 † | [+1.11, +18.0] | |
| 0.0005 | 1,282 | 34.5 | 7.75 | 7.47 | premature | +3.41 | [-0.0253, +8.63] | 2.86 | -0.750 ‡ ∗ | [-1.35, -0.298] | |
| 0.001 | 1,538 | 29.8 | 6.86 | 6.59 | premature | -1.01 | [-3.10, +0.228] | 7.29 | +1.02 | [-0.854, +3.49] | |
| 0.002 | 502 | 33.4 | 1.39 | 1.19 | open | +0.0102 | [-0.410, +0.546] | 3.52 | -0.700 ‡ | [-1.38, -0.166] | |
| Seeds 1–50 | Pre-registered test | ||||
|---|---|---|---|---|---|
| Bench. | Gate | H2-S EI-S [band] | UCL | H2-S EI-S [band] | |
| LCBench | 0.0001 | 34% | [-0.00008, +0.00036] | [-0.00015, +0.00021] | |
| 0.0002 | 25% | [-0.00018, +0.00035] | [-0.00017, +0.00015] | ||
| 0.0005 | 21% | [-0.00131, +0.00036] | [-0.00083, -0.00009] ‡ | ||
| 0.001 | 18% | [-0.00072, -0.00025] ‡ | [-0.00059, -0.00017] ‡ | ||
| 0.002 | 12% | [-0.00080, -0.00003] ‡ | [-0.00079, -0.00026] ‡ | ||
| Benchmark | Gate opened | Certified lower | Realized | Realized certified [band] | Realized |
|---|---|---|---|---|---|
| LCBench | 2,060 | +0.11 | -1.71 | -1.82 [-2.60, -1.13] | 94% |
| PD1 | 1,199 | +0.29 | +6.50 | +6.21 [+4.99, +7.40] | 78% |
| ODBench | 683 | +0.46 | +3.28 | +2.81 [+0.95, +4.77] | 83% |
| Surrogate | Bench. | 0.0001 | 0.0002 | 0.0005 | 0.001 | 0.002 | 0.005 | 0.01 |
|---|---|---|---|---|---|---|---|---|
| GP-at, seeds 51–100 | LCBench | open | open | NI | NI | NI | NI | NI |
| PD1 | open | open | open | open | open | open | NI | |
| ODBench | open | open | open | open | NI | NI | NI | |
| GP-a, seeds 51–100 | LCBench | open | open | NI | NI | NI | NI | NI |
| PD1 | open | open | open | open | open | NI | NI | |
| Surrogate | Bench. | 0.0001 | 0.0002 | 0.0005 | 0.001 | 0.002 | 0.005 | 0.01 |
|---|---|---|---|---|---|---|---|---|
| GP-atJ, seeds 1–50 | LCBench | open | open | NI | NI | NI | NI | NI |
| PD1 | open | open | open | open | open | NI | NI | |
| ODBench | open | open | open | open | open | NI | NI | |
| GP-at, seeds 1–50 | LCBench | open | open | NI | NI | NI | NI | NI |
| PD1 | open | open | open | open | open | NI | NI | |
| Seeds 1–50 | Fresh seeds 51–100 | ||||||
|---|---|---|---|---|---|---|---|
| Bench. | GP-at GP-atJ [band] | GP-a GP-atJ [band] | GP-at GP-atJ [band] | GP-a GP-atJ [band] | |||
| LCBench | 0.0001 | [-0.0011, -0.0002] ‡ | [-0.0014, -0.0004] ‡ | [-0.0007, -0.0000] ‡ | [-0.0008, -0.0002] ‡ | ||
| 0.0002 | [-0.0017, -0.0008] ‡ | [-0.0019, -0.0009] ‡ | [-0.0012, -0.0005] ‡ | [-0.0014, -0.0007] ‡ | |||
| 0.0005 | [-0.0030, -0.0012] ‡ | [-0.0029, -0.0010] ‡ | [-0.0035, -0.0010] ‡ | [-0.0030, -0.0007] ‡ | |||
| 0.001 | [-0.0026, -0.0009] ‡ | [-0.0036, -0.0016] ‡ | [-0.0023, -0.0005] ‡ | [-0.0029, -0.0008] ‡ | |||
| 0.002 | [-0.0035, -0.0015] ‡ | [-0.0047, -0.0025] ‡ | [-0.0033, -0.0015] ‡ | [-0.0042, -0.0023] ‡ | |||
| Seeds | Step | Mean | [Wilcoxon] | Pos. | Above | Below |
|---|---|---|---|---|---|---|
| 1–50 | GP-at GP-a | [ ] | 49/50 | 9 | 0 | |
| GP-atJ GP-at | [ ] | 47/50 | 14 | 0 | ||
| GP-atJ GP-a | [ ] | 49/50 | 16 | 0 | ||
| 51–100 | GP-at GP-a | [ ] | 47/50 | 8 | 0 | |
| GP-atJ GP-at | [ ] | 48/50 | 13 | 0 | ||
| GP-atJ GP-a | [ ] | 49/50 | 15 | 0 |
| Group | seeds 1–50 [band] | seeds 51–100 [band] | Tasks | Cap at EI stop | No cap hit | |
|---|---|---|---|---|---|---|
| (seeds 1–50; 51–100) | (seeds 1–50; 51–100) | |||||
| LCBench | 0.0001 | [-0.00009, +0.00037] | [-0.00016, +0.00021] | 13/35 ; 8/35 | ; | |
| LCBench | 0.0002 | [-0.00019, +0.00036] | [-0.00017, +0.00016] | 11/35 ; 9/35 | ; | |
| LCBench | 0.0005 | [-0.00135, +0.00040] | [-0.00084, -0.00007] ‡ | 8/35 ; 4/35 | ; | |
| LCBench | 0.001 | [-0.00073, -0.00024] ‡ | [-0.00060, -0.00017] ‡ | 5/35 ; 4/35 | ; | |
| LCBench | 0.002 | [-0.00082, -0.00002] ‡ | [-0.00080, -0.00025] ‡ | 6/35 ; 4/35 | ; |
| Bench. | Noise | Noise | Prior | Prior | Noise | Noise | Prior | Prior | |
|---|---|---|---|---|---|---|---|---|---|
| LCBench | 0.0001 | ||||||||
| 0.0002 | |||||||||
| 0.0005 | |||||||||
| 0.001 | |||||||||
| 0.002 | |||||||||
| Bench. | Primary | Noise | Noise | Prior | Prior | Calibrated noise | |
|---|---|---|---|---|---|---|---|
| LCBench | 0.0001 | ||||||
| 0.0002 | |||||||
| 0.0005 | |||||||
| 0.001 | |||||||
| 0.002 | |||||||
| 0.005 |
| FT-PFN | Deep-kernel GP | |||||
| CFBO-NT [band] | ifBO, [band] | DyHPO, [band] | EI-S on FT-PFN [band] | |||
| LCBench | ||||||
| 0.0001 | [+0.0176, +0.0241] † | [+0.0086, +0.0119] † | 0.00004 | [+0.0023, +0.0049] † | 0.00005 | [-0.0109, -0.0076] ‡ |
| 0.0002 | [+0.0179, +0.0251] † | [+0.0290, +0.0334] † | 0.00005 | [+0.0097, +0.0127] † | 0.00005 | [-0.0244, -0.0207] ‡ |
| 0.0005 | [+0.0174, +0.0266] † | [+0.1000, +0.1060] † | 0.00006 | [+0.0351, +0.0399] † | 0.00005 | [-0.0714, -0.0647] ‡ |
| 0.001 | [+0.0236, +0.0399] † | [+0.2310, +0.2386] † | 0.00007 | [+0.0839, +0.0931] † | 0.00005 | [-0.1549, -0.1428] ‡ |
Bayesian Optimization
Bayesian optimization uses probabilistic models and sequential experiments to find strong solutions when evaluating a black-box objective is costly.
See all 253 papers