arXiv cs.AI· Tianjue Lin, Jianan Zhou, Jieyi Bi, Yaoxin Wu, Wen Song, Zhiguang Cao, Jie Zhang·· 4 小时前AI 评分24
NeurPRISE:面向两阶段鲁棒优化的学习式场景缩减方法
Learning Scenario Reduction for Two-Stage Robust Optimization with Discrete Uncertainty
AI 导读
论文提出 NeurPRISE,一种基于 GNN-Transformer 骨干的神经代理模型,用于两阶段鲁棒优化(2RO)离散不确定性下的场景缩减,通过模仿学习从 PRISE 启发式蒸馏边际增益信息。
正文 · 原文
Abstract:Two-Stage Robust Optimization (2RO) with discrete uncertainty is challenging, often rendering exact solutions prohibitive. Scenario reduction alleviates this issue by selecting a small, representative subset of scenarios to enable tractable computation. However, existing methods are largely problem-agnostic, operating solely on the uncertainty set without consulting the feasible region or recourse structure. In this paper, we introduce PRISE, a problem-driven sequential lookahead heuristic that constructs reduced scenario sets by evaluating the marginal impact of each scenario. While PRISE yields high-quality scenario subsets, each selection step requires solving multiple subproblems, making it computationally expensive at scale. To address this, we propose NeurPRISE, a neural surrogate model built on a GNN-Transformer backbone that encodes the per-scenario structure via graph convolution and captures cross-scenario interactions through attention. NeurPRISE is trained via imitation learning with a gain-aware ranking objective, which distills marginal gain information from PRISE into a learned scoring function for scenario ranking and selection. Extensive results on three 2RO problems show that NeurPRISE consistently achieves competitive regret relative to comprehensive methods, maintains strong calability with varying numbers of scenarios, and delivers 17-200x speedup over PRISE. NeurPRISE also exhibits strong zero-shot generalization, effectively handling instances with larger problem scales (up to 5x), more scenarios (up to 4x), and distribution shifts.
| Comments: | Accepted at NeurIPS 2026 |
| Subjects: | Artificial Intelligence (cs.AI); Machine Learning (cs.LG) |
| Cite as: | arXiv:2605.14494 [cs.AI] |
| (or arXiv:2605.14494v2 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2605.14494 arXiv-issued DOI via DataCite |
Submission history
From: Tianjue Lin [view email]
[v1]
Thu, 14 May 2026 07:34:13 UTC (1,358 KB)
[v2]
Mon, 26 Oct 2026 12:26:33 UTC (1,357 KB)
来源:arXiv cs.AI · arxiv.org