mlsanityData to 14 Oct 2026

Understanding SAM's Robustness to Noisy Labels through Gradient Down-weighting

arXiv 2411.17132 ↗First posted 26 Nov 2024cs.LGcited by 0

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
  3. 2016Densely Connected Convolutional Networkscited by 6,720
  4. 2016Wide Residual Networkscited by 3,070
  5. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
  6. 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,059
  7. 2018Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labelscited by 997
  8. 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
  9. 2020DivideMix: Learning with Noisy Labels as Semi-supervised Learningcited by 692
  10. 2018Learning to Reweight Examples for Robust Deep Learningcited by 675
  11. 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
  12. 2020Early-Learning Regularization Prevents Memorization of Noisy Labelscited by 416
26 more
  1. 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
  2. 2018Joint Optimization Framework for Learning with Noisy Labelscited by 293
  3. 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
  4. 2017WebVision Database: Visual Learning and Understanding from Web Datacited by 262
  5. 2020Normalized Loss Functions for Deep Learning with Noisy Labelscited by 257
  6. 2021Learning with Noisy Labels Revisited: A Study Using Real-World Human Annotationscited by 256
  7. 2020Does label smoothing mitigate label noise?cited by 204
  8. 2022Surrogate Gap Minimization Improves Sharpness-Aware Trainingcited by 172
  9. 2019SELF: Learning to Filter Noisy Labels with Self-Ensemblingcited by 164
  10. 2022Towards Understanding Sharpness-Aware Minimizationcited by 146
  11. 2021Understanding and Improving Early Stopping for Learning with Noisy Labelscited by 113
  12. 2022Selective-Supervised Contrastive Learning with Noisy Labelscited by 112
  13. 2021Learning with Feature-Dependent Label Noise: A Progressive Approachcited by 102
  14. 2022Robust Training under Label Noise by Over-parameterizationcited by 93
  15. 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
  16. 2024Friendly Sharpness-Aware Minimizationcited by 53
  17. 2022The Dynamics of Sharpness-Aware Minimization: Bouncing Across Ravines and Drifting Towards Wide Minimacited by 51
  18. 2023An Adaptive Policy to Employ Sharpness-Aware Minimizationcited by 43
  19. 2022Mitigating Memorization of Noisy Labels by Clipping the Model Predictioncited by 40
  20. 2023Enhancing Sharpness-Aware Optimization Through Variance Suppressioncited by 37
  21. 2023Why Does Sharpness-Aware Minimization Generalize Better Than SGD?cited by 28
  22. 2024Why is SAM Robust to Label Noise?cited by 21
  23. 2024Fundamental Convergence Analysis of Sharpness-Aware Minimizationcited by 14
  24. 2020MetaInfoNet: Learning Task-Guided Information for Sample Reweightingcited by 5
  25. 2023On Statistical Properties of Sharpness-Aware Minimization: Provable Guaranteescited by 5
  26. 2023A New Inexact Gradient Descent Method with Applications to Nonsmooth Convex Optimizationcited by 2

Built on by

Not cited yet.