Sharpness Aware Minimization
Sharpness-Aware Minimization (SAM) is an optimization technique aiming to improve the generalization ability of machine learning models by finding "flatter" minima in the loss landscape, reducing overfitting and enhancing robustness. Current research focuses on refining SAM's algorithm, including adaptive radius selection, bilateral sharpness estimation, and integrating it with other techniques like federated learning and uncertainty quantification, often applied to models such as transformers and convolutional neural networks. This approach holds significant promise for improving the performance and reliability of machine learning models across diverse applications, particularly in areas like medical image analysis and time series forecasting where generalization is crucial.
Papers
Sharpness-Aware Minimization Leads to Low-Rank Features
Maksym Andriushchenko, Dara Bahri, Hossein Mobahi, Nicolas Flammarion
Sharpness-Aware Minimization Revisited: Weighted Sharpness as a Regularization Term
Yun Yue, Jiadi Jiang, Zhiling Ye, Ning Gao, Yongchao Liu, Ke Zhang
How to escape sharp minima with random perturbations
Kwangjun Ahn, Ali Jadbabaie, Suvrit Sra