Prescribe-then-Select: Adaptive Policy Selection for Contextual Stochastic Optimization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Iglesias, Caio de Prospero, Carballo, Kimberly Villalobos, Bertsimas, Dimitris
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910267781152768
author Iglesias, Caio de Prospero
Carballo, Kimberly Villalobos
Bertsimas, Dimitris
author_facet Iglesias, Caio de Prospero
Carballo, Kimberly Villalobos
Bertsimas, Dimitris
contents We address the problem of policy selection in contextual stochastic optimization (CSO), where covariates are available as contextual information and decisions must satisfy hard feasibility constraints. In many CSO settings, multiple candidate policies--arising from different modeling paradigms--exhibit heterogeneous performance across the covariate space, with no single policy uniformly dominating. We propose Prescribe-then-Select (PS), a modular framework that first constructs a library of feasible candidate policies and then learns a meta-policy to select the best policy for the observed covariates. We implement the meta-policy using ensembles of Optimal Policy Trees trained via cross-validation on the training set, making policy choice entirely data-driven. Across two benchmark CSO problems--single-stage newsvendor and two-stage shipment planning--PS consistently outperforms the best single policy in heterogeneous regimes of the covariate space and converges to the dominant policy when such heterogeneity is absent. All the code to reproduce the results can be found at https://anonymous.4open.science/r/Prescribe-then-Select-TMLR.
format Preprint
id arxiv_https___arxiv_org_abs_2509_08194
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Prescribe-then-Select: Adaptive Policy Selection for Contextual Stochastic Optimization
Iglesias, Caio de Prospero
Carballo, Kimberly Villalobos
Bertsimas, Dimitris
Machine Learning
We address the problem of policy selection in contextual stochastic optimization (CSO), where covariates are available as contextual information and decisions must satisfy hard feasibility constraints. In many CSO settings, multiple candidate policies--arising from different modeling paradigms--exhibit heterogeneous performance across the covariate space, with no single policy uniformly dominating. We propose Prescribe-then-Select (PS), a modular framework that first constructs a library of feasible candidate policies and then learns a meta-policy to select the best policy for the observed covariates. We implement the meta-policy using ensembles of Optimal Policy Trees trained via cross-validation on the training set, making policy choice entirely data-driven. Across two benchmark CSO problems--single-stage newsvendor and two-stage shipment planning--PS consistently outperforms the best single policy in heterogeneous regimes of the covariate space and converges to the dominant policy when such heterogeneity is absent. All the code to reproduce the results can be found at https://anonymous.4open.science/r/Prescribe-then-Select-TMLR.
title Prescribe-then-Select: Adaptive Policy Selection for Contextual Stochastic Optimization
topic Machine Learning
url https://arxiv.org/abs/2509.08194