Distributionally Robust Logistic Regression
Not on the map. Why?
Abstract (arXiv)
This paper proposes a distributionally robust approach to logistic regression. We use the Wasserstein distance to construct a ball in the space of probability distributions centered at the uniform distribution on the training samples. If the radius of this ball is chosen judiciously, we can guarantee that it contains the unknown data-generating distribution with high confidence. We then formulate a distributionally robust logistic regression model that minimizes a worst-case expected logloss function, where the worst case is taken over all distributions in the Wasserstein ball. We prove that this optimization problem admits a tractable reformulation and encapsulates the classical as well as the popular regularized logistic regression problems as special cases. We further propose a distributionally robust approach based on Wasserstein balls to compute upper and lower confidence bounds on the misclassification probability of the resulting classifier. These bounds are given by the optimal values of two highly tractable linear programs. We validate our theoretical out-of-sample guarantees through simulated and empirical experiments.
Key facts. Cited by 124 papers in the map (data through 7 Oct 2026). First posted on arXiv 30 Sep 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 30; 2022: 21; 2023: 28; 2024: 23; 2025: 15; 2026: 7.
124 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2021Distributional Robustness Loss for Long-tail Learningcited by 84
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 52
- 2021Twice regularized MDPs and the equivalence between robustness and regularizationcited by 46
- 2021Distributionally Robust Learningcited by 44
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
- 2023A Finite Sample Complexity Bound for Distributionally Robust Q-learningcited by 35
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2024Robust Reinforcement Learning from Corrupted Human Feedbackcited by 25
38 more
- 2021Certifying Robustness to Programmable Data Bias in Decision Treescited by 21
- 2023Understanding Contrastive Learning via Distributionally Robust Optimizationcited by 21
- 2021Fast Distributionally Robust Learning with Variance Reduced Min-Max Optimizationcited by 19
- 2021Zeroth-Order Methods for Convex-Concave Minmax Problems: Applications to Decision-Dependent Risk Minimizationcited by 18
- 2023The Dataset Multiplicity Problem: How Unreliable Data Impacts Predictionscited by 17
- 2022Distributionally Robust Bayesian Optimization with varphi-divergencescited by 16
- 2023Outlier-Robust Wasserstein DROcited by 16
- 2024Two-Timescale Gradient Descent Ascent Algorithms for Nonconvex Minimax Optimizationcited by 15
- 2021Distributionally Robust Learning with Stable Adversarial Trainingcited by 14
- 2023Towards Artificial General Intelligence (AGI) in the Internet of Things (IoT): Opportunities and Challengescited by 14
- 2021Semi-Discrete Optimal Transport: Hardness, Regularization and Numerical Solutioncited by 13
- 2023Unifying Distributionally Robust Optimization via Optimal Transport Theorycited by 12
- 2022Global-Local Regularization Via Distributional Robustnesscited by 11
- 2021The s-value: evaluating stability with respect to distributional shiftscited by 10
- 2021Coordinate Linear Variance Reduction for Generalized Linear Programmingcited by 10
- 2021Quasi-Newton Methods for Saddle Point Problems and Beyondcited by 10
- 2024Towards Robust Out-of-Distribution Generalization Bounds via Sharpnesscited by 10
- 2021Robust W-GAN-Based Estimation Under Wasserstein Contaminationcited by 9
- 2022Minimax risk classifiers with 0-1 losscited by 9
- 2022On the Generalization of Wasserstein Robust Federated Learningcited by 9
- 2023Bitrate-Constrained DRO: Beyond Worst Case Robustness To Unknown Group Shiftscited by 9
- 2023Exact Generalization Guarantees for (Regularized) Wasserstein Distributionally Robust Modelscited by 9
- 2021Distributed Saddle-Point Problems Under Similaritycited by 8
- 2023A first-order augmented Lagrangian method for constrained minimax optimizationcited by 8
- 2023Nash Equilibria, Regularization and Computation in Optimal Transport-Based Distributionally Robust Optimizationcited by 8
- 2021Robust Generalization despite Distribution Shift via Minimum Discriminating Informationcited by 6
- 2022Metrizing Fairnesscited by 6
- 2022Tikhonov Regularization is Optimal Transport Robust under Martingale Constraintscited by 6
- 2022Coresets for Wasserstein Distributionally Robust Optimization Problemscited by 6
- 2022On Generalization and Regularization via Wasserstein Distributionally Robust Optimizationcited by 6
- 2023Uncertainty-Aware Robust Learning on Noisy Graphscited by 6
- 2022Learning Distributionally Robust Models at Scale via Composite Optimizationcited by 5
- 2023Optimal Transport Model Distributional Robustnesscited by 5
- 2024Universal generalization guarantees for Wasserstein distributionally robust modelscited by 5
- 2024Evaluating Model Performance Under Worst-case Subpopulationscited by 5
- 2024Toward Robust Neural Reconstruction from Sparse Point Setscited by 5
- 2025Generalization Bounds and Stopping Rules for Learning with Self-Selected Datacited by 5
- 2021Adversarially Robust Kernel Smoothingcited by 4
50 most cited of 124 All 124 (JSON)
- 2026Learning Risk Scores Robust to Unobserved Confounders22 Sep 2026
- 2026Single-Loop Stochastic Projected Damped Extragradient Methods for Stochastic Nonconvex--(Strongly) Concave Minimax Optimization18 Sep 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026Wasserstein Distributionally Robust Risk-Sensitive Estimation via Conditional Value-at-Risk20 Apr 2026
- 2026Safe Distributionally Robust Feature Selection under Covariate Shift17 Mar 2026
- 2026Efficient Distribution Learning with Error Bounds in Wasserstein Distance8 Feb 2026
- 2025A first-order method for nonconvex-strongly-concave constrained minimax optimization28 Dec 2025
- 2025Iterative Sampling Methods for Sinkhorn Distributionally Robust Optimization14 Dec 2025
- 2025Distributionally Robust Multimodal Machine Learning7 Nov 2025
- 2025Wasserstein Distributionally Robust Optimization Through the Lens of Structural Causal Models and Individual Fairness30 Sep 2025
- 2025Distributionally Robust Federated Learning with Outlier Resilience29 Sep 2025
38 more
- 2025Stochastic Bilevel Optimization with Heavy-Tailed Noise18 Sep 2025
- 2025Representation-Aware Distributionally Robust Optimization: A Knowledge Transfer Framework11 Sep 2025
- 2025Group Distributionally Robust Machine Learning under Group Level Distributional Uncertainty10 Sep 2025
- 2025Improving Mixup Calibration with Wasserstein Distributionally Robust Optimization22 Jun 2025
- 2025SFT-GO: Supervised Fine-Tuning with Group Optimization for Large Language Models17 Jun 2025
- 2025Unregularized limit of stochastic gradient method for Wasserstein distributionally robust optimization5 Jun 2025
- 2025Generalization Bounds and Stopping Rules for Learning with Self-Selected Data12 May 2025
- 2025Nested Stochastic Algorithm for Generalized Sinkhorn distance-Regularized Distributionally Robust Optimization29 Mar 2025
- 2025Mixed-feature Logistic Regression Robust to Distribution Shifts15 Mar 2025
- 2025A stochastic smoothing framework for nonconvex-nonconcave minEmax problems with applications to Wasserstein distributionally robust optimization24 Feb 2025
- 2024Alternating minimization for square root principal component pursuit31 Dec 2024
- 2024Toward Robust Neural Reconstruction from Sparse Point Sets20 Dec 2024
- 2024Distributionally Robust Clustered Federated Learning: A Case Study in Healthcare9 Oct 2024
- 2024FDR-SVM: A Federated Distributionally Robust Support Vector Machine via a Mixture of Wasserstein Balls Ambiguity Set4 Oct 2024
- 2024Wasserstein Distributionally Robust Multiclass Support Vector Machine12 Sep 2024
- 2024Relative Translation Invariant Wasserstein Distance4 Sep 2024
- 2024Two-Timescale Gradient Descent Ascent Algorithms for Nonconvex Minimax Optimization21 Aug 2024
- 2024Generalizing Few Data to Unseen Domains Flexibly Based on Label Smoothing Integrated with Distributionally Robust Optimization9 Aug 2024
- 2024Completely Parameter-Free Single-Loop Algorithms for Nonconvex-Concave Minimax Problems31 Jul 2024
- 2024Distributionally and Adversarially Robust Logistic Regression via Intersecting Wasserstein Balls18 Jul 2024
- 2024Statistical Reachability Analysis of Stochastic Cyber-Physical Systems under Distribution Shift16 Jul 2024
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimization10 Jul 2024
- 2024Distributionally robust risk evaluation with an isotonic constraint9 Jul 2024
- 2024Evaluating Model Performance Under Worst-case Subpopulations1 Jul 2024
- 2024Robust Reinforcement Learning from Corrupted Human Feedback21 Jun 2024
- 2024Language-guided Detection and Mitigation of Unknown Dataset Bias5 Jun 2024
- 2024Learning from Uncertain Data: From Possible Worlds to Possible Models28 May 2024
- 2024Taking a Moment for Distributional Robustness8 May 2024
- 2024Automatic Outlier Rectification via Optimal Transport21 Mar 2024
- 2024Towards Robust Out-of-Distribution Generalization Bounds via Sharpness11 Mar 2024
- 2024Universal generalization guarantees for Wasserstein distributionally robust models19 Feb 2024
- 2024An Inexact Halpern Iteration with Application to Distributionally Robust Optimization8 Feb 2024
- 2024A Distributionally Robust Optimisation Approach to Fair Credit Scoring2 Feb 2024
- 2023Robust Regression over Averaged Uncertainty12 Nov 2023
- 2023Outlier-Robust Wasserstein DRO9 Nov 2023
- 2023Fair Graph Machine Learning under Adversarial Missingness Processes2 Nov 2023
- 2023Responsible AI (RAI) Games and Ensembles28 Oct 2023
- 2023Learning Optimal Classification Trees Robust to Distribution Shifts26 Oct 2023
50 newest of 124 All 124 (JSON)
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2021Distributional Robustness Loss for Long-tail Learningcited by 84
- 2021Twice regularized MDPs and the equivalence between robustness and regularizationcited by 46
- 2021Distributionally Robust Learningcited by 44
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
- 2023A Finite Sample Complexity Bound for Distributionally Robust Q-learningcited by 35
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2021Certifying Robustness to Programmable Data Bias in Decision Treescited by 21
- 2023Understanding Contrastive Learning via Distributionally Robust Optimizationcited by 21
12 most cited of 122 All 122 (JSON)
- 2026Learning Risk Scores Robust to Unobserved Confounders22 Sep 2026
- 2026Single-Loop Stochastic Projected Damped Extragradient Methods for Stochastic Nonconvex--(Strongly) Concave Minimax Optimization18 Sep 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026Wasserstein Distributionally Robust Risk-Sensitive Estimation via Conditional Value-at-Risk20 Apr 2026
- 2026Safe Distributionally Robust Feature Selection under Covariate Shift17 Mar 2026
- 2026Efficient Distribution Learning with Error Bounds in Wasserstein Distance8 Feb 2026
- 2025A first-order method for nonconvex-strongly-concave constrained minimax optimization28 Dec 2025
- 2025Iterative Sampling Methods for Sinkhorn Distributionally Robust Optimization14 Dec 2025
- 2025Distributionally Robust Multimodal Machine Learning7 Nov 2025
- 2025Wasserstein Distributionally Robust Optimization Through the Lens of Structural Causal Models and Individual Fairness30 Sep 2025
- 2025Distributionally Robust Federated Learning with Outlier Resilience29 Sep 2025
12 newest of 122 All 122 (JSON)