Learning Sample Reweighting for Accuracy and Adversarial Robustness
Not on the map. Why?
Abstract (arXiv)
There has been great interest in enhancing the robustness of neural network classifiers to defend against adversarial perturbations through adversarial training, while balancing the trade-off between robust accuracy and standard accuracy. We propose a novel adversarial training framework that learns to reweight the loss associated with individual training samples based on a notion of class-conditioned margin, with the goal of improving robust generalization. We formulate weighted adversarial training as a bilevel optimization problem with the upper-level problem corresponding to learning a robust classifier, and the lower-level problem corresponding to learning a parametric function that maps from a sample's textitmulti-class margin to an importance weight. Extensive experiments demonstrate that our approach consistently improves both clean and robust accuracy compared to related methods and state-of-the-art baselines.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 34 of its 42 references that are in the map. First posted on arXiv 20 Oct 2022.
Builds on
34 of 42 references.
References · 34
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2013Intriguing properties of neural networkscited by 5,114
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2016Wide Residual Networkscited by 3,057
- 2019Theoretically Principled Trade-off between Robustness and Accuracycited by 1,510
- 2020Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attackscited by 1,442
- 2020Fast is better than free: Revisiting adversarial trainingcited by 723
- 2018Learning to Reweight Examples for Robust Deep Learningcited by 670
- 2017One pixel attack for fooling deep neural networkscited by 579
- 2017Provable defenses against adversarial examples via the convex outer adversarial polytopecited by 564
22 more
- 2019Meta-Learning with Implicit Gradientscited by 523
- 2020Overfitting in adversarially robust deep learningcited by 504
- 2018Meta-Learning with Latent Embedding Optimizationcited by 476
- 2019Provably Robust Deep Learning via Adversarially Trained Smoothed Classifierscited by 328
- 2020Attacks Which Do Not Kill Training Make Adversarial Learning Strongercited by 242
- 2021Improving Robustness using Generated Datacited by 233
- 2020Geometry-aware Instance-reweighted Adversarial Trainingcited by 198
- 2021Data Augmentation Can Improve Robustnesscited by 193
- 2018MMA Training: Direct Input Space Margin Maximization through Adversarial Trainingcited by 171
- 2021Probabilistic Margins for Instance Reweighting in Adversarial Trainingcited by 57
- 2019Instance adaptive adversarial training: Improved accuracy tradeoffs in neural netscited by 55
- 2018Deep Bilevel Learningcited by 48
- 2020How benign is benign overfitting?cited by 40
- 2019Provable robustness against all adversarial lp-perturbations for p≥ 1cited by 39
- 2021Understanding the role of importance weighting for deep learningcited by 38
- 2020Calibrated Surrogate Losses for Adversarially Robust Classificationcited by 36
- 2019Identifying Mislabeled Instances in Classification Datasetscited by 26
- 2017Robustness to Adversarial Examples through an Ensemble of Specialistscited by 21
- 2021Reweighting Augmented Samples by Minimizing the Maximal Expected Losscited by 21
- 2020A Closer Look at Accuracy vs. Robustnesscited by 14
- 2020Are Adversarial Examples Created Equal? A Learnable Weighted Minimax Risk for Robustness under Non-uniform Attackscited by 14
- 2018Gradient Agreement as an Optimization Objective for Meta-Learningcited by 5
Built on by
Citing papers per year: 2023: 2; 2024: 2; 2025: 1; 2026: 1.
6 papers.
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learningcited by 39
- 2024A Short Survey on Importance Weighting for Machine Learningcited by 15
- 2023Identifying Adversarially Attackable and Robust Samplescited by 5
- 2024Improving Generalization via Meta-Learning on Hard Samplescited by 5
- 2025Revisiting Meta-Learning with Noisy Labels: Reweighting Dynamics and Theoretical Guaranteescited by 0
- 2026Rethinking Loss Reweighting for Imbalance Learning as an Inverse Problem: A Neural Collapse Point of Viewcited by 0
- 2026Rethinking Loss Reweighting for Imbalance Learning as an Inverse Problem: A Neural Collapse Point of View11 May 2026
- 2025Revisiting Meta-Learning with Noisy Labels: Reweighting Dynamics and Theoretical Guarantees14 Oct 2025
- 2024Improving Generalization via Meta-Learning on Hard Samples18 Mar 2024
- 2024A Short Survey on Importance Weighting for Machine Learning15 Mar 2024
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learning1 Aug 2023
- 2023Identifying Adversarially Attackable and Robust Samples30 Jan 2023