arrow
Return

Optimal Model Selection for Conformalized Robust Optimization

delete2026-09-19
delete0
PRE
AI
Y
Yajie Bao
Y
Yang Hu
H
Haojie Ren *
P
Peng Zhao
C
Changliang Zou *
DOI:10.1080/01621459.2026.2736788delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In decision-making under uncertainty, Contextual Robust Optimization (CRO) provides reliability by minimizing the worst-case decision loss over a prediction set. While recent advances use conformal prediction to construct prediction sets for machine learning models, the downstream decisions critically depend on model selection. This paper introduces novel model selection frameworks for CRO that unify robustness control with decision risk minimization. We first propose Conformalized Robust Optimization with Model Selection (CROMS), a framework that selects the model to approximately minimize the averaged decision risk in CRO solutions. Given the target robustness level 1 − 𝛼 , we present a computationally efficient algorithm called E-CROMS, which achieves asymptotic robustness control and decision optimality. To correct the control bias in finite samples, we further develop two algorithms: F-CROMS, which ensures a 1 − 𝛼 robustness but requires searching the label space; and J-CROMS, which offers lower computational cost while achieving a robustness. Furthermore, we extend the CROMS framework to the individualized setting, where model selection is performed by minimizing the conditional decision risk given the covariates of the test data. This framework advances conformal prediction methodology by enabling covariate-aware model selection. Numerical results demonstrate significant improvements in decision efficiency across diverse synthetic and real-world applications, outperforming baseline approaches.
Keywords:
Conformal prediction
Contextual robust optimization
Empirical risk minimization
Individualized model selection
Uncertainty set

Journal

J
Journal of the American Statistical Association
IF:
3
Papers:
5.2K
Citations:
4.8W

Organization

N
Nankai University
Scholars:
930
Papers: 300
Citations: 0
J
Jiangsu Normal University
Scholars:
176
Papers: 55
Citations: 0
S
Shanghai Jiao Tong University
Scholars:
4.5K
Papers: 1.3K
Citations: 0
F
fudan university
Scholars:
2.5K
Papers: 674
Citations: 0
researcher View more organizations
Cited Papers

Cited Papers

No cited papers available