End-to-End Learning for Stochastic Optimization: A Bayesian Perspective
Yves Rychener, Daniel Kuhn, Tobias Sutter
Abstract
We develop a principled approach to end-to-end learning in stochastic optimization. First, we show that the standard end-to-end learning algorithm admits a Bayesian interpretation and trains a posterior Bayes action map. Building on the insights of this analysis, we then propose new end-to-end learning algorithms for training decision maps that output solutions of empirical risk minimization and distributionally robust optimization problems, two dominant modeling paradigms in optimization under uncertainty. Numerical results for a synthetic newsvendor problem illustrate the key differences between alternative training schemes. We also investigate an economic dispatch problem based on real data to showcase the impact of the neural network architecture of the decision maps on their test performance.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b5d4d390-c0d7-4f62-88b5-63d4e9a328b9Cited by top-tier papers3
- Contextual Stochastic Bilevel OptimizationYifan Hu, Jie Wang, Yao Xie, Andreas Krause et al.NeurIPS 2023 · 22 citations
- Multi-Stage Predict+Optimize for (Mixed Integer) Linear ProgramsXinyi Hu, Jasper C. H. Lee, Jimmy H. M. Lee, Peter J. StuckeyNeurIPS 2024 · 9 citations
- Smart Surrogate Losses for Contextual Stochastic Linear Optimization with Robust ConstraintsHyungki Im, Wyame Benslimane, Paul GrigasNeurIPS 2025 · 3 citations
Builds on1
Related papers
- Decision Making under the Exponential Family: Distributionally Robust Optimisation with Bayesian Ambiguity SetsCharita Dellaporta, Patrick O'Hara, Theodoros DamoulasICML 2025
- Bayesian Nonparametrics Meets Data-Driven Distributionally Robust OptimizationNicola Bariletto, Nhat HoNeurIPS 2024 · 4 citations
- Distributionally Robust Performative OptimizationZhuangzhuang Jia, Yijie Wang, Roy Dong, Grani A. HanasusantoNeurIPS 2025 · 3 citations
- Loss-aware distributionally robust optimization via trainable optimal transport ambiguity setsJonas Ohnemus, Marta Fochesato, Riccardo Zuliani, John LygerosICML 2026
- E2E-AT: A Unified Framework for Tackling Uncertainty in Task-Aware End-to-End LearningWangkun Xu, Jianhong Wang, Fei TengAAAI 2024 · 5 citations
