cs.LGSep 14, 2026

Optimizing for the decision not the prediction: an exploration of Smooth Net Benefit as a training objective

Authors: Koen M. F. GorgelsLasai BarreñadaMaarten van SmedenBen Van CalsterEwout W. SteyerbergWouter A. C. van Amsterdam

Abstract

Objective Prediction models are commonly trained using objectives such as Bernoulli negative log-likelihood (NLL), although downstream clinical decisions may depend on specific risk thresholds. We introduce Smooth Net Benefit (σ\sigmaNB), a differentiable approximation of Net Benefit designed to align model training with threshold-specific clinical utility. Materials and Methods We evaluated σ\sigmaNB as a training objective for logistic regression, generalized additive models (GAMs), and XGBoost with three Hessian implementations. Experiments used the Framingham cardiovascular risk dataset and 44 TabZilla datasets comprising 72 dataset-threshold combinations. Results σ\sigmaNB training did not consistently improve Net Benefit in Framingham. Across the TabZilla benchmark, mean standardized Net Benefit for logistic regression increased from 0.5669 with NLL to 0.5765 with σ\sigmaNB (mean difference 0.0096, 95% CI -0.0001 to 0.0193). For GAMs, mean standardized Net Benefit decreased from 0.5921 to 0.5625 (mean difference -0.0296, 95% CI -0.0721 to 0.0129). For XGBoost, NLL achieved 0.6745 compared with 0.6723--0.6735 across σ\sigmaNB implementations. In logistic regression, σ\sigmaNB gains were positively associated with the performance advantage of XGBoost over NLL-trained logistic regression. Discussion The effect of σ\sigmaNB was context dependent, with modest gains concentrated in logistic regression and little benefit for more flexible model classes. This suggests that decision-focused optimization may be most useful when limited model flexibility leaves greater scope for improvement. Conclusion Our results do not support σ\sigmaNB as a general replacement for NLL training, but support further investigation of decision-focused objectives in settings where conventional likelihood-based training may not adequately capture decision-relevant structure.

Explore similar work

CardsList