Strong Mixed-Integer Programming Formulations for Trained Neural Networks 2: Under Strict Activity Every Inequality of the Exponential Family (6b) Is Facet-DefiningResearch Paper
Motivation
Trained neural networks are increasingly embedded inside optimization models: to verify that a classifier is robust to small input perturbations, to optimize over a learned surrogate of an expensive system, or to choose decisions whose outcome is predicted by a network. When the network uses rectified linear units (ReLU), each neuron is piecewise linear, and the whole network can be written exactly as a mixed-integer program (MIP) with one binary variable per neuron. How fast a branch-and-bound solver closes such a model depends on how tight the linear-programming relaxation of each neuron's formulation is.
Anderson, Huchette, Tjandraatmadja and Vielma (arXiv:1811.08359v2, the IPCO 2019 extended abstract) gave a formulation (6) of a single ReLU neuron over a box that uses only the original variables and one binary variable, and is ideal: its LP relaxation has integral extreme points (Proposition 1, p. 6). The price is an exponential family of inequalities (6b), one for every subset of the support of . The present mission formalizes their Proposition 2: each of these inequalities is facet-defining, so no member of the family can be dropped without weakening the relaxation. A longer journal version of the work, with different numbering, appeared later (arXiv:1811.01988); this mission follows the extended abstract.
Setting
Fix , a weight vector , a bias , and bounds with for every (§1.3, p. 4). Write and . The sign-adjusted bounds are
so that and are the maximum and minimum of over . The support is . Strict activity means : the neuron is neither always off nor always on over the box. The paper assumes it throughout (§1.3).
Formulation (6) (p. 6) consists of the points with
In Lean, form6 w b L U is this set, and relax6 w b L U is its LP relaxation ( in place of ). The right-hand side of (6b) for the subset is rhs6b w b L U I x z.
An inequality is facet-defining for a set when it holds on , its face is nonempty, and , the dimension of a set being that of its affine hull. The paper uses this standard notion without defining it.
Formalization targets
Goal: Proposition 2 (p. 6)
Under and strict activity, for every , the inequality (6b) for is facet-defining for
The page states the result as "Each inequality in (6b) is facet-defining", and adds right after the proof line: "We require the assumption of strict activity above, as introduced in Section 1.3."
Milestones (App. A.2, p. 15)
- For some , the points , , for , and for are feasible with respect to (6) and satisfy (6b) for at equality; here is the sign of (with when ).
- For every these points are affinely independent.
Significance
The result. Proposition 1 shows that (6) is ideal; Proposition 2 shows it cannot be made smaller: removing any single inequality (6b) produces a strictly weaker relaxation. Since the family has members, this is what justifies the paper's practical recommendation to start from the big- formulation and separate inequalities of (6b) on demand (Proposition 3, p. 7) rather than to search for a smaller ideal description in the same variables. The facet structure also gives the geometric picture the paper describes after Proposition 2: each facet is the convex combination of an -dimensional face at and an -dimensional face at .
Formalizing it. The result is proved in the paper, in a half-page appendix; it has not been machine-checked. The formalization adds two things. First, the proof is written for "without loss of generality by appropriately interchanging and "; the Lean statements are for every sign pattern, including zero weights. Second, the appendix exhibits affinely independent points on the face, which bounds the face dimension from below; the statement that the face has dimension exactly one less than the polyhedron also needs the polyhedron to be full-dimensional and the face to lie in a proper hyperplane, steps the extended abstract leaves implicit and a complete proof must supply.
Difficulty
The arithmetic in each step is elementary. The work lies in the bookkeeping: choosing a single that keeps every perturbed point inside the box and on the correct side of (this is exactly where strict activity enters), checking the perturbed points against all inequalities of (6b) and not only the one for , and turning a row-reduction argument on an matrix into a statement about AffineIndependent and finrank of a vectorSpan in Lean. The natural shortcut, proving only that affinely independent tight points exist, is not Proposition 2: it says nothing about the dimension of the polyhedron itself.
Formalization scope
Inputs are Fin η → ℝ (indices for the paper's ); a point is p : (Fin η → ℝ) × ℝ × ℝ with p.1 = x, p.2.1 = y, p.2.2 = z. are given by their closed forms and . In (6b), "" ranges over all indices outside , zero weights included; ranges over subsets of , as on the page.
Every goal and milestone keeps the standing assumptions of §1.3, for all and strict activity, except the affine-independence milestone, which holds without them and is stated without them (a stronger statement). No other hypothesis is added. Strict activity excludes , so no nonemptiness assumption on the index set is needed.
IsFacetDefining P g is the standard notion: validity, a nonempty face, and with dimensions the finrank of the vectorSpan. The equation is written with on the left so that no natural-number subtraction can make the empty set or a point a facet. The polyhedron is the convex hull of the points of (6), not the set form6 itself (which is not convex, since ); by Proposition 1 it equals the LP relaxation relax6, but the statement does not depend on that.
The shared objects of this paper (the ReLU graph, , , , support, strict activity, formulation (6), ideality) come from the shared definitions module ReluMIP.Ideal.Setting, common to the companion mission on Proposition 1; this mission's own definitions module adds form6, IsFacetDefining, the sign inward, and the indexed family facetPts of the points. The facet notion and the affine-independence argument are reusable for other facet proofs of polyhedra in product spaces. Proofs of the milestones, of the goal, and of the full-dimensionality step are all welcome.
Selected references
- R. Anderson, J. Huchette, C. Tjandraatmadja, J. P. Vielma, Strong mixed-integer programming formulations for trained neural networks, IPCO 2019, LNCS 11480, pp. 27–42; preprint arXiv:1811.08359v2, 2019. https://arxiv.org/abs/1811.08359v2
- R. Anderson, J. Huchette, W. Ma, C. Tjandraatmadja, J. P. Vielma, Strong mixed-integer programming formulations for trained neural networks, Mathematical Programming 183 (2020), 3–39. https://doi.org/10.1007/s10107-020-01474-5
- G. L. Nemhauser, L. A. Wolsey, Integer and Combinatorial Optimization, Wiley, 1988. https://doi.org/10.1002/9781118627372
- M. Conforti, G. Cornuéjols, G. Zambelli, Integer Programming, Springer, 2014. https://doi.org/10.1007/978-3-319-11008-0
- J. P. Vielma, Mixed integer linear programming formulation techniques, SIAM Review 57 (2015), 3–57. https://doi.org/10.1137/130915303