Search papers, labs, and topics across Lattice.
This paper introduces GoBOED, a goal-driven Bayesian optimal experimental design framework that directly optimizes experimental designs for decision-making objectives by combining an amortized variational posterior surrogate with a differentiable convex decision layer. The authors theoretically prove and empirically demonstrate that GoBOED gradients are insensitive to parameter directions irrelevant to the decision objective, leading to better decision quality across a wider range of experimental designs compared to information-gain maximization. Experiments across source localization, epidemic management, and pharmacokinetic control show GoBOED identifies designs that better align with downstream decision objectives.
Stop wasting experiments: optimizing directly for your decision objective yields better results and more robust designs than just chasing information gain.
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.
