Better Diffusion Models Further Improve Adversarial Training
On the map: Diffusion models → Robustness, privacy and adversarial aspects
Abstract (arXiv)
It has been recognized that the data generated by the denoising diffusion probabilistic model (DDPM) improves adversarial training. After two years of rapid development in diffusion models, a question naturally arises: can better diffusion models further improve adversarial training? This paper gives an affirmative answer by employing the most recent diffusion model which has higher efficiency (~ 20 sampling steps) and image quality (lower FID score) compared with DDPM. Our adversarially trained models achieve state-of-the-art performance on RobustBench using only generated data (no external datasets). Under the ℓ_∞-norm threat model with ε=8/255, our models achieve 70.69% and 42.67% robust accuracy on CIFAR-10 and CIFAR-100, respectively, i.e. improving upon previous state-of-the-art models by +4.58% and +8.03%. Under the ℓ2-norm threat model with ε=128/255, our models achieve 84.86% on CIFAR-10 (+4.44%). These results also beat previous works that use external data. We also provide compelling results on the SVHN and TinyImageNet datasets. Our code is available at https://github.com/wzekai99/DM-Improves-AT.
Key facts. Cited by 207 papers in the map (data through 7 Oct 2026). Placed in Robustness, privacy and adversarial aspects, one of 14 directions in Diffusion models. Builds on 89 of its 112 references that are in the map. First posted on arXiv 9 Feb 2023.
Builds on
89 of 112 references.
Closest ancestors · 5
Same direction, newest first
- 2022DensePure: Understanding Diffusion Models towards Adversarial Robustness1 Nov
- 2022(Certified!!) Adversarial Robustness for Free!21 Jun
- 2022Guided Diffusion Model for Adversarial Purification30 May
- 2022Diffusion Models for Adversarial Purification16 May
- 2021Adversarial purification with Score-based generative models11 Jun
Other references · 84
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
38 more
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2016Wide Residual Networkscited by 3,057
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,609
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,623
- 2019Theoretically Principled Trade-off between Robustness and Accuracycited by 1,510
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2020Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attackscited by 1,442
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2021Variational Diffusion Modelscited by 1,145
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,036
- 2019When Does Label Smoothing Help?cited by 985
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2017Ensemble Adversarial Training: Attacks and Defensescited by 833
- 2020Fast is better than free: Revisiting adversarial trainingcited by 723
- 2020RobustBench: a standardized adversarial robustness benchmarkcited by 659
- 2019Adversarial Training for Free!cited by 638
- 2020On Adaptive Attacks to Adversarial Example Defensescited by 547
- 2020Overfitting in adversarially robust deep learningcited by 504
- 2019Unlabeled Data Improves Adversarial Robustnesscited by 415
- 2019Using Pre-Training Can Improve Model Robustness and Uncertaintycited by 373
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Modelscited by 331
- 2018Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problemcited by 323
- 2018Adversarially Robust Generalization Requires More Datacited by 318
- 2020Uncovering the Limits of Adversarial Training against Norm-Bounded Adversarial Examplescited by 275
- 2017PixelDefend: Leveraging Generative Models to Understand and Defend against Adversarial Examplescited by 243
- 2020Attacks Which Do Not Kill Training Make Adversarial Learning Strongercited by 242
- 2021Improving Robustness using Generated Datacited by 233
- 2021Adversarial GLUE: A Multi-Task Benchmark for Robustness Evaluation of Language Modelscited by 227
- 2019Improving Adversarial Robustness via Promoting Ensemble Diversitycited by 213
- 2020Understanding and Improving Fast Adversarial Trainingcited by 213
- 2021Fixing Data Augmentation to Improve Adversarial Robustnesscited by 201
- 2020Bag of Tricks for Adversarial Trainingcited by 199
50 most cited of 84 All 84 (JSON)
Built on by
Citing papers per year: 2023: 49; 2024: 76; 2025: 57; 2026: 25.
207 papers.
Sort:
By area:
- 2024HarmBench: A Standardized Evaluation Framework for Automated Red Teaming and Robust Refusalcited by 1,330
- 2023Image Hijacks: Adversarial Images can Control Generative Models at Runtimecited by 165
- 2023Robust Classification via a Single Diffusion Modelcited by 79
- 2023How Deep Learning Sees the World: A Survey on Adversarial Attacks & Defensescited by 65
- 2023Decoupled Kullback-Leibler Divergence Losscited by 58
- 2024One Prompt Word is Enough to Boost Adversarial Robustness for Pre-trained Vision-Language Modelscited by 48
- 2023Robust Principles: Architectural Design Principles for Adversarially Robust CNNscited by 46
- 2024Adversarial Robustness Limits via Scaling-Law and Human-Alignment Studiescited by 38
- 2023Diffusion Models and Semi-Supervised Learners Benefit Mutually with Few Labelscited by 37
- 2024Self-Consuming Generative Models with Curated Data Provably Optimize Human Preferencescited by 35
- 2023DiffProtect: Generate Adversarial Examples with Diffusion Models for Facial Privacy Protectioncited by 34
- 2024Revisiting Adversarial Training at Scalecited by 33
38 more
- 2024Safeguarding Vision-Language Models Against Patched Visual Prompt Injectorscited by 26
- 2023Toward Understanding Generative Data Augmentationcited by 25
- 2024ADBM: Adversarial diffusion bridge model for reliable adversarial purificationcited by 24
- 2024An End-to-End Robust Point Cloud Semantic Segmentation Network with Single-Step Conditional Diffusion Modelscited by 22
- 2025X-Transfer Attacks: Towards Super Transferable Adversarial Attacks on CLIPcited by 21
- 2024MeanSparse: Post-Training Robustness Enhancement Through Mean-Centered Feature Sparsificationcited by 20
- 2023The Devil's Advocate: Shattering the Illusion of Unexploitable Data using Diffusion Modelscited by 19
- 2023Enhancing Adversarial Robustness via Score-Based Optimizationcited by 19
- 2023MimicDiffusion: Purifying Adversarial Perturbation via Mimicking Clean Diffusion Modelcited by 18
- 2024MixedNUTS: Training-Free Accuracy-Robustness Balance via Nonlinearly Mixed Classifierscited by 16
- 2023Wasserstein distributional robustness of neural networkscited by 15
- 2024Adversarial Training on Purification (AToP): Advancing Both Robustness and Generalizationcited by 15
- 2024AttackBench: Evaluating Gradient-based Attacks for Adversarial Examplescited by 14
- 2024Instant Adversarial Purification with Adversarial Consistency Distillationcited by 14
- 2023Adversarial Training Should Be Cast as a Non-Zero-Sum Gamecited by 13
- 2024Revisiting Adversarial Training under Long-Tailed Distributionscited by 13
- 2023MIMIR: Masked Image Modeling for Mutual Information-based Adversarial Robustnesscited by 12
- 2024Preventing Catastrophic Overfitting in Fast Adversarial Training: A Bi-level Optimization Perspectivecited by 12
- 2024Adversarial Training Can Provably Improve Robustness: Theoretical Analysis of Feature Learning Process Under Structured Datacited by 12
- 2024DAT: Improving Adversarial Robustness via Generative Amplitude Mix-up in Frequency Domaincited by 12
- 2025CLIPure: Purification in Latent Space via CLIP for Adversarially Robust Zero-Shot Classificationcited by 12
- 2023Mitigating Adversarial Vulnerability through Causal Parameter Estimation by Adversarial Double Machine Learningcited by 11
- 2025Enhancing Robust Fairness via Confusional Spectral Regularizationcited by 11
- 2024Detecting Brittle Decisions for Free: Leveraging Margin Consistency in Deep Robust Classifierscited by 10
- 2023An Extended Study of Human-like Behavior under Adversarial Trainingcited by 9
- 2024Shedding More Light on Robust Classifiers under the lens of Energy-based Modelscited by 9
- 2024Adversarial Guided Diffusion Models for Adversarial Purificationcited by 8
- 2024Towards Understanding the Robustness of Diffusion-Based Purification: A Stochastic Perspectivecited by 8
- 2023Raising the Bar for Certified Adversarial Robustness with Diffusion Modelscited by 7
- 2023Benchmark Generation Framework with Customizable Distortions for Image Classifier Robustnesscited by 7
- 2024Boosting Adversarial Training via Fisher-Rao Norm-based Regularizationcited by 7
- 2024PartImageNet++ Dataset: Scaling up Part-based Models for Robust Recognitioncited by 7
- 2024LoRID: Low-Rank Iterative Diffusion for Adversarial Purificationcited by 7
- 2023AROID: Improving Adversarial Robustness Through Online Instance-Wise Data Augmentationcited by 6
- 2023Assessing Robustness via Score-Based Adversarial Image Generationcited by 6
- 2024Robust Overfitting Does Matter: Test-Time Adversarial Purification With FGSMcited by 6
- 2024Adversarial Training: A Surveycited by 6
- 2024Decoupled Data Augmentation for Improving Image Classificationcited by 6
50 most cited of 207 All 207 (JSON)
- 2026Don't Waste the Noise: Importance-Guided Perturbation Allocation under Joint Global and Local Constraints1 Oct 2026
- 2026Improving Generative Model Self-Training with Geometrically Modified Outputs28 Sep 2026
- 2026Beyond Decision Boundaries: Relational Geometry Attacks on Contrastive Embedding Manifolds10 Aug 2026
- 2026Adversarial Frontiers: Minimum-Norm Attack Ensembles for Robustness Evaluation22 Jul 2026
- 2026VanillaBench: The Hidden Accuracy Cost of Adversarial Robustness14 Jul 2026
- 2026Post-Generation Curation of Synthetic Images via Homogeneous-Heterogeneous Splitting2 Jul 2026
- 2026Generative AI and Federated Learning for Intrusion Detection Systems: A Survey1 Jul 2026
- 2026MAPE: Defending Against Transferable Adversarial Attacks Using Multi-Source Adversarial Perturbations Elimination30 Jun 2026
- 2026Class-frequency Guided Noise Schedule for Diffusion Models26 Jun 2026
- 2026The Scissors Effect: When Resize-Based Input Diversity Helps or Hurts Transfer Attacks21 Jun 2026
- 2026Flux-Guard: Facial Identity Protection using diffusion models16 Jun 2026
- 2026Adaptive Causal Alignment for High-Confidence Adversarial Training2 Jun 2026
38 more
- 2026A combination of noise and bilateral filters achieve supralinear and scalable adversarial robustness in CNNs1 Jun 2026
- 2026Sensitivity as a Double-Edged Sword: A Trade-off Between Discriminability and Adversarial Robustness1 Jun 2026
- 2026Taming the Long Tail: Rebalancing Adversarial Training via Adaptive Perturbation13 May 2026
- 2026TARO: Temporal Adversarial Rectification Optimization Using Diffusion Models as Purifiers8 May 2026
- 2026Detecting Adversarial Data via Provable Adversarial Noise Amplification4 May 2026
- 2026Useful nonrobust features are ubiquitous in biomedical images24 Apr 2026
- 2026Compression as an Adversarial Amplifier Through Decision Space Reduction8 Apr 2026
- 2026Solving adversarial examples requires solving exponential misalignment3 Mar 2026
- 2026S2O: Enhancing Adversarial Training with Second-Order Statistics of Weights1 Mar 2026
- 2026Expanding the Role of Diffusion Models for Robust Classifier Training23 Feb 2026
- 2026Closing the Distribution Gap in Adversarial Training for LLMs16 Feb 2026
- 2026FMVP: Masked Flow Matching for Adversarial Video Purification5 Jan 2026
- 2026PartImageNet++ Dataset: Enhancing Visual Models with High-Quality Part Annotations4 Jan 2026
- 2025Stylized Synthetic Augmentation further improves Corruption Robustness17 Dec 2025
- 2025Sample-wise Adaptive Weighting for Transfer Consistency in Adversarial Distillation11 Dec 2025
- 2025Convergence and Stability Analysis of Self-Consuming Generative Models with Heterogeneous Human Curation12 Nov 2025
- 2025Tight Robustness Certificates and Wasserstein Distributional Attacks for Deep Neural Networks11 Oct 2025
- 2025MANI-Pure: Magnitude-Adaptive Noise Injection for Adversarial Purification29 Sep 2025
- 2025On the Escaping Efficiency of Distributed Adversarial Training Algorithms14 Sep 2025
- 2025Differentiable Entropy Regularization: A Complexity-Aware Approach for Neural Optimization3 Sep 2025
- 2025Does simple trump complex? Comparing strategies for adversarial robustness in DNNs25 Aug 2025
- 2025Robustness Feature Adapter for Efficient Adversarial Training25 Aug 2025
- 2025DASH: A Meta-Attack Framework for Synthesizing Effective and Stealthy Adversarial Examples18 Aug 2025
- 2025ViT-EnsembleAttack: Augmenting Ensemble Models for Stronger Adversarial Transferability in Vision Transformers17 Aug 2025
- 2025A Guide to Robust Generalization: The Impact of Architecture, Pre-training, and Optimization Strategy12 Aug 2025
- 2025DBLP: Noise Bridge Consistency Distillation For Efficient And Reliable Adversarial Purification1 Aug 2025
- 2025Rectifying Adversarial Sample with Low Entropy Prior for Test-Time Defense4 Jul 2025
- 2025Are Fast Methods Stable in Adversarially Robust Transfer Learning?27 Jun 2025
- 2025Busting the Paper Ballot: Voting Meets Adversarial Machine Learning17 Jun 2025
- 2025On the Existence of Consistent Adversarial Attacks in High-Dimensional Linear Classification14 Jun 2025
- 2025Canonical Latent Representations in Conditional Diffusion Models11 Jun 2025
- 2025Effective Data Pruning through Score Extrapolation10 Jun 2025
- 2025Identifying and Understanding Cross-Class Features in Adversarial Training5 Jun 2025
- 2025Robustness in Both Domains: CLIP Needs a Robust Text Encoder3 Jun 2025
- 2025A Red Teaming Roadmap Towards System-Level Safety30 May 2025
- 2025Understanding Adversarial Training with Energy-based Models28 May 2025
- 2025What is Adversarial Training for Diffusion Models?27 May 2025
- 2025Your Classifier Can Do More: Towards Balancing the Gaps in Classification, Robustness, and Generation26 May 2025
50 newest of 207 All 207 (JSON)
- 2023Robust Classification via a Single Diffusion Modelcited by 79
- 2023Diffusion Models and Semi-Supervised Learners Benefit Mutually with Few Labelscited by 37
- 2023DiffProtect: Generate Adversarial Examples with Diffusion Models for Facial Privacy Protectioncited by 34
- 2024ADBM: Adversarial diffusion bridge model for reliable adversarial purificationcited by 24
- 2024An End-to-End Robust Point Cloud Semantic Segmentation Network with Single-Step Conditional Diffusion Modelscited by 22
- 2023The Devil's Advocate: Shattering the Illusion of Unexploitable Data using Diffusion Modelscited by 19
- 2023Enhancing Adversarial Robustness via Score-Based Optimizationcited by 19
- 2023MimicDiffusion: Purifying Adversarial Perturbation via Mimicking Clean Diffusion Modelcited by 18
- 2024Adversarial Guided Diffusion Models for Adversarial Purificationcited by 8
- 2024Towards Understanding the Robustness of Diffusion-Based Purification: A Stochastic Perspectivecited by 8
- 2023Raising the Bar for Certified Adversarial Robustness with Diffusion Modelscited by 7
- 2024LoRID: Low-Rank Iterative Diffusion for Adversarial Purificationcited by 7
12 most cited of 34 All 34 (JSON)
- 2026Class-frequency Guided Noise Schedule for Diffusion Models26 Jun 2026
- 2026Flux-Guard: Facial Identity Protection using diffusion models16 Jun 2026
- 2026TARO: Temporal Adversarial Rectification Optimization Using Diffusion Models as Purifiers8 May 2026
- 2026Expanding the Role of Diffusion Models for Robust Classifier Training23 Feb 2026
- 2026FMVP: Masked Flow Matching for Adversarial Video Purification5 Jan 2026
- 2025MANI-Pure: Magnitude-Adaptive Noise Injection for Adversarial Purification29 Sep 2025
- 2025Canonical Latent Representations in Conditional Diffusion Models11 Jun 2025
- 2025What is Adversarial Training for Diffusion Models?27 May 2025
- 2025Defending Against Frequency-Based Attacks with Diffusion Models15 Apr 2025
- 2025Active Learning Inspired ControlNet Guidance for Augmenting Semantic Segmentation Datasets12 Mar 2025
- 2025Improved Diffusion-based Generative Model with Better Adversarial Robustness24 Feb 2025
- 2025Diffusion or Non-Diffusion Adversarial Defenses: Rethinking the Relation between Classifier and Adversarial Purifier28 Jan 2025
12 newest of 34 All 34 (JSON)
- 2024One Prompt Word is Enough to Boost Adversarial Robustness for Pre-trained Vision-Language Modelscited by 48
- 2024Safeguarding Vision-Language Models Against Patched Visual Prompt Injectorscited by 26
- 2025X-Transfer Attacks: Towards Super Transferable Adversarial Attacks on CLIPcited by 21
- 2023MIMIR: Masked Image Modeling for Mutual Information-based Adversarial Robustnesscited by 12
- 2024Towards Adversarially Robust Vision-Language Models: Insights from Design Choices and Prompt Formatting Techniquescited by 5
- 2025X-Transfer Attacks: Towards Super Transferable Adversarial Attacks on CLIP8 May 2025
- 2024Towards Adversarially Robust Vision-Language Models: Insights from Design Choices and Prompt Formatting Techniques15 Jul 2024
- 2024Safeguarding Vision-Language Models Against Patched Visual Prompt Injectors17 May 2024
- 2024One Prompt Word is Enough to Boost Adversarial Robustness for Pre-trained Vision-Language Models4 Mar 2024
- 2023MIMIR: Masked Image Modeling for Mutual Information-based Adversarial Robustness8 Dec 2023
- 2023Image Hijacks: Adversarial Images can Control Generative Models at Runtimecited by 165
- 2023How Deep Learning Sees the World: A Survey on Adversarial Attacks & Defensescited by 65
- 2023Decoupled Kullback-Leibler Divergence Losscited by 58
- 2023Robust Principles: Architectural Design Principles for Adversarially Robust CNNscited by 46
- 2024Adversarial Robustness Limits via Scaling-Law and Human-Alignment Studiescited by 38
- 2024Revisiting Adversarial Training at Scalecited by 33
- 2023Toward Understanding Generative Data Augmentationcited by 25
- 2024MeanSparse: Post-Training Robustness Enhancement Through Mean-Centered Feature Sparsificationcited by 20
- 2024MixedNUTS: Training-Free Accuracy-Robustness Balance via Nonlinearly Mixed Classifierscited by 16
- 2023Wasserstein distributional robustness of neural networkscited by 15
- 2024Adversarial Training on Purification (AToP): Advancing Both Robustness and Generalizationcited by 15
- 2024AttackBench: Evaluating Gradient-based Attacks for Adversarial Examplescited by 14
12 most cited of 165 All 165 (JSON)
- 2026Don't Waste the Noise: Importance-Guided Perturbation Allocation under Joint Global and Local Constraints1 Oct 2026
- 2026Improving Generative Model Self-Training with Geometrically Modified Outputs28 Sep 2026
- 2026Beyond Decision Boundaries: Relational Geometry Attacks on Contrastive Embedding Manifolds10 Aug 2026
- 2026Adversarial Frontiers: Minimum-Norm Attack Ensembles for Robustness Evaluation22 Jul 2026
- 2026VanillaBench: The Hidden Accuracy Cost of Adversarial Robustness14 Jul 2026
- 2026Post-Generation Curation of Synthetic Images via Homogeneous-Heterogeneous Splitting2 Jul 2026
- 2026Generative AI and Federated Learning for Intrusion Detection Systems: A Survey1 Jul 2026
- 2026MAPE: Defending Against Transferable Adversarial Attacks Using Multi-Source Adversarial Perturbations Elimination30 Jun 2026
- 2026The Scissors Effect: When Resize-Based Input Diversity Helps or Hurts Transfer Attacks21 Jun 2026
- 2026Adaptive Causal Alignment for High-Confidence Adversarial Training2 Jun 2026
- 2026A combination of noise and bilateral filters achieve supralinear and scalable adversarial robustness in CNNs1 Jun 2026
- 2026Sensitivity as a Double-Edged Sword: A Trade-off Between Discriminability and Adversarial Robustness1 Jun 2026
12 newest of 165 All 165 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025Demystifying Adversarial Robustness in Diffusion Models: Compression, Randomness, and Geometry
- 2025Diffusion Models are Robust Pretrainers
- 2023Denoising Diffusion Probabilistic Models as a Defense against Adversarial Attacks
- 2025Beyond Classification: Evaluating Diffusion Denoised Smoothing for Security-Utility Trade off
- 2021Improved Denoising Diffusion Probabilistic Models
- 2025ADT: Tuning Diffusion Models with Adversarial Supervision
- 2026A Low-Rank Defense Method for Adversarial Attack on Diffusion Models
- 2024Attacks and Defenses for Generative Diffusion Models: A Comprehensive Survey