Necessary Optimality Conditions for Integrated Learning and Optimization Problem in Contextual Optimization
Yuan Tao, Huifu Xu
TL;DR
This work derives first-order necessary optimality conditions for integrated learning and optimization (ILO), a two-stage stochastic bilevel framework that jointly learns a predictive distribution and optimizes downstream decisions. It develops convex-case conditions using Mordukhovich coderivatives for a VI-based reformulation and nonconvex-case conditions via a value-function approach under stochastic partial calmness, with measurability and selection results to support sample-based analysis. The theory is instantiated in practical SPO settings, including portfolio optimization and newsvendor problems, and demonstrates how explicit coderivative expressions can be obtained for polyhedral and orthant feasible sets. The results enable gradient-based algorithms for ILO and provide a principled way to incorporate predictive models (linear, Gaussian mixtures, kernel regression) into decision problems with uncertainty.
Abstract
Integrated learning and optimization (ILO) is a framework in contextual optimization which aims to train a predictive model for the probability distribution of the underlying problem data uncertainty, with the goal of enhancing the quality of downstream decisions. This framework represents a new class of stochastic bilevel programs, which are extensively utilized in the literature of operations research and management science, yet remain underexplored from the perspective of optimization theory. In this paper, we fill the gap. Specifically, we derive the first-order necessary optimality conditions in terms of Mordukhovich limiting subdifferentials. To this end, we formulate the bilevel program as a two-stage stochastic program with variational inequality constraints when the lower-level decision-making problem is convex, and establish an optimality condition via sensitivity analysis of the second-stage value function. In the case where the lower level optimization problem is nonconvex, we adopt the value function approach in the literature of bilevel programs and derive the first-order necessary conditions under stochastic partial calmness conditions. The derived optimality conditions are applied to several existing ILO problems in the literature. These conditions may be used for the design of gradient-based algorithms for solving ILO problems.
