Bayesian optimization is a coherent, ubiquitous approach to decision-making under uncertainty, with applications including multi-arm bandits, active learning, and black-box optimization. Bayesian optimization selects decisions (i.e. objective function queries) with maximal expected utility with respect to the posterior distribution of a Bayesian model, which quantifies reducible, epistemic uncertainty about query outcomes. In practice, subjectively implausible outcomes can occur regularly for two reasons: 1) model misspecification and 2) covariate shift. Conformal prediction is an uncertainty quantification method with coverage guarantees even for misspecified models and a simple mechanism to correct for covariate shift. We propose conformal Bayesian optimization, which directs queries towards regions of search space where the model predictions have guaranteed validity, and investigate its behavior on a suite of black-box optimization tasks and tabular ranking tasks. In many cases we find that query coverage can be significantly improved without harming sample-efficiency.
翻译:贝叶斯优化是一种在不确性条件下进行决策的连贯且广泛采用的方法,其应用包括多臂老虎机、主动学习和黑箱优化。贝叶斯优化通过最大化贝叶斯模型后验分布的期望效用(该分布量化了关于查询结果的可约简认知不确定性)来选择决策(即目标函数查询)。在实践中,主观上不合理的查询结果可能因两种原因频繁出现:1)模型误设定;2)协变量偏移。保形预测是一种不确定性量化方法,即使对于误设定模型也能提供覆盖保证,并具有修正协变量偏移的简单机制。我们提出保形贝叶斯优化,该方法将查询引导至模型预测具有保证有效性的搜索空间区域,并在一系列黑箱优化任务和表格排序任务中研究其行为。在许多案例中,我们发现无需牺牲样本效率即可显著提升查询覆盖度。