English
Related papers

Related papers: Unveiling m-Sharpness Through the Structure of Sto…

200 papers

Improving the generalization ability of modern deep neural networks (DNNs) is a fundamental challenge in machine learning. Two branches of methods have been proposed to seek flat minima and improve generalization: one led by sharpness-aware…

Machine Learning · Computer Science 2024-04-02 Tao Li , Qinghua Tao , Weihao Yan , Zehao Lei , Yingwen Wu , Kun Fang , Mingzhen He , Xiaolin Huang

Sharpness-Aware Minimization (SAM) has proven highly effective in improving model generalization in machine learning tasks. However, SAM employs a fixed hyperparameter associated with the regularization to characterize the sharpness of the…

Machine Learning · Computer Science 2024-12-24 Jinping Zou , Xiaoge Deng , Tao Sun

Deep neural networks have been increasingly used in safety-critical applications such as medical diagnosis and autonomous driving. However, many studies suggest that they are prone to being poorly calibrated and have a propensity for…

Machine Learning · Computer Science 2025-06-02 Chengli Tan , Yubo Zhou , Haishan Ye , Guang Dai , Junmin Liu , Zengjie Song , Jiangshe Zhang , Zixiang Zhao , Yunda Hao , Yong Xu

Sharpness-aware minimization (SAM) has well documented merits in enhancing generalization of deep neural networks, even without sizable data augmentation. Embracing the geometry of the loss function, where neighborhoods of 'flat minima'…

Machine Learning · Computer Science 2023-12-25 Bingcong Li , Georgios B. Giannakis

Sharpness-Aware Minimization (SAM) improves generalization by minimizing the worst-case loss within a fixed parameter-space radius neighborhood. SAM and its variants mainly rely on a first-order linearized surrogate, while flat minima are…

Machine Learning · Computer Science 2026-05-12 Jinping Wang , Qinhan Liu , Zhiwu Xie , Zhiqiang Gao

The recently proposed optimization algorithm for deep neural networks Sharpness Aware Minimization (SAM) suggests perturbing parameters before gradient calculation by a gradient ascent step to guide the optimization into parameter space…

Machine Learning · Computer Science 2025-10-03 Marlon Becker , Frederick Altrock , Benjamin Risse

Sharpness-Aware Minimization (SAM) has attracted considerable attention for its effectiveness in improving generalization in deep neural network training by explicitly minimizing sharpness in the loss landscape. Its success, however, relies…

Machine Learning · Computer Science 2025-06-16 Sungbin Shin , Dongyeop Lee , Maksym Andriushchenko , Namhoon Lee

Sharpness-Aware Minimization (SAM) has recently emerged as an effective technique for improving DNN robustness to input variations. However, its interplay with the compactness requirements of on-device DNN deployments remains less explored.…

Computer Vision and Pattern Recognition · Computer Science 2026-01-29 Jialuo He , Huangxun Chen

Sharpness-Aware Minimization (SAM) and adaptive sharpness-aware minimization (ASAM) aim to improve the model generalization. And in this project, we proposed three experiments to valid their generalization from the sharpness aware…

Machine Learning · Computer Science 2022-08-16 Jozef Marus Coldenhoff , Chengkun Li , Yurui Zhu

Sharpness-aware minimization (SAM) aims to improve the generalisation of gradient-based learning by seeking out flat minima. In this work, we establish connections between SAM and Mean-Field Variational Inference (MFVI) of neural network…

Machine Learning · Statistics 2022-10-20 Szilvia Ujváry , Zsigmond Telek , Anna Kerekes , Anna Mészáros , Ferenc Huszár

The recently proposed Sharpness-Aware Minimization (SAM) improves generalization by minimizing a \textit{perturbed loss} defined as the maximum loss within a neighborhood in the parameter space. However, we show that both sharp and flat…

Machine Learning · Computer Science 2022-03-22 Juntang Zhuang , Boqing Gong , Liangzhe Yuan , Yin Cui , Hartwig Adam , Nicha Dvornek , Sekhar Tatikonda , James Duncan , Ting Liu

Sharpness-aware minimization (SAM) has been shown to improve the generalization of neural networks. However, each SAM update requires \emph{sequentially} computing two gradients, effectively doubling the per-iteration cost compared to base…

Machine Learning · Computer Science 2024-10-15 Wanyun Xie , Thomas Pethick , Volkan Cevher

Curvature regularization techniques like Sharpness Aware Minimization (SAM) have shown great promise in improving generalization on vision tasks. However, we find that SAM performs poorly in domains like natural language processing (NLP),…

Machine Learning · Computer Science 2025-02-05 Sidak Pal Singh , Hossein Mobahi , Atish Agarwala , Yann Dauphin

Sharpness-Aware Minimization (SAM) has attracted significant attention for its effectiveness in improving generalization across various tasks. However, its underlying principles remain poorly understood. In this work, we analyze SAM's…

Machine Learning · Computer Science 2025-01-23 Haocheng Luo , Tuan Truong , Tung Pham , Mehrtash Harandi , Dinh Phung , Trung Le

While Sharpness-Aware Minimization (SAM) improves generalization in deep neural networks by minimizing both loss and sharpness, it suffers from inefficiency in distributed large-batch training. We present Landscape-Smoothed SAM (LSAM), a…

Machine Learning · Computer Science 2025-09-04 Yunfei Teng , Sixin Zhang

Sharpness Aware Minimization (SAM) enhances performance across various neural architectures and datasets. As models are continually scaled up to improve performance, a rigorous understanding of SAM's scaling behaviour is paramount. To this…

Machine Learning · Computer Science 2025-02-12 Moritz Haas , Jin Xu , Volkan Cevher , Leena Chennuru Vankadara

By driving models to converge to flat minima, sharpness-aware learning algorithms (such as SAM) have shown the power to achieve state-of-the-art performances. However, these algorithms will generally incur one extra forward-backward…

Machine Learning · Computer Science 2023-04-11 Yang Zhao , Hao Zhang , Xiuyuan Hu

Sharpness-Aware Minimization (SAM) improves model generalization but doubles the computational cost of Stochastic Gradient Descent (SGD) by requiring twice the gradient calculations per optimization step. To mitigate this, we propose…

Computer Vision and Pattern Recognition · Computer Science 2025-10-07 Jiaxin Deng , Junbiao Pang

Sharpness-Aware Minimization (SAM) has recently emerged as a robust technique for improving the accuracy of deep neural networks. However, SAM incurs a high computational cost in practice, requiring up to twice as much computation as…

Machine Learning · Computer Science 2022-10-25 Renkun Ni , Ping-yeh Chiang , Jonas Geiping , Micah Goldblum , Andrew Gordon Wilson , Tom Goldstein

Sharpness-Aware Minimization (SAM) is an optimization technique designed to improve generalization by favoring flatter loss minima. To achieve this, SAM optimizes a modified objective that penalizes sharpness, using computationally…

Machine Learning · Computer Science 2024-11-05 Nalin Tiwary , Siddarth Aananth