Searching for Robustness: Loss Learning for Noisy Classification Tasks
Not on the map. Why?
Abstract (arXiv)
We present a "learning to learn" approach for automatically constructing white-box classification loss functions that are robust to label noise in the training data. We parameterize a flexible family of loss functions using Taylor polynomials, and apply evolutionary strategies to search for noise-robust losses in this space. To learn re-usable loss functions that can apply to new tasks, our fitness function scores their performance in aggregate across a range of training dataset and architecture combinations. The resulting white-box loss provides a simple and fast "plug-and-play" module that enables effective noise-robust learning in diverse downstream tasks, without requiring a special training procedure or network architecture. The efficacy of our method is demonstrated on a variety of datasets with both synthetic and real label noise, where we compare favourably to previous work.
Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 39 of its 54 references that are in the map. First posted on arXiv 27 Feb 2021.
Builds on
39 of 54 references.
References · 39
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,053
- 2018Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labelscited by 994
- 2019When Does Label Smoothing Help?cited by 985
- 2020Meta-Learning in Neural Networks: A Surveycited by 944
- 2018Regularized Evolution for Image Classifier Architecture Searchcited by 902
27 more
- 2018Learning to Reweight Examples for Robust Deep Learningcited by 670
- 2016Making Deep Neural Networks Robust to Label Noise: a Loss Correction Approachcited by 663
- 2017MentorNet: Learning Data-Driven Curriculum for Very Deep Neural Networks on Corrupted Labelscited by 652
- 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 580
- 2018The Open Images Dataset V4: Unified image classification, object detection, and visual relationship detection at scalecited by 542
- 2019Symmetric Cross Entropy for Robust Learning with Noisy Labelscited by 499
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 461
- 2017Robust Loss Functions under Label Noise for Deep Neural Networkscited by 441
- 2018Deep Learning for Classical Japanese Literaturecited by 415
- 2019Lookahead Optimizer: k steps forward, 1 step backcited by 333
- 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
- 2020Combating noisy labels by agreement: A joint training method with co-regularizationcited by 289
- 2020Normalized Loss Functions for Deep Learning with Noisy Labelscited by 257
- 2018Learning to Learn from Noisy Labeled Datacited by 160
- 2019Feature-Critic Networks for Heterogeneous Domain Generalizationcited by 130
- 2017Learned Optimizers that Scale and Generalizecited by 109
- 2015Webly Supervised Learning of Convolutional Networkscited by 95
- 2018Evolved Policy Gradientscited by 86
- 2019Improving Generalization in Meta Reinforcement Learning using Learned Objectivescited by 68
- 2019Robust Bi-Tempered Logistic Loss Based on Bregman Divergencescited by 61
- 2019Meta-Learning via Learned Losscited by 60
- 2018Deep Bilevel Learningcited by 48
- 2018Learning to Teach with Dynamic Loss Functionscited by 40
- 2019Addressing the Loss-Metric Mismatch with Adaptive Loss Alignmentcited by 33
- 2019Improved Training Speed, Accuracy, and Data Utilization Through Loss Function Optimizationcited by 28
- 2020Robustness of Accuracy Metric and its Inspirations in Learning with Noisy Labelscited by 20
- 2017Two-temperature logistic regression based on the Tsallis divergencecited by 14
Built on by
Citing papers per year: 2022: 3; 2023: 3; 2024: 4; 2025: 2; 2026: 1.
13 papers.
Sort:
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learningcited by 9
- 2022Meta Mirror Descent: Optimiser Learning for Fast Convergencecited by 7
- 2022Truncate-Split-Contrast: A Framework for Learning from Mislabeled Videoscited by 4
- 2023Meta-Learning Adaptive Loss Functionscited by 4
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learningcited by 4
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awarenesscited by 3
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networkscited by 2
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learningcited by 1
- 2024Data augmentation with automated machine learning: approaches and performance comparison with classical data augmentation methodscited by 1
- 2024Evolving Loss Functions for Specific Image Augmentation Techniquescited by 0
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learningcited by 0
- 2025Label-Noise Resistant Learning via Optimal Brain Damage Maskingcited by 0
- 2026Leveraging Data Symmetries to Select an Optimal Subset of Training Data under Label Noise3 May 2026
- 2025Label-Noise Resistant Learning via Optimal Brain Damage Masking13 Aug 2025
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learning21 Jan 2025
- 2024Evolving Loss Functions for Specific Image Augmentation Techniques9 Apr 2024
- 2024Data augmentation with automated machine learning: approaches and performance comparison with classical data augmentation methods13 Mar 2024
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learning1 Mar 2024
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networks30 Jan 2024
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
- 2023Meta-Learning Adaptive Loss Functions30 Jan 2023
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awareness18 Jan 2023
- 2022Truncate-Split-Contrast: A Framework for Learning from Mislabeled Videos27 Dec 2022
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learning19 Sep 2022
1 more
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learningcited by 9
- 2022Meta Mirror Descent: Optimiser Learning for Fast Convergencecited by 7
- 2022Truncate-Split-Contrast: A Framework for Learning from Mislabeled Videoscited by 4
- 2023Meta-Learning Adaptive Loss Functionscited by 4
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learningcited by 4
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awarenesscited by 3
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networkscited by 2
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learningcited by 1
- 2024Data augmentation with automated machine learning: approaches and performance comparison with classical data augmentation methodscited by 1
- 2024Evolving Loss Functions for Specific Image Augmentation Techniquescited by 0
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learningcited by 0
- 2025Label-Noise Resistant Learning via Optimal Brain Damage Maskingcited by 0
12 most cited of 13 All 13 (JSON)
- 2026Leveraging Data Symmetries to Select an Optimal Subset of Training Data under Label Noise3 May 2026
- 2025Label-Noise Resistant Learning via Optimal Brain Damage Masking13 Aug 2025
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learning21 Jan 2025
- 2024Evolving Loss Functions for Specific Image Augmentation Techniques9 Apr 2024
- 2024Data augmentation with automated machine learning: approaches and performance comparison with classical data augmentation methods13 Mar 2024
- 2024Fast and Efficient Local Search for Genetic Programming Based Loss Function Learning1 Mar 2024
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networks30 Jan 2024
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
- 2023Meta-Learning Adaptive Loss Functions30 Jan 2023
- 2023Improve Noise Tolerance of Robust Loss via Noise-Awareness18 Jan 2023
- 2022Truncate-Split-Contrast: A Framework for Learning from Mislabeled Videos27 Dec 2022
- 2022Learning Symbolic Model-Agnostic Loss Functions via Meta-Learning19 Sep 2022
12 newest of 13 All 13 (JSON)