DGSAM: Domain Generalization via Individual Sharpness-Aware Minimization
Not on the map. Why?
Abstract (arXiv)
Domain generalization (DG) aims to learn models that perform well on unseen target domains by training on multiple source domains. Sharpness-Aware Minimization (SAM), known for finding flat minima that improve generalization, has therefore been widely adopted in DG. However, our analysis reveals that SAM in DG may converge to textitfake flat minima, where the total loss surface appears flat in terms of global sharpness but remains sharp with respect to individual source domains. To understand this phenomenon more precisely, we formalize the average worst-case domain risk as the maximum loss under domain distribution shifts within a bounded divergence, and derive a generalization bound that reveals the limitations of global sharpness-aware minimization. In contrast, we show that individual sharpness provides a valid upper bound on this risk, making it a more suitable proxy for robust domain generalization. Motivated by these insights, we shift the DG paradigm toward minimizing individual sharpness across source domains. We propose textitDecreased-overhead Gradual SAM (DGSAM), which applies gradual domain-wise perturbations in a computationally efficient manner to consistently reduce individual sharpness. Extensive experiments demonstrate that DGSAM not only improves average accuracy but also reduces performance variance across domains, while incurring less computational overhead than SAM.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 59 of its 75 references that are in the map. First posted on arXiv 30 Mar 2025.
Builds on
59 of 75 references.
References · 59
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015Domain-Adversarial Training of Neural Networkscited by 3,044
- 2019Invariant Risk Minimizationcited by 1,854
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,331
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,200
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,188
- 2020In Search of Lost Domain Generalizationcited by 1,054
- 2017Deeper, Broader and Artier Domain Generalizationcited by 986
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 763
38 more
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 742
- 2018Recognition in Terra Incognitacited by 658
- 2013Domain Generalization via Invariant Feature Representationcited by 623
- 2017Exploring Generalization in Deep Learningcited by 600
- 2021Domain Generalization with MixStylecited by 566
- 2018Generalizing to Unseen Domains via Adversarial Data Augmentationcited by 451
- 2021SWAD: Domain Generalization by Seeking Flat Minimacited by 434
- 2020Self-Challenging Improves Cross-Domain Generalizationcited by 426
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 359
- 2019Lookahead Optimizer: k steps forward, 1 step backcited by 336
- 2017Unified Deep Supervised Domain Adaptation and Generalizationcited by 323
- 2020Learning to Learn Single Domain Generalizationcited by 296
- 2020Learning to Generate Novel Domains for Domain Generalizationcited by 296
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 292
- 2019SGD: General Analysis and Improved Ratescited by 207
- 2020Adaptive Risk Minimization: Learning to Adapt to Domain Shiftcited by 183
- 2017Domain Generalization by Marginal Transfer Learningcited by 180
- 2022Surrogate Gap Minimization Improves Sharpness-Aware Trainingcited by 172
- 2019Learning to Optimize Domain Specific Normalization for Domain Generalizationcited by 161
- 2020Better Theory for SGD in the Nonconvex Worldcited by 158
- 2017Generalization in Deep Learningcited by 146
- 2018Domain Generalization via Conditional Invariant Representationcited by 146
- 2022Generalized Federated Learning via Sharpness Aware Minimizationcited by 144
- 2020Robust and Generalizable Visual Representation Learning via Random Convolutionscited by 142
- 2020Single-Side Domain Generalization for Face Anti-Spoofingcited by 139
- 2021Efficient Sharpness-aware Minimization for Improved Training of Neural Networkscited by 137
- 2019Feature-Critic Networks for Heterogeneous Domain Generalizationcited by 130
- 2022Towards Efficient and Scalable Sharpness-Aware Minimizationcited by 129
- 2020Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularizationcited by 123
- 2023Sharpness-Aware Gradient Matching for Domain Generalizationcited by 102
- 2022Improving Generalization in Federated Learning by Seeking Flat Minimacited by 98
- 2015Scatter Component Analysis: A Unified Framework for Domain Adaptation and Domain Generalizationcited by 96
- 2020Towards Universal Representation Learning for Deep Face Recognitioncited by 78
- 2023Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalizationcited by 57
- 2019Adversarial Invariant Feature Learning with Accuracy Constraint for Domain Generalizationcited by 43
- 2023Flatness-Aware Minimization for Domain Generalizationcited by 27
- 2024Gradient Alignment for Cross-Domain Face Anti-Spoofingcited by 27
50 most cited of 59 All 59 (JSON)
Built on by
Not cited yet.