A Solver-Free Training Method for Predict-then-Optimize
Beichen Wan, Mo Liu
Abstract
We propose a scalable method for training prediction (machine learning) models in the predict-then-optimize paradigm, where model outputs serve as coefficients for a subsequent linear optimization task. Directly minimizing the empirical decision regret is intractable for linear programming and combinatorial optimization since the decision mapping is piecewise constant, and the gradients are zero almost everywhere. While existing methods address this by smoothing the differentiation process, they suffer from scalability issues, since a computationally expensive solver call is required for every gradient evaluation. To address this, we propose a decision-focused learning pipeline based on a measure transformation principle, which yields a new surrogate loss that is completely optimization-solver-free during training. We establish theoretical guarantees, including Fisher consistency and excess risk bounds. Empirically, our method achieves decision quality competitive with state-of-the-art methods while reducing training time by orders of magnitude.
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 d14dfa1a-06a8-4d20-a4c0-6e7e9ce414d5Builds on12
- Differentiation of Blackbox Combinatorial SolversMarin Vlastelica Pogancic, Anselm Paulus, Vít Musil, Georg Martius et al.ICLR 2020 · 341 citations
- Smart Predict-and-Optimize for Hard Combinatorial Optimization ProblemsJayanta Mandi, Emir Demirovic, Peter J. Stuckey, Tias GunsAAAI 2020 · 184 citations
- Learning with Differentiable Pertubed OptimizersQuentin Berthet, Mathieu Blondel, Olivier Teboul, Marco Cuturi et al.NeurIPS 2020 · 181 citations
- Interior Point Solving for LP-based prediction+optimisationJayanta Mandi, Tias GunsNeurIPS 2020 · 138 citations
- Decision-Focused Learning without Decision-Making: Learning Locally Optimized Decision LossesSanket Shah, Kai Wang, Bryan Wilder, Andrew Perrault et al.NeurIPS 2022 · 79 citations
Related papers
- Solver-Free Decision-Focused Learning for Linear Optimization ProblemsSenne Berden, Ali Irfan Mahmutogullari, Dimos Tsouros, Tias GunsNeurIPS 2025 · 13 citations
- Decision-Focused Learning: Through the Lens of Learning to RankJayanta Mandi, Víctor Bucarey, Maxime Mulamba Ke Tchomba, Tias GunsICML 2022 · 73 citations
- Decision-Focused Learning via Tangent-Space Projection of Prediction ErrorJunhyeong Lee, Sangjin Jin, Yongjae LeeICML 2026
- MIPaaL: Mixed Integer Program as a LayerAaron M. Ferber, Bryan Wilder, Bistra Dilkina, Milind TambeAAAI 2020 · 169 citations
- Learning MDPs from Features: Predict-Then-Optimize for Sequential Decision Making by Reinforcement LearningKai Wang, Sanket Shah, Haipeng Chen, Andrew Perrault et al.NeurIPS 2021 · 44 citations
