Regime-Conditioned Evaluation in Multi-Context Bayesian Optimization
TLDR
This paper introduces the Portable Regime Score (PRS) for multi-context Bayesian Optimization, predicting optimal acquisition strategies based on experimental conditions.
Key contributions
- An audit reveals 98% of transfer-BO papers ignore budget ratio (B/|A|), leading to unstable acquisition rankings.
- Introduces Portable Regime Score (PRS) = (B/|A|)(1-rho) to predict optimal acquisition based on budget and prior quality.
- PRS predicts winners in published reversal cases; a PRS-guided planner outperforms matched oracles by 18%.
- Proposes a practical protocol: report B/|A|, rho, K, and metric for robust acquisition advantage claims.
Why it matters
Current BO evaluations yield unstable, unconditional rankings, hindering practitioners. This paper introduces a framework and Portable Regime Score (PRS) to condition evaluations on experimental parameters, providing reliable, context-aware recommendations.
Original Abstract
Published transfer-BO comparisons often estimate an average treatment effect of acquisition choice over hidden regime variables, while practitioners need the conditional effect for their specific prior quality, budget ratio, and metric. An audit of 40 transfer-BO papers from NeurIPS, ICML, ICLR, AISTATS, UAI, TMLR, JMLR, and AutoML-Conf (2022-2025) finds that 98% never vary B/|A| as a controlled axis. On the same GDSC2 benchmark, changing only the budget reverses the ranking: at B=50, Greedy outperforms UCB by 0.050 Hit@1, while at B=100, UCB outperforms Greedy by 0.035. We capture this transition with the Portable Regime Score PRS=(B/|A|)(1-rho), where rho is the prior rank correlation and can be estimated from pilot contexts before the main comparison. Across 79 conditions spanning chemistry, drug-response biology, and HPO, a hierarchical model gives beta=0.50 (p=1.1e-9), and 19% of conditions fall in an equivalence zone where |advantage|<0.01 Hit@1. In five published reversal cases, PRS predicts the winner from pre-comparison observables. A No-Free-Leaderboard proposition explains why unconditional rankings are unstable: when CATE changes sign across regimes, the reported ATE becomes a function of benchmark mixture. RegimePlanner, which estimates rho online and switches acquisition accordingly, wins all 16 HPO-B search spaces at B=100 and exceeds the matched {Greedy,UCB} per-context oracle on GDSC2 by 18%. Pre-registered predictions achieve 27/40=67.5% overall accuracy and above 90% within EMA prior families. The practical protocol is simple: report B/|A|, rho, K, and metric alongside any claimed acquisition advantage.
📬 Weekly AI Paper Digest
Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.