paper-with-me

홈 › Papers

Decision-Weighted Flow Matching for Contextual Stochastic Optimization

2026-06-15 · Jize Xie, Haomiao Wu, Qiang Chen, Xiu Su, Yi Chen arxiv

Conditional generative models are increasingly used as scenario generators for stochastic optimization, but standard training objectives emphasize uniform distributional fit rather than the downstream decisions induced by generated scenarios. This creates an objective mismatch: errors in statistically common regions may have little effect on decision regret, whereas errors in decision-sensitive regions can substantially change the optimal action. We propose Decision-Weighted Flow Matching (DW-FM), a regret-aligned training framework that preserves the simplicity of standard flow matching while reweighting its velocity-regression objective using decision-sensitive endpoint information. Theoretically, we connect downstream regret to pathwise velocity mismatch through a loss-induced decision discrepancy and an adjoint transport argument, yielding an ideal regret-aligned surrogate and practical endpoint-weighted objectives with regret guarantees. Empirically, we demonstrate the effectiveness of DW-FM on three CVaR-based contextual stochastic optimization benchmarks spanning synthetic portfolio, semi-real financial, and traffic-CVaR tasks, where DW-FM improves downstream regret over standard baselines.

📄 PDF Abstract BibTeX arXiv:2606.16790

Code (0)

등록된 구현이 없습니다.

Tasks

Stochastic Optimization

Similar Papers 제목 키워드 기반

Variational Flow Matching for Graph Generation

2024-06-07 · Floor Eijkelboom, Grigory Bartosh, Christian Andersson Naesseth, Max Welling 외

We present a formulation of flow matching as variational inference, which we refer to as variational flow matching (VFM). Based on this formulation we develop CatFlow, a flow matching method for categorical data. CatFlow…

Graph GenerationVariational Inference

Statistical Inference for Weighted Sample Average Approximation in Contextual Stochastic Optimization

2025-03-17 · Yanyuan Wang, Xiaowei Zhang

Contextual stochastic optimization provides a framework for decision-making under uncertainty incorporating observable contextual information through covariates. We analyze statistical inference for weighted sample avera…

Decision MakingDecision Making Under UncertaintyStochastic Optimizationvalid

Online Statistical Inference for Contextual Bandits via Stochastic Gradient Descent

2022-12-30 · Xi Chen, Zehua Lai, He Li, Yichen Zhang

With the fast development of big data, it has been easier than before to learn the optimal decision rule by updating the decision rule recursively and making online decisions. We study the online statistical inference of…

Decision MakingMulti-Armed BanditsSequential Decision Making

Maximum Likelihood Training of Score-Based Diffusion Models

2021-01-22 · NeurIPS 2021 12 · Yang song, Conor Durkan, Iain Murray, Stefano Ermon

Score-based diffusion models synthesize samples by reversing a stochastic process that diffuses data to noise, and are trained by minimizing a weighted combination of score matching losses. The log-likelihood of score-ba…

Data AugmentationImage Generation

Energy-Weighted Flow Matching for Offline Reinforcement Learning

2025-03-06 · Shiyuan Zhang, Weitong Zhang, Quanquan Gu

This paper investigates energy guidance in generative modeling, where the target distribution is defined as $q(\mathbf x) \propto p(\mathbf x)\exp(-\beta \mathcal E(\mathbf x))$, with $p(\mathbf x)$ being the data distri…

Offline RLreinforcement-learningReinforcement LearningReinforcement Learning (RL)