Saved in:
Bibliographic Details
Main Authors: Chen, Shuyi, Zhou, Wenbin, Zhu, Shixiang
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2605.08506
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918500583342080
author Chen, Shuyi
Zhou, Wenbin
Zhu, Shixiang
author_facet Chen, Shuyi
Zhou, Wenbin
Zhu, Shixiang
contents Robust optimization (RO) provides a principled framework for decision-making under uncertainty, but its performance critically depends on the choice of the uncertainty set. While large sets ensure reliability, they often lead to overly conservative decisions, whereas small sets risk excluding the true outcome. Recent data-driven approaches, particularly conformal prediction, offer finite-sample validity guarantees but remain largely task-agnostic, ignoring the downstream decision structure. In this paper, we propose a decision-aware conformal framework that learns uncertainty sets tailored to robust optimization objectives. Our approach parameterizes a flexible family of polyhedral sets via data-driven hyperplanes and learns their geometry by directly minimizing the induced robust loss, while preserving statistical validity through conformal calibration. To correct for data-dependent selection, we incorporate a re-calibration step on an independent dataset to restore coverage. The resulting sets capture directional and anisotropic uncertainty aligned with the decision objective while remaining computationally tractable. We provide finite-sample coverage guarantees and bounds on the sub-optimality gap to an oracle decision. This work bridges the gap between statistical validity and decision optimality, providing a principled framework for data-driven robust optimization.
format Preprint
id arxiv_https___arxiv_org_abs_2605_08506
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning Polyhedral Conformal Sets for Robust Optimization
Chen, Shuyi
Zhou, Wenbin
Zhu, Shixiang
Machine Learning
Robust optimization (RO) provides a principled framework for decision-making under uncertainty, but its performance critically depends on the choice of the uncertainty set. While large sets ensure reliability, they often lead to overly conservative decisions, whereas small sets risk excluding the true outcome. Recent data-driven approaches, particularly conformal prediction, offer finite-sample validity guarantees but remain largely task-agnostic, ignoring the downstream decision structure. In this paper, we propose a decision-aware conformal framework that learns uncertainty sets tailored to robust optimization objectives. Our approach parameterizes a flexible family of polyhedral sets via data-driven hyperplanes and learns their geometry by directly minimizing the induced robust loss, while preserving statistical validity through conformal calibration. To correct for data-dependent selection, we incorporate a re-calibration step on an independent dataset to restore coverage. The resulting sets capture directional and anisotropic uncertainty aligned with the decision objective while remaining computationally tractable. We provide finite-sample coverage guarantees and bounds on the sub-optimality gap to an oracle decision. This work bridges the gap between statistical validity and decision optimality, providing a principled framework for data-driven robust optimization.
title Learning Polyhedral Conformal Sets for Robust Optimization
topic Machine Learning
url https://arxiv.org/abs/2605.08506