Abstract:Recently, sharpness-aware minimization (SAM) has emerged as a promising method to improve generalization by minimizing sharpness, which is known to correlate well with generalization ability. Since the original proposal of SAM, many variants of SAM have been proposed to improve its accuracy and efficiency, but comparisons have mainly been restricted to the i.i.d. setting. In this paper we study SAM for out-of-distribution (OOD) generalization. First, we perform a comprehensive comparison of eight SAM variants on zero-shot OOD generalization, finding that the original SAM outperforms the Adam baseline by $4.76\%$ and the strongest SAM variants outperform the Adam baseline by $8.01\%$ on average. We then provide an OOD generalization bound in terms of sharpness for this setting. Next, we extend our study of SAM to the related setting of gradual domain adaptation (GDA), another form of OOD generalization where intermediate domains are constructed between the source and target domains, and iterative self-training is done on intermediate domains, to improve the overall target domain error. In this setting, our experimental results demonstrate that the original SAM outperforms the baseline of Adam on each of the experimental datasets by $0.82\%$ on average and the strongest SAM variants outperform Adam by $1.52\%$ on average. We then provide a generalization bound for SAM in the GDA setting. Asymptotically, this generalization bound is no better than the one for self-training in the literature of GDA. This highlights a further disconnection between the theoretical justification for SAM versus its empirical performance, with recent work finding that low sharpness alone does not account for all of SAM's generalization benefits. For future work, we provide several potential avenues for obtaining a tighter analysis for SAM in the OOD setting.

softmax is not enough (for sharp out-of-distribution)

How the Softmax Output is Misleading for Evaluating the Strength of Adversarial Examples

Understanding Softmax Confidence and Uncertainty

Sharpness Minimization Algorithms Do Not Only Minimize Sharpness To Achieve Better Generalization

Towards Robust Out-of-Distribution Generalization Bounds Via Sharpness.

Convex Bounds on the Softmax Function with Applications to Robustness Verification

Sigsoftmax: Reanalysis of the Softmax Bottleneck

Exploring the Frontiers of Softmax: Provable Optimization, Applications in Diffusion Model, and Beyond

Making Robust Generalizers Less Rigid with Soft Ascent-Descent

Statistical Perspective of Top-K Sparse Softmax Gating Mixture of Experts

Imbalance Robust Softmax for Deep Embeeding Learning

Critical Influence of Overparameterization on Sharpness-aware Minimization

Density-Softmax: Efficient Test-time Model for Uncertainty Estimation and Robustness under Distribution Shifts

Online normalizer calculation for softmax

Rethinking Softmax Cross-Entropy Loss for Adversarial Robustness.

In Defense of Softmax Parametrization for Calibrated and Consistent Learning to Defer

Towards Understanding the Role of Sharpness-Aware Minimization Algorithms for Out-of-Distribution Generalization

Money on the Table: Statistical information ignored by Softmax can improve classifier accuracy

Why Robust Generalization in Deep Learning is Difficult: Perspective of Expressive Power

Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimization

Sharpness-Aware Black-Box Optimization