Search papers, labs, and topics across Lattice.
This paper introduces Decision-Weighted Flow Matching (DW-FM), a novel training framework for conditional generative models that aligns scenario generation with decision-making outcomes in stochastic optimization. By reweighting the velocity-regression objective based on decision-sensitive endpoint information, DW-FM addresses the critical mismatch between uniform distributional fit and the actual impact of generated scenarios on decision regret. Empirical results across various CVaR-based benchmarks show that DW-FM significantly reduces downstream regret compared to standard methods, highlighting its practical advantages in real-world applications.
DW-FM reveals that aligning generative model training with decision-sensitive outcomes can drastically reduce decision regret in stochastic optimization tasks.
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.
