Distributionally Robust Logistic Regression I: The Worst-Case Expected Logloss over a Wasserstein Ball Is a Tractable Convex ProgramResearch Paper
Motivation
Logistic regression is among the most widely used classification methods in statistics and machine learning. Its maximum-likelihood estimator minimizes the average logloss on the training data and is known to overfit when data are scarce; practitioners respond with ad hoc regularization, typically a norm penalty on the weight vector. Shafieezadeh-Abadeh, Mohajerin Esfahani and Kuhn (NIPS 2015, arXiv:1509.09259) replace the empirical average by a worst case over all distributions within a Wasserstein ball around the empirical distribution. The resulting model has a finite convex reformulation, contains classical and norm-regularized logistic regression as special cases, and comes with out-of-sample guarantees. It is one of the early instances of Wasserstein distributionally robust optimization in learning, building on the duality theory of Mohajerin Esfahani and Kuhn (Math. Program. 2018, arXiv:1505.05116); the regularization interpretation was later extended to general losses by Shafieezadeh-Abadeh, Kuhn and Mohajerin Esfahani (JMLR 2019, arXiv:1710.10016).
Setting
Let be the feature space with an arbitrary norm , and let be the dual norm of a weight vector . Labels are , and the feature-label space is . The logloss of at is
For a label weight , the metric of Definition 2 on is
so that changing a label costs . The Wasserstein distance between probability distributions on (Definition 1) is the infimum of over all couplings of and , and . Given training samples , the empirical distribution is , and the distributionally robust logistic regression problem (6) is
Program (7) has variables , , , objective , and constraints , for all , and .
Formalization targets
Goal: Theorem 1 (tractable reformulation)
For every , , and every norm on the feature space,
and for the infimum of (7) is attained.
Milestones
- §3.1 — the feasible set of (7) is convex.
- §2 — for the worst-case expected logloss is the empirical average logloss, so (6) reduces to classical logistic regression (2).
- Theorem 1 for fixed — equals the attained minimum of (7) over with fixed.
- Remark 2, eq. (9) — at an optimal solution ,
- Remark 1 — as the optimal value of (7) converges to .
- Theorem 2, implication — if , then .
Significance
Theorem 1 turns a minimax problem over an infinite-dimensional family of distributions into a finite convex program whose size grows linearly in ; with the , or norm it is a standard exponential-cone or conic program. Remark 1 explains norm-regularized logistic regression as a distributionally robust model: the regularizer is the dual norm of the transport cost on features, and the regularization weight is the radius of the ambiguity set. Remark 2 exposes an additional term that accounts for label noise and vanishes as label changes become prohibitively expensive. Theorem 2 makes the optimal value a certificate on the out-of-sample logloss whenever the ball contains the true distribution.
The paper's proofs are in a technical appendix and have not been machine-checked. Mathlib contains no Wasserstein distributionally robust duality. This mission produces a formal statement of the reformulation with an arbitrary norm and a label-dependent cost, together with formal versions of the paper's printed consequences of it (Remarks 1 and 2, the reduction, and the implication in Theorem 2).
Difficulty
The worst-case expectation ranges over every Borel probability distribution within transport distance of the empirical distribution, including distributions with unbounded support and distributions that move mass across labels. Exhibiting good distributions in the ball shows only that the robust value is at least the value of (7); the reverse inequality must control every distribution in the ball at once, and nothing in the definition of the ball bounds its elements' supports. The obvious simplification, restricting attention to distributions supported on finitely many points, again yields only a one-sided bound unless the supremum is shown to be approached by such distributions. The label term of the metric couples the two label classes, so results for a pure norm cost on the features do not apply directly, and the dual norm enters through an arbitrary norm rather than the Euclidean one.
Formalization scope
- The feature space is an abstract finite-dimensional real normed space
Vstanding for with an arbitrary norm; weights are continuous linear functionalsV →L[ℝ] ℝ, and is their operator norm, which is exactly the dual norm. Labels areBool, embedded as ; the label is Boolean negation. The metric of Definition 2 is written literally. - The Wasserstein distance is of type 1, valued in , with couplings ranging over all probability measures on with the two prescribed marginals. The ball consists of probability measures.
- Expectations of the positive logloss are lower Lebesgue integrals in , and the supremum over the ball is taken there; the optimal value of (7) is the infimum of its (nonnegative) objective over the feasible set, also in . A Bochner integral, which vanishes on non-integrable functions, would make the worst case trivially finite and is not used.
- The standing hypotheses are , , .
- Correction. The paper prints "min" in (7) for all . At the minimum can fail to be attained (, , , : the value is but every feasible point has positive objective). The goal states the value identity for and attainment for .
- Remark 1 is formalized as convergence of optimal values as ; a metric with is not formalized. Only convexity, not tractability, of (7) is stated. The first claim of Theorem 2 (the radius (8) and the light-tail assumption) is not formalized; the confidence of the ball event is a hypothesis of milestone 6.
- A formalization in which the ball is taken only over distributions supported on the training samples, or in which the label term of the metric is dropped, trivializes the second constraint group of (7) and is ruled out: the ball here contains every Borel probability distribution on within the prescribed distance.
- Infrastructure needed and reusable beyond this mission: type-1 optimal transport on product spaces with a label component, couplings and their marginals, and elementary properties of the logloss as a function of . Contributions of such supporting lemmas as independent theorems 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://arxiv.org/abs/1509.09259
- P. Mohajerin Esfahani, D. Kuhn, Data-driven distributionally robust optimization using the Wasserstein metric: performance guarantees and tractable reformulations, Mathematical Programming 171 (2018). https://arxiv.org/abs/1505.05116
- N. Fournier, A. Guillin, On the rate of convergence in Wasserstein distance of the empirical measure, Probability Theory and Related Fields 162 (2015). https://arxiv.org/abs/1312.2128
- S. Shafieezadeh-Abadeh, D. Kuhn, P. Mohajerin Esfahani, Regularization via Mass Transportation, Journal of Machine Learning Research 20 (2019). https://arxiv.org/abs/1710.10016