Search papers, labs, and topics across Lattice.
This paper introduces decision-aware training for sample-based generative models, addressing the limitation of traditional training objectives that ignore the decision maker's cost structure. By augmenting the energy score with a differentiable decision loss that penalizes costly forecast errors, the authors provide a theoretically grounded approach that aligns model training with downstream decision-making. Experiments on synthetic and real-world tasks demonstrate significant improvements in cost-sensitive areas while maintaining the integrity of probabilistic forecasts.
Training generative models with a decision-aware approach can significantly enhance performance in high-stakes environments where forecast errors carry different costs.
Sample-based generative models are increasingly used for probabilistic forecasting in high-stakes decision settings, yet their training objectives are blind to the decision maker's cost structure. These models are commonly trained with strictly proper scoring rules, such as the energy score, which allocate their training signal in proportion to data density, with no awareness of where forecast errors are most costly for downstream decisions. We therefore propose decision-aware training for sample-based generative models, augmenting the energy score objective with a differentiable decision loss that directly penalises the cost incurred by acting on the model's forecast. This combined loss is theoretically grounded, as the decision loss is itself a proper scoring rule. We validate our method on one synthetic and two real-world tasks, showing targeted improvements in cost-sensitive regions while retaining full probabilistic forecasts.
