Learning to Reweight Examples for Robust Deep Learning
Not on the map. Why?
Abstract (arXiv)
Deep neural networks have been shown to be very powerful modeling tools for many supervised learning tasks involving complex input patterns. However, they can also easily overfit to training set biases and label noises. In addition to various regularizers, example reweighting algorithms are popular solutions to these problems, but they require careful tuning of additional hyperparameters, such as example mining schedules and regularization hyperparameters. In contrast to past reweighting methods, which typically consist of functions of the cost value of each example, in this work we propose a novel meta-learning algorithm that learns to assign weights to training examples based on their gradient directions. To determine the example weights, our method performs a meta gradient descent step on the current mini-batch example weights (which are initialized from zero) to minimize the loss on a clean unbiased validation set. Our proposed method can be easily implemented on any type of deep network, does not require any additional hyperparameter tuning, and achieves impressive performance on class imbalance and corrupted label problems where only a small amount of clean validation data is available.
Key facts. Cited by 670 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Mar 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 172; 2022: 135; 2023: 115; 2024: 114; 2025: 65; 2026: 69.
670 papers.
Sort:
By area:
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 522
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
- 2021Gradient Matching for Domain Generalizationcited by 258
- 2021Curriculum Learning: A Surveycited by 257
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
- 2021Parametric Contrastive Learningcited by 217
- 2021Distribution Alignment: A Unified Framework for Long-tail Visual Recognitioncited by 198
- 2021CReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learningcited by 174
- 2021Contrastive Learning based Hybrid Networks for Long-Tailed Image Classificationcited by 130
- 2021Self-supervised Learning is More Robust to Dataset Imbalancecited by 122
- 2022Selective-Supervised Contrastive Learning with Noisy Labelscited by 112
- 2022Long-Tailed Recognition via Weight Balancingcited by 107
38 more
- 2021Regularizing Generative Adversarial Networks under Limited Datacited by 105
- 2021Learning with Feature-Dependent Label Noise: A Progressive Approachcited by 102
- 2021MetaSAug: Meta Semantic Augmentation for Long-Tailed Visual Recognitioncited by 99
- 2021Provably End-to-end Label-Noise Learning without Anchor Pointscited by 97
- 2021ResLT: Residual Learning for Long-tailed Recognitioncited by 96
- 2026TIP: Token Importance in On-Policy Distillationcited by 90
- 2022Spread Spurious Attribute: Improving Worst-group Accuracy with Spurious Attribute Estimationcited by 88
- 2021FINE Samples for Learning with Noisy Labelscited by 82
- 2021Jo-SRC: A Contrastive Approach for Combating Noisy Labelscited by 80
- 2021Robust Audio-Visual Instance Discriminationcited by 80
- 2021OOD-GNN: Out-of-Distribution Generalized Graph Neural Networkcited by 78
- 2023LLMaAA: Making Large Language Models as Active Annotatorscited by 78
- 2022A Survey on Long-Tailed Visual Recognitioncited by 75
- 2023Long-tailed Visual Recognition via Gaussian Clouded Logit Adjustmentcited by 75
- 2021RETRIEVE: Coreset Selection for Efficient and Robust Semi-Supervised Learningcited by 72
- 2023DELTA: degradation-free fully test-time adaptationcited by 72
- 2021Learning Fast Sample Re-weighting Without Reward Datacited by 69
- 2022Learn From All: Erasing Attention Consistency for Noisy Label Facial Expression Recognitioncited by 67
- 2021ABC: Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised Learningcited by 65
- 2022Nested Collaborative Learning for Long-Tailed Visual Recognitioncited by 63
- 2023Guiding Pseudo-labels with Uncertainty Estimation for Source-free Unsupervised Domain Adaptationcited by 63
- 2022The Devil is in the Labels: Noisy Label Correction for Robust Scene Graph Generationcited by 60
- 2022Part-based Pseudo Label Refinement for Unsupervised Person Re-identificationcited by 59
- 2022Towards Intersectionality in Machine Learning: Including More Identities, Handling Underrepresentation, and Performing Evaluationcited by 59
- 2025Rethinking LLM Unlearning Objectives: A Gradient Perspective and Go Beyondcited by 59
- 2021Open-set Label Noise Can Improve Robustness Against Inherent Label Noisecited by 58
- 2021NGC: A Unified Framework for Learning with Open-World Noisy Datacited by 58
- 2022Confidence Score for Source-Free Unsupervised Domain Adaptationcited by 58
- 2021Topology-Imbalance Learning for Semi-Supervised Node Classificationcited by 56
- 2021Exploring Memorization in Adversarial Trainingcited by 55
- 2022DrugOOD: Out-of-Distribution (OOD) Dataset Curator and Benchmark for AI-aided Drug Discovery -- A Focus on Affinity Prediction Problems with Noise Annotationscited by 54
- 2022Meta Balanced Network for Fair Face Recognitioncited by 53
- 2021DASO: Distribution-Aware Semantics-Oriented Pseudo-label for Imbalanced Semi-Supervised Learningcited by 51
- 2021Adversarial Robustness under Long-Tailed Distributioncited by 50
- 2021Self-Training with Weak Supervisioncited by 49
- 2022Self-Guided Noise-Free Data Generation for Efficient Zero-Shot Learningcited by 49
- 2021Sample Selection for Fair and Robust Trainingcited by 48
- 2022Investigating Why Contrastive Learning Benefits Robustness Against Label Noisecited by 46
50 most cited of 670 All 670 (JSON)
- 2026Scalable, Transferable Meta-network for Data Selection Requires a Different Loss (and Why the Obvious Choice is Problematic)1 Oct 2026
- 2026Dr. OPD: Learning What to Follow for Optimal On-Policy Distillation of Large Language Models29 Sep 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026Age of Learning: Temporal Persistence of Prediction Errors as a Learning Signal26 Sep 2026
- 2026Which Influence Are We Estimating? The Role of Counterfactual Specifications in Data Attribution25 Sep 2026
- 2026LWCal: Loss-Weighted Calibration for Tabular Classifiers with Noisy Calibration Labels22 Sep 2026
- 2026Trajectory Learnability for Offline On-Policy Distillation with Imperfect Teachers16 Sep 2026
- 2026Bounded Adjustment with Reliability-Guided Embedding for Imbalanced Learning with Noisy Labels14 Sep 2026
- 2026Online Language Adaptive Sampling for Better Distributed Cross-lingual Gains14 Sep 2026
- 2026Rebalancing Token Importance in Language Models with TF-IDF Weighted Cross-Entropy Loss10 Sep 2026
- 2026Bayesian Data Reweighting Improves Multimodal Retrieval for Knowledge-Based Visual Question Answering3 Aug 2026
- 2026HERO: History-Enriched Rollout Training for Long-Horizon Autoregressive Neural Operators31 Jul 2026
38 more
- 2026Harnessing the Potential of Optimizing Data Mixtures via Bayesian Domain Reweighting30 Jul 2026
- 2026Dynamic Influence-Weighted Distillation for Single-IMU Activity Recognition17 Jul 2026
- 2026NodeImport: Imbalanced Node Classification with Node Importance Assessment15 Jul 2026
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimization11 Jul 2026
- 2026A Strong Balanced-Softmax Classifier-Retraining Baseline for Long-Tailed Recognition10 Jul 2026
- 2026POTracker: Optimizing Large Language Models for Standard-Compliant Power Outage Report Generation22 Jun 2026
- 2026Data Pruning: Redundant, Problematic, and Interdependent Samples20 Jun 2026
- 2026Deep Active Re-Labeling: Toward Noise-Resilient Annotation Efficiency7 Jun 2026
- 2026Quality-Guided Semi-Supervised Learning for Medical Image Segmentation1 Jun 2026
- 2026On the Difficulty of Learning a Meta-network for Training Data Selection30 May 2026
- 2026On Revisiting Entropy for Identifying Mislabeled Images29 May 2026
- 2026DAMEL: Dual-Axis Multi-Expert Learning for Class-Imbalanced Learning28 May 2026
- 2026Accelerated Dynamic Importance Weighting with Versatile Divergence-Minimizing Estimators25 May 2026
- 2026Factorize to Generalize: Retrieval-Guided Invariant-Dynamic Decomposition for Time Series Forecasting24 May 2026
- 2026CARE: Class-Adaptive Expert Consensus for Reliable Learning with Long-Tailed Noisy Labels22 May 2026
- 2026Cumulative Meta-Learning from Active Learning Queries for Robustness to Spurious Correlations20 May 2026
- 2026Balancing Knowledge Distillation for Imbalance Learning with Bilevel Optimization18 May 2026
- 2026Efficient Bilevel Optimization for Meta Label Correction in Noisy Label Learning18 May 2026
- 2026When Molecular Similarity Works: Property Cliffs Reveal Hidden Errors17 May 2026
- 2026Weighted Flow Matching and Physics-Informed Nonlinear Filtering for Parameter Estimation in Digital Twins16 May 2026
- 2026DynaMiCS: Fine-tuning LLMs with Performance Constraints using Dynamic Mixtures11 May 2026
- 2026BROS: Bias-Corrected Randomized Subspaces for Memory-Efficient Single-Loop Bilevel Optimization11 May 2026
- 2026Concordia: Self-Improving Synthetic Tables for Federated LLMs11 May 2026
- 2026Robust Server Defense Against Unreliable Clients in One-Shot Fair Collaborative Machine Learning9 May 2026
- 2026Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems8 May 2026
- 2026Multimodal Deep Generative Model for Semi-Supervised Learning under Class Imbalance7 May 2026
- 2026Architecture-agnostic Lipschitz-constant Bayesian header and its application to resolve semantically proximal classification errors with vision transformers7 May 2026
- 2026Not Every Subject Should Stay: Machine Unlearning for Noisy Engagement Recognition6 May 2026
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Training2 May 2026
- 2026Learning from Imperfect Text Guidance: Robust Long-Tail Visual Recognition with High-Noise Label25 Apr 2026
- 2026Model-Agnostic Meta Learning for Class Imbalance Adaptation20 Apr 2026
- 2026TIP: Token Importance in On-Policy Distillation15 Apr 2026
- 2026AbLWR:A Context-Aware Listwise Ranking Framework for Antibody-Antigen Binding Affinity Prediction via Positive-Unlabeled Learning13 Apr 2026
- 2026Nexus: Same Pretraining Loss, Better Downstream Generalization via Common Minima10 Apr 2026
- 2026Dynamic Class-Aware Active Learning for Unbiased Satellite Image Segmentation10 Apr 2026
- 2026Fine-grained Analysis of Stability and Generalization for Stochastic Bilevel Optimization5 Apr 2026
- 2026DataFlex: A Unified Framework for Data-Centric Dynamic Training of Large Language Models27 Mar 2026
- 2026AI Security in the Foundation Model Era: A Comprehensive Survey from a Unified Perspective25 Mar 2026
50 newest of 670 All 670 (JSON)
- 2024Training Unbiased Diffusion Models From Biased Datasetcited by 25
- 2024Don't drop your samples! Coherence-aware training benefits Conditional diffusioncited by 13
- 2025Robust Learning of Diffusion Models with Extremely Noisy Conditionscited by 2
- 2025Restoring Noisy Demonstration for Imitation Learning With Diffusion Modelscited by 1
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Trainingcited by 1
- 2026Weighted Flow Matching and Physics-Informed Nonlinear Filtering for Parameter Estimation in Digital Twinscited by 0
- 2026Weighted Flow Matching and Physics-Informed Nonlinear Filtering for Parameter Estimation in Digital Twins16 May 2026
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Training2 May 2026
- 2025Restoring Noisy Demonstration for Imitation Learning With Diffusion Models16 Oct 2025
- 2025Robust Learning of Diffusion Models with Extremely Noisy Conditions11 Oct 2025
- 2024Don't drop your samples! Coherence-aware training benefits Conditional diffusion30 May 2024
- 2024Training Unbiased Diffusion Models From Biased Dataset2 Mar 2024
- 2022How to Sift Out a Clean Data Subset in the Presence of Data Poisoning?cited by 17
- 2025LoReUn: Data Itself Implicitly Provides Cues to Improve Machine Unlearningcited by 3
- 2025Classifying Long-tailed and Label-noise Data via Disentangling and Unlearningcited by 2
- 2026AI Security in the Foundation Model Era: A Comprehensive Survey from a Unified Perspectivecited by 1
- 2025Machine Unlearning for Robust DNNs: Attribution-Guided Partitioning and Neuron Pruning in Noisy Environmentscited by 0
- 2026Not Every Subject Should Stay: Machine Unlearning for Noisy Engagement Recognitioncited by 0
- 2026Not Every Subject Should Stay: Machine Unlearning for Noisy Engagement Recognition6 May 2026
- 2026AI Security in the Foundation Model Era: A Comprehensive Survey from a Unified Perspective25 Mar 2026
- 2025LoReUn: Data Itself Implicitly Provides Cues to Improve Machine Unlearning30 Jul 2025
- 2025Machine Unlearning for Robust DNNs: Attribution-Guided Partitioning and Neuron Pruning in Noisy Environments13 Jun 2025
- 2025Classifying Long-tailed and Label-noise Data via Disentangling and Unlearning14 Mar 2025
- 2022How to Sift Out a Clean Data Subset in the Presence of Data Poisoning?12 Oct 2022
- 2024ROPO: Robust Preference Optimization for Large Language Modelscited by 16
- 2025Pre-DPO: Improving Data Utilization in Direct Preference Optimization Using a Guiding Reference Modelcited by 6
- 2026CTPD: Cross Tokenizer Preference Distillationcited by 3
- 2025What Is Preference Optimization Doing, and Why?cited by 2
- 2026SP^2DPO: An LLM-assisted Semantic Per-Pair DPO Generalizationcited by 0
- 2026SP^2DPO: An LLM-assisted Semantic Per-Pair DPO Generalization29 Jan 2026
- 2026CTPD: Cross Tokenizer Preference Distillation17 Jan 2026
- 2025What Is Preference Optimization Doing, and Why?30 Nov 2025
- 2025Pre-DPO: Improving Data Utilization in Direct Preference Optimization Using a Guiding Reference Model22 Apr 2025
- 2024ROPO: Robust Preference Optimization for Large Language Models5 Apr 2024
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 522
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
- 2021Gradient Matching for Domain Generalizationcited by 258
- 2021Curriculum Learning: A Surveycited by 257
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
- 2021Parametric Contrastive Learningcited by 217
- 2021Distribution Alignment: A Unified Framework for Long-tail Visual Recognitioncited by 198
- 2021CReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learningcited by 174
- 2021Contrastive Learning based Hybrid Networks for Long-Tailed Image Classificationcited by 130
- 2021Self-supervised Learning is More Robust to Dataset Imbalancecited by 122
- 2022Selective-Supervised Contrastive Learning with Noisy Labelscited by 112
- 2022Long-Tailed Recognition via Weight Balancingcited by 107
12 most cited of 639 All 639 (JSON)
- 2026Scalable, Transferable Meta-network for Data Selection Requires a Different Loss (and Why the Obvious Choice is Problematic)1 Oct 2026
- 2026Dr. OPD: Learning What to Follow for Optimal On-Policy Distillation of Large Language Models29 Sep 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026Age of Learning: Temporal Persistence of Prediction Errors as a Learning Signal26 Sep 2026
- 2026Which Influence Are We Estimating? The Role of Counterfactual Specifications in Data Attribution25 Sep 2026
- 2026LWCal: Loss-Weighted Calibration for Tabular Classifiers with Noisy Calibration Labels22 Sep 2026
- 2026Trajectory Learnability for Offline On-Policy Distillation with Imperfect Teachers16 Sep 2026
- 2026Bounded Adjustment with Reliability-Guided Embedding for Imbalanced Learning with Noisy Labels14 Sep 2026
- 2026Online Language Adaptive Sampling for Better Distributed Cross-lingual Gains14 Sep 2026
- 2026Rebalancing Token Importance in Language Models with TF-IDF Weighted Cross-Entropy Loss10 Sep 2026
- 2026Bayesian Data Reweighting Improves Multimodal Retrieval for Knowledge-Based Visual Question Answering3 Aug 2026
- 2026HERO: History-Enriched Rollout Training for Long-Horizon Autoregressive Neural Operators31 Jul 2026
12 newest of 639 All 639 (JSON)