Certifying Some Distributional Robustness with Principled Adversarial Training
Not on the map. Why?
Abstract (arXiv)
Neural networks are vulnerable to adversarial examples and researchers have proposed many heuristic attack and defense mechanisms. We address this problem through the principled lens of distributionally robust optimization, which guarantees performance under adversarial input perturbations. By considering a Lagrangian penalty formulation of perturbing the underlying data distribution in a Wasserstein ball, we provide a training procedure that augments model parameter updates with worst-case perturbations of training data. For smooth losses, our procedure provably achieves moderate levels of robustness with little computational or statistical cost relative to empirical risk minimization. Furthermore, our statistical guarantees allow us to efficiently certify robustness for the population loss. For imperceptible perturbations, our method matches or outperforms heuristic approaches.
Key facts. Cited by 336 papers in the map (data through 13 Oct 2026). First posted on arXiv 29 Oct 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 75; 2022: 68; 2023: 62; 2024: 50; 2025: 41; 2026: 39.
336 papers.
Sort:
By area:
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,889
- 2021Domain Generalization: A Surveycited by 760
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 432
- 2021Data Collection and Quality Challenges in Deep Learning: A Data-Centric AI Perspectivecited by 116
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2023A Survey of Safety and Trustworthiness of Large Language Models through the Lens of Verification and Validationcited by 92
- 2021Towards Principled Disentanglement for Domain Generalizationcited by 91
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Adversarially Adaptive Normalization for Single Domain Generalizationcited by 85
- 2021Improved OOD Generalization via Adversarial Training and Pre-trainingcited by 66
- 2021Cross-Domain Few-Shot Classification via Adversarial Task Augmentationcited by 63
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 53
38 more
- 2021The Complexity of Nonconvex-Strongly-Concave Minimax Optimizationcited by 45
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2021Distributionally Robust Learningcited by 44
- 2021Constrained Learning with Non-Convex Lossescited by 42
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 42
- 2022Enhancing Adversarial Training with Second-Order Statistics of Weightscited by 42
- 2023Out-of-distribution Detection with Implicit Outlier Transformationcited by 42
- 2021Uncertainty-guided Model Generalization to Unseen Domainscited by 41
- 2021On the Certified Robustness for Ensemble Models and Beyondcited by 40
- 2021Adversarial Robustness with Semi-Infinite Constrained Learningcited by 39
- 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
- 2021TRS: Transferability Reduced Ensemble via Encouraging Gradient Diversity and Model Smoothnesscited by 38
- 2021How to Certify Machine Learning Based Safety-critical Systems? A Systematic Literature Reviewcited by 38
- 2022Unleashing the Power of Graph Data Augmentation on Covariate Distribution Shiftcited by 38
- 2023Learning to Augment Distributions for Out-of-Distribution Detectioncited by 38
- 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
- 2022Adversarial Feature Augmentation for Cross-domain Few-shot Classificationcited by 36
- 2021Quantifying and Improving Transferability in Domain Generalizationcited by 35
- 2023Anomaly Detection under Distribution Shiftcited by 35
- 2023Mind the Label Shift of Augmentation-based Graph OOD Generalizationcited by 34
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 34
- 2022Trustworthy Reinforcement Learning Against Intrinsic Vulnerabilities: Robustness, Safety, and Generalizabilitycited by 32
- 2022Learning to Solve Routing Problems via Distributionally Robust Optimizationcited by 31
- 2022Grounding Visual Representations with Texts for Domain Generalizationcited by 30
- 2021Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform Stabilitycited by 29
- 2021Modeling the Second Player in Distributionally Robust Optimizationcited by 29
- 2023Certified Minimax Unlearning with Generalization Rates and Deletion Capacitycited by 29
- 2022The Many Faces of Adversarial Riskcited by 28
- 2021Stochastic Bias-Reduced Gradient Methodscited by 25
- 2021Taxonomy of Machine Learning Safety: A Survey and Primercited by 23
- 2022Differentially Private SGDA for Minimax Problemscited by 23
- 2021Mixed Nash Equilibria in the Adversarial Examples Gamecited by 22
- 2021What Doesn't Kill You Makes You Robust(er): How to Adversarially Train against Data Poisoningcited by 22
- 2021P-Adapters: Robustly Extracting Factual Information from Language Models with Diverse Promptscited by 22
- 2021Robust Models Are More Interpretable Because Attributions Look Normalcited by 21
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
- 2023Understanding Contrastive Learning via Distributionally Robust Optimizationcited by 21
- 2023Rethinking the Effect of Data Augmentation in Adversarial Contrastive Learningcited by 20
50 most cited of 336 All 336 (JSON)
- 2026Stochastic Gradient Descent Ascent is Suboptimal for Nonconvex-PL Min-Max Games6 Oct 2026
- 2026Beyond Empirical Support: Structured Outlier Generation via Sinkhorn Optimal Transport25 Sep 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Distributionally Robust Federated Learning with Multi-Source Data17 Sep 2026
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalization4 Sep 2026
- 2026Layered LLM Defenses as an Ensemble: Access Tiers, Inference Cost, and the Measured Failure Correlation Between Defense Layers28 Aug 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026End-to-End Fairness Optimization with Fair Decision-Focused Learning31 Jul 2026
- 2026Generative Distributionally Robust Optimization27 Jul 2026
- 2026Distributionally Robust and Safe Imitation Learning15 Jul 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026Which Directions Matter? Sparse Design for Affine Robust Optimization12 Jun 2026
38 more
- 2026Amnesia: A Stealthy Replay Attack on Continual Learning Dreams10 Jun 2026
- 2026Reinforcement Learning Disrupts Gradient-Based Adversarial Optimization10 Jun 2026
- 2026Ambiguous Strategic Classification8 Jun 2026
- 2026Unification and Optimization of Robust Supervised Learning27 May 2026
- 2026Lipschitz Optimization for Formal Verification of Homographies22 May 2026
- 2026Robust Representation Learning through Explicit Environment Modeling28 Apr 2026
- 2026A Distributionally Robust Reinforcement Learning Framework for Constrained Urban EV Dispatch28 Apr 2026
- 2026Deep Learning for Sequential Decision Making under Uncertainty: Foundations, Frameworks, and Frontiers13 Apr 2026
- 2026Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization9 Apr 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026rSDNet: Unified Robust Neural Learning against Label Noise and Adversarial Attacks18 Mar 2026
- 2026Safe Distributionally Robust Feature Selection under Covariate Shift17 Mar 2026
- 2026Evaluating Black-Box Vulnerabilities with Wasserstein-Constrained Data Perturbations16 Mar 2026
- 2026Worst-case low-rank approximations11 Mar 2026
- 2026Robustness of Agentic AI Systems via Adversarially-Aligned Jacobian Regularization4 Mar 2026
- 2026The Price of Robustness: Stable Classifiers Need Overparameterization3 Mar 2026
- 2026S2O: Enhancing Adversarial Training with Second-Order Statistics of Weights1 Mar 2026
- 2026Risk-Averse Wasserstein Distributionally Robust Online Learning23 Feb 2026
- 2026Agentic Adversarial QA for Improving Domain-Specific LLMs20 Feb 2026
- 2026Bigger Is Safer: Provable Robustness in In-Context Learning Scales with Capacity19 Feb 2026
- 2026Differentially Private Non-convex Distributionally Robust Optimization18 Feb 2026
- 2026Distributionally Robust Optimization via Generative Ambiguity Modeling9 Feb 2026
- 2026Robust Policy Optimization to Prevent Catastrophic Forgetting9 Feb 2026
- 2026Certified Learning under Distribution Shift: Sound Verification and Identifiable Structure6 Feb 2026
- 2026Multivariate Time Series Data Imputation via Distributionally Robust Regularization31 Jan 2026
- 2026Robust Learning of a Group DRO Neuron26 Jan 2026
- 2026Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distance23 Jan 2026
- 2025A first-order method for nonconvex-strongly-concave constrained minimax optimization28 Dec 2025
- 2025Maximum Risk Minimization with Random Forests11 Dec 2025
- 2025Natural Geometry of Robust Data Attribution: From Convex Models to Deep Networks9 Dec 2025
- 2025Worst-case generation via minimax optimization in Wasserstein space9 Dec 2025
- 2025Structured Matching via Cost-Regularized Unbalanced Optimal Transport24 Nov 2025
- 2025Wasserstein Distributionally Robust Nash Equilibrium Seeking with Heterogeneous Data: A Lagrangian Approach18 Nov 2025
- 2025Evolving Graph Learning for Out-of-Distribution Generalization in Non-stationary Environments4 Nov 2025
- 2025ODP-Bench: Benchmarking Out-of-Distribution Performance Prediction31 Oct 2025
- 2025Gradient Flow Sampler-based Distributionally Robust Optimization29 Oct 2025
- 2025Distributionally Robust Optimization via Diffusion Ambiguity Modeling26 Oct 2025
- 2025Exploring Scale Shift in Crowd Localization under the Context of Domain Generalization22 Oct 2025
50 newest of 336 All 336 (JSON)
- 2024Meta Stackelberg Game: Robust Federated Learning against Adaptive and Mixed Poisoning Attackscited by 9
- 2024Risk-Sensitive Soft Actor-Critic for Robust Deep Reinforcement Learning under Distribution Shiftscited by 6
- 2024Model-Free Robust φ-Divergence Reinforcement Learning Using Both Offline and Online Datacited by 2
- 2026A Distributionally Robust Reinforcement Learning Framework for Constrained Urban EV Dispatchcited by 0
- 2026Reinforcement Learning Disrupts Gradient-Based Adversarial Optimizationcited by 0
- 2026Reinforcement Learning Disrupts Gradient-Based Adversarial Optimization10 Jun 2026
- 2026A Distributionally Robust Reinforcement Learning Framework for Constrained Urban EV Dispatch28 Apr 2026
- 2024Meta Stackelberg Game: Robust Federated Learning against Adaptive and Mixed Poisoning Attacks22 Oct 2024
- 2024Model-Free Robust φ-Divergence Reinforcement Learning Using Both Offline and Online Data8 May 2024
- 2024Risk-Sensitive Soft Actor-Critic for Robust Deep Reinforcement Learning under Distribution Shifts15 Feb 2024
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 34
- 2025Improved Diffusion-based Generative Model with Better Adversarial Robustnesscited by 3
- 2025Robustifying Diffusion-Denoised Smoothing Against Covariate Shiftcited by 0
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalizationcited by 0
- 2026PAPT++: Risk-Aware Adversarial Tuning and Generation for Single Domain Generalization4 Sep 2026
- 2025Robustifying Diffusion-Denoised Smoothing Against Covariate Shift13 Sep 2025
- 2025Improved Diffusion-based Generative Model with Better Adversarial Robustness24 Feb 2025
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Future11 Sep 2024
- 2023Certified Minimax Unlearning with Generalization Rates and Deletion Capacitycited by 29
- 2021What Doesn't Kill You Makes You Robust(er): How to Adversarially Train against Data Poisoningcited by 22
- 2025RLCracker: Evaluating the Worst-Case Vulnerability of LLM Watermarks with Adaptive RL Attackscited by 1
- 2025RLCracker: Evaluating the Worst-Case Vulnerability of LLM Watermarks with Adaptive RL Attacks25 Sep 2025
- 2023Certified Minimax Unlearning with Generalization Rates and Deletion Capacity16 Dec 2023
- 2021What Doesn't Kill You Makes You Robust(er): How to Adversarially Train against Data Poisoning26 Feb 2021
- 2021Domain Generalization: A Surveycited by 760
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 432
- 2021Data Collection and Quality Challenges in Deep Learning: A Data-Centric AI Perspectivecited by 116
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2023A Survey of Safety and Trustworthiness of Large Language Models through the Lens of Verification and Validationcited by 92
- 2021Towards Principled Disentanglement for Domain Generalizationcited by 91
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Adversarially Adaptive Normalization for Single Domain Generalizationcited by 85
- 2021Improved OOD Generalization via Adversarial Training and Pre-trainingcited by 66
- 2021Cross-Domain Few-Shot Classification via Adversarial Task Augmentationcited by 63
- 2021The Complexity of Nonconvex-Strongly-Concave Minimax Optimizationcited by 45
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
12 most cited of 313 All 313 (JSON)
- 2026Stochastic Gradient Descent Ascent is Suboptimal for Nonconvex-PL Min-Max Games6 Oct 2026
- 2026Beyond Empirical Support: Structured Outlier Generation via Sinkhorn Optimal Transport25 Sep 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Distributionally Robust Federated Learning with Multi-Source Data17 Sep 2026
- 2026Layered LLM Defenses as an Ensemble: Access Tiers, Inference Cost, and the Measured Failure Correlation Between Defense Layers28 Aug 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026End-to-End Fairness Optimization with Fair Decision-Focused Learning31 Jul 2026
- 2026Generative Distributionally Robust Optimization27 Jul 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026Which Directions Matter? Sparse Design for Affine Robust Optimization12 Jun 2026
- 2026Amnesia: A Stealthy Replay Attack on Continual Learning Dreams10 Jun 2026
- 2026Ambiguous Strategic Classification8 Jun 2026
12 newest of 313 All 313 (JSON)