Distributionally Robust Logistic Regression II: Worst- and Best-Case Misclassification Risks over a Wasserstein Ball Are Linear ProgramsResearch Paper
Motivation
A logistic regression model is fitted on finitely many samples, and the quantity a practitioner cares about is the misclassification risk of the fitted classifier on new data. Its empirical counterpart, the training error, is biased downwards, and classical generalization bounds give it an additive margin that depends on a complexity measure of the model class rather than on the data at hand.
Shafieezadeh-Abadeh, Mohajerin Esfahani and Kuhn (NIPS 2015) take a distributionally robust route. They surround the empirical distribution of the training data by a ball of distributions in the Wasserstein metric and, for a given weight vector, compute the largest and the smallest misclassification probability over that ball. Their Theorem 3 shows that both extremes are optimal values of explicit linear programs. Combined with a measure-concentration result for the empirical distribution in the Wasserstein metric (Fournier and Guillin, PTRF 2015), the two values bracket the true risk with a prescribed confidence. The same Wasserstein-ball construction underlies the data-driven optimization framework of Mohajerin Esfahani and Kuhn (Math. Program. 2018).
Setting
Let be the feature space with an arbitrary norm , and let labels take the values . The feature-label space is with points . A weight vector acts on features by ; its dual norm is .
Metric (Definition 2). For a weight ,
Changing a label costs ; moving a feature costs its norm distance.
Wasserstein distance (Definition 1). For distributions on , is the infimum of over all couplings of and . The Wasserstein ball of radius is .
Data. Training samples , , define the empirical distribution .
Classifier and risk. Logistic regression models (eq. (1)). The classifier is if and otherwise, and its risk under the data-generating distribution is .
Worst- and best-case risks.
The worst case counts a nonpositive margin, the best case a strictly negative one.
The linear programs. For data , a weight vector and variables , , program (10a) minimizes subject to, for every ,
Program (10b) has the same objective and bounds, with the signs of the two margin terms exchanged.
Formalization targets
Goal: Theorem 3 (i)–(ii)
For every , , , all samples and every weight vector , both programs attain their minima and , and
The identities hold for each fixed , so they apply to any computed from the data.
Milestone: Theorem 3(i) alone
equals the minimum of (10a).
Milestones: the confidence clauses
If the training samples are i.i.d. from and the radius is such that , then for any sample-dependent
Significance
The result. Theorem 3 replaces an optimization over an infinite-dimensional set of distributions by a linear program with variables and constraints plus sign constraints. That makes the worst- and best-case misclassification probabilities computable at the scale of the training set, for any norm on the features whose dual norm can be evaluated. With the confidence clauses, the two values are data-driven upper and lower confidence bounds on the out-of-sample risk of the classifier actually deployed, including one fitted on the same data.
Formalizing it. The paper states Theorem 3 without proof in the main text; the argument is deferred to a technical appendix. No part of it is machine-checked. A formal proof needs the evaluation of a worst-case probability of a closed set over a type-1 Wasserstein ball around a discrete distribution, and the analogous best-case probability of an open set. Both are reusable in any Wasserstein-robust treatment of chance constraints or classification error.
Difficulty
The objective is neither continuous nor concave, so the duality theorems for Wasserstein balls stated for continuous or Lipschitz losses do not apply directly. Upper semicontinuity of the indicator of a closed set is what matters, and the strict inequality in has to be handled as the complement of a closed set. The transport cost couples a norm on the features with a discrete label-flip cost, so a sample can reach the misclassification region either by moving its feature to the hyperplane or by flipping its label, and the two options interact through the shared budget . Distances to the hyperplane are measured in the given norm and produce the dual norm . The degenerate weight (every point on the hyperplane) must come out correctly without any division by .
Formalization scope
The feature space is a finite-dimensional real normed space V with an arbitrary norm, standing for ; the Euclidean norm is not assumed. A weight vector is a continuous linear functional V →L[ℝ] ℝ, and is its operator norm, which is exactly the dual norm. Labels are Bool with an explicit embedding , ; the metric of Definition 2 is written literally. The Wasserstein distance is ℝ≥0∞-valued, probabilities and expectations of indicators are measure values in , and suprema and infima range exactly over the probability measures in the ball. "min" in (10a)/(10b) is formalized as attainment (IsLeast) of the objective over the feasible set. Samples are indexed by Fin N with .
The following choices differ from a literal reading of the page:
- The paper says the risk "can be expressed as" . This fails on the hyperplane , where is correct for . The mission defines from (1) and includes the true statement as a helper item.
- The choice of (8) and the measure-concentration theorem behind it (Theorem 2) are not formalized. The confidence clauses take their conclusion, , as a hypothesis, and "with probability " is read as "with probability at least ". The printed level is kept for the two-sided bound.
Swapping the strict and non-strict inequalities in and , restricting the supremum to measures supported on the sample points, or replacing the ball by a set that excludes non-discrete distributions would each change the theorem. None of these is an acceptable reformulation of the goal.
Useful infrastructure: couplings of a discrete measure with an arbitrary one, the distance from a point to a closed half-space in a general norm, and LP-duality arguments for fractional-knapsack-type programs. Proofs of the helper and confidence items, and any reusable lemma about worst-case probabilities of closed sets over Wasserstein balls, are welcome.
Selected references
- S. Shafieezadeh-Abadeh, P. Mohajerin Esfahani, D. Kuhn, Distributionally Robust Logistic Regression, Advances in Neural Information Processing Systems 28 (NIPS 2015). https://papers.nips.cc/paper/2015/hash/cc1aa436277138f61cda703991069eaf-Abstract.html
- N. Fournier, A. Guillin, On the rate of convergence in Wasserstein distance of the empirical measure, Probability Theory and Related Fields 162 (2015). https://doi.org/10.1007/s00440-014-0583-7
- P. Mohajerin Esfahani, D. Kuhn, Data-driven distributionally robust optimization using the Wasserstein metric: performance guarantees and tractable reformulations, Mathematical Programming 171 (2018). https://doi.org/10.1007/s10107-017-1172-1