Generalization Bounds in the Predict-then-Optimize Framework II: Margin-Based Generalization Bound for the SPO Loss under the Strength PropertyResearch Paper
Motivation
In the predict-then-optimize paradigm a model first predicts the cost vector of a linear optimization problem from contextual features, and the prediction is then fed to an optimization solver that returns a decision. Examples include routing with predicted travel times and portfolio choice with predicted returns. The quality of a prediction is judged by the decision it produces. The Smart Predict-then-Optimize (SPO) loss of Elmachtoub and Grigas (Management Science 2022) measures exactly that: the excess cost of acting on the prediction instead of on the true cost vector.
El Balghiti, Elmachtoub, Grigas and Tewari (arXiv:1905.11488v3) ask when a model with small empirical SPO loss also has small expected SPO loss. The SPO loss is non-convex and discontinuous, so standard Lipschitz-contraction arguments do not apply to it directly. Their Section 4 introduces a margin version of the SPO loss, in the spirit of the margin theory of Koltchinskii and Panchenko (Ann. Statist. 2002) for classification. They show that it is Lipschitz under a geometric condition on the feasible region, and derive a generalization bound in terms of the multivariate Rademacher complexity of the hypothesis class. This mission formalizes that bound.
Setting
Decisions live in with a norm ; cost vectors are linear functionals with the dual norm . The feasible region is nonempty, compact and convex, and throughout Section 4 it is not a singleton. An optimization oracle maps each cost vector to some minimizer . The SPO loss of a prediction against the realized cost is
and the linear optimization gap is , with and for the set of possible true costs.
A cost vector is degenerate if has more than one optimal solution; is the set of degenerate costs. The distance to degeneracy is . The region has the strength property with parameter if
For the -margin SPO loss equals when and otherwise. It dominates the SPO loss.
Data are drawn from a distribution on features and costs in , and is a class of prediction functions . The SPO risk is and the empirical margin risk is . The multivariate empirical Rademacher complexity is with i.i.d. Rademacher vectors , and is its expectation over the sample.
Formalization targets
Goal: Theorem 4, second display (pp. 19–20)
In the set-up, under the strength property with and for fixed , for every , with probability at least over an i.i.d. sample of size , for all :
Milestones
- Theorem 3(a): .
- Theorem 3(b): the same Lipschitz-like bound for , with an extra factor .
- Theorem 3(c): is -Lipschitz for the dual norm.
- Eq. (7) with (Maurer's vector contraction inequality): for -Lipschitz on Euclidean ,
- Theorem 4, first display: for any fixed sample with costs in ,
Theorem 3 is stated for a general norm, as in the paper. Eq. (7), Theorem 4 and the goal are Euclidean. The paper's Theorem 5 (p. 20), a version of the goal uniform over , is not part of this mission.
Significance
The bound replaces the loss-class complexity of the SPO loss, which is controlled only through combinatorial dimensions (Natarajan dimension in the polyhedral case, Section 3 of the paper), by the multivariate Rademacher complexity of itself. For norm-bounded linear hypothesis classes this complexity has mild, even logarithmic, dependence on the dimensions and (Section 4.4). The result applies to every feasible region with the strength property. By Section 5 of the paper these include strongly convex sets and polytopes, where can also be computed. When most predictions stay far from degeneracy, and the bound is much sharper than the combinatorial one. It is also a strict generalization of margin bounds for binary classification (Example 7).
The theorem is proved in the paper, which imports two external tools without proof: the Rademacher generalization bound of Bartlett and Mendelson, applied to the margin loss, and Maurer's inequality. To our knowledge none of these results has a machine-checked proof. The mission produces a checked proof of the margin bound and a Lean statement of Maurer's inequality. It also formalizes the strength property and the Lipschitz estimates of Theorem 3, which the companion missions on strongly convex sets and polytopes rely on.
Difficulty
The SPO loss is discontinuous in at degenerate predictions. The standard route, scalar Ledoux–Talagrand contraction applied to the loss class, therefore fails at the first step. It would fail even for a Lipschitz loss, because it relates the loss class only to a scalar class, and is vector valued. Lipschitz continuity of the margin loss needs the oracle to be stable away from . Convexity and compactness of alone do not give that: for an ball with the strength property fails for every (p. 14). The vector contraction inequality of Maurer (2016) is a nontrivial probabilistic inequality, and its constant must not depend on the dimension . The final concentration step is McDiarmid's inequality for a supremum over a possibly uncountable class, which in a formal proof needs measurability of that supremum.
Formalization scope
The decision space is a finite-dimensional real normed space E. Cost vectors and predictions are continuous linear functionals, StrongDual ℝ E, whose operator norm is the paper's dual norm. In the statements E = EuclideanSpace ℝ (Fin d), where the dual norm is Euclidean. Every statement carries the standing assumptions: nonempty, compact, convex and not a singleton, an arbitrary oracle (no tie-breaking rule), and , . The Lipschitz-like bounds of Theorem 3(a)–(b) are stated multiplied out, because the paper reads as . Expectations over signs are finite averages over sign patterns. and are suprema over a nonempty bounded containing the cost almost surely. "With probability at least " is the statement that the outer -measure of the failure event is at most .
Added hypotheses, all disclosed in the statements: the multivariate Rademacher sums are bounded above (almost surely in the goal) and is integrable, since otherwise Lean's junk value would replace an infinite complexity and make the bound false rather than vacuous. Hypotheses and measurable, and the uniform deviation and margin Rademacher suprema a.e.-measurable, are also added; the paper is silent on measurability. A singleton would make empty and the strength property hold for free; this is excluded explicitly, so the strength property is not vacuous.
A complete development needs the Bartlett–Mendelson symmetrization bound for bounded losses, McDiarmid's inequality, Maurer's inequality, and the Lipschitz and distance-to-degeneracy facts of Section 4.1. Maurer's inequality and the multivariate Rademacher complexity are reusable across vector-valued learning theory. Proofs of any milestone, and of Maurer's inequality in particular, are welcome.
Selected references
- O. El Balghiti, A. N. Elmachtoub, P. Grigas, A. Tewari, Generalization Bounds in the Predict-then-Optimize Framework, Mathematics of Operations Research, 2023; preprint arXiv:1905.11488v3, 2022. https://arxiv.org/abs/1905.11488
- A. N. Elmachtoub, P. Grigas, Smart "Predict, then Optimize", Management Science 68(1), 2022. https://doi.org/10.1287/mnsc.2020.3922
- A. Maurer, A Vector-Contraction Inequality for Rademacher Complexities, Algorithmic Learning Theory (ALT), 2016. https://arxiv.org/abs/1605.00251
- P. L. Bartlett, S. Mendelson, Rademacher and Gaussian Complexities: Risk Bounds and Structural Results, Journal of Machine Learning Research 3, 2002. https://www.jmlr.org/papers/v3/bartlett02a.html
- V. Koltchinskii, D. Panchenko, Empirical Margin Distributions and Bounding the Generalization Error of Combined Classifiers, Annals of Statistics 30(1), 2002. https://doi.org/10.1214/aos/1015362183