Understanding SAM's Robustness to Noisy Labels through Gradient Down-weighting
Not on the map. Why?
Abstract (arXiv)
Sharpness-Aware Minimization (SAM) was introduced to improve generalization by seeking flat minima, yet it also exhibits robustness to label noise, a phenomenon that remains only partially understood. Prior work has mainly attributed this effect to SAM's tendency to prolong the learning of clean samples. In this work, we provide a complementary explanation by analyzing SAM at the element-wise level. We show that when noisy gradients dominate a parameter direction, their influence is reduced by the stronger amplification of clean gradients. This slows the memorization of noisy labels while sustaining clean learning, offering a more complete account of SAM's robustness. Building on this insight, we propose SANER (Sharpness-Aware Noise-Explicit Reweighting), a simple variant of SAM that explicitly magnifies this down-weighting effect. Experiments on benchmark image classification tasks with noisy labels demonstrate that SANER significantly mitigates noisy-label memorization and improves generalization over both SAM and SGD. Moreover, since SANER is designed from the mechanism of SAM, it can also be seamlessly integrated into SAM-like variants, further boosting their robustness.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 38 of its 60 references that are in the map. First posted on arXiv 26 Nov 2024.
Builds on
38 of 60 references.
References · 38
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2016Wide Residual Networkscited by 3,070
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,059
- 2018Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labelscited by 997
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2020DivideMix: Learning with Noisy Labels as Semi-supervised Learningcited by 692
- 2018Learning to Reweight Examples for Robust Deep Learningcited by 675
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2020Early-Learning Regularization Prevents Memorization of Noisy Labelscited by 416
26 more
- 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
- 2018Joint Optimization Framework for Learning with Noisy Labelscited by 293
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2017WebVision Database: Visual Learning and Understanding from Web Datacited by 262
- 2020Normalized Loss Functions for Deep Learning with Noisy Labelscited by 257
- 2021Learning with Noisy Labels Revisited: A Study Using Real-World Human Annotationscited by 256
- 2020Does label smoothing mitigate label noise?cited by 204
- 2022Surrogate Gap Minimization Improves Sharpness-Aware Trainingcited by 172
- 2019SELF: Learning to Filter Noisy Labels with Self-Ensemblingcited by 164
- 2022Towards Understanding Sharpness-Aware Minimizationcited by 146
- 2021Understanding and Improving Early Stopping for Learning with Noisy Labelscited by 113
- 2022Selective-Supervised Contrastive Learning with Noisy Labelscited by 112
- 2021Learning with Feature-Dependent Label Noise: A Progressive Approachcited by 102
- 2022Robust Training under Label Noise by Over-parameterizationcited by 93
- 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
- 2024Friendly Sharpness-Aware Minimizationcited by 53
- 2022The Dynamics of Sharpness-Aware Minimization: Bouncing Across Ravines and Drifting Towards Wide Minimacited by 51
- 2023An Adaptive Policy to Employ Sharpness-Aware Minimizationcited by 43
- 2022Mitigating Memorization of Noisy Labels by Clipping the Model Predictioncited by 40
- 2023Enhancing Sharpness-Aware Optimization Through Variance Suppressioncited by 37
- 2023Why Does Sharpness-Aware Minimization Generalize Better Than SGD?cited by 28
- 2024Why is SAM Robust to Label Noise?cited by 21
- 2024Fundamental Convergence Analysis of Sharpness-Aware Minimizationcited by 14
- 2020MetaInfoNet: Learning Task-Guided Information for Sample Reweightingcited by 5
- 2023On Statistical Properties of Sharpness-Aware Minimization: Provable Guaranteescited by 5
- 2023A New Inexact Gradient Descent Method with Applications to Nonsmooth Convex Optimizationcited by 2
Built on by
Not cited yet.