Goal-driven Bayesian Optimal Experimental Design for Robust Decision-Making Under Model Uncertainty
面向模型不确定性下鲁棒决策的目标驱动贝叶斯最优实验设计
Jinwoo Go, Xiaoning Qian, Byung-Jun Yoon
AI总结 提出GoBOED框架,通过结合变分后验代理与可微凸决策层,直接优化实验设计以提升下游决策质量,并理论证明其对决策无关参数方向不敏感。
详情
贝叶斯最优实验设计(BOED)选择实验以最大化关于模型参数的信息增益。然而,在决策关键场景中,减少参数不确定性并不一定能改善下游决策,因为只有与目标相关的特定参数方向才真正重要。我们提出了GoBOED,一个目标驱动的BOED框架,它直接针对指定的决策目标优化实验设计。GoBOED结合了摊销变分后验代理与可微凸决策层,实现了完全以决策为中心的基于梯度的设计优化。我们从理论上证明,GoBOED梯度对决策目标无关的参数方向不敏感,这为为什么目标驱动设计在更广泛的实验设计集合上实现与信息增益最大化等效的决策质量提供了形式化依据。在源定位、流行病管理和药代动力学控制等实证任务中,GoBOED识别出与下游决策目标更一致的设计,并揭示了接近最优的设计窗口比目标无关的BOED方法预测的要宽得多。
Bayesian optimal experimental design (BOED) selects experiments to maximize information gain about model parameters. However, in decision-critical settings, reducing parameter uncertainty does not necessarily improve downstream decisions, as only specific parameter directions relevant to the objective truly matter. We propose GoBOED, a goal-driven BOED framework that directly optimizes experimental designs for a specified decision-making objective. GoBOED combines an amortized variational posterior surrogate with a differentiable convex decision layer, enabling gradient-based design optimization that is fully decision-focused. We theoretically show that GoBOED gradients are insensitive to parameter directions irrelevant to the decision objective, providing a formal justification for why goal-driven design achieves equivalent decision quality over a wider set of experimental designs than information-gain maximization. Empirically, across source localization, epidemic management, and pharmacokinetic control, GoBOED identifies designs that better align with downstream decision objectives and reveals that near-optimal design windows are substantially wider than those predicted by goal-agnostic BOED approaches.