English
Related papers

Related papers: Implicit Regularization of Sharpness-Aware Minimiz…

200 papers

Class imbalance is a common challenge in real-world recognition tasks, where the majority of classes have few samples, also known as tail classes. We address this challenge with the perspective of generalization and empirically find that…

Computer Vision and Pattern Recognition · Computer Science 2023-08-16 Yixuan Zhou , Yi Qu , Xing Xu , Hengtao Shen

Modern deep neural networks (DNNs) have achieved state-of-the-art performances but are typically over-parameterized. The over-parameterization may result in undesirably large generalization error in the absence of other customized training…

Machine Learning · Computer Science 2023-03-03 Jiawei Du , Daquan Zhou , Jiashi Feng , Vincent Y. F. Tan , Joey Tianyi Zhou

In multimodal learning, dominant modalities often overshadow others, limiting generalization. We propose Modality-Aware Sharpness-Aware Minimization (M-SAM), a model-agnostic framework that applies to many modalities and supports early and…

Computer Vision and Pattern Recognition · Computer Science 2025-10-30 Hossein R. Nowdeh , Jie Ji , Xiaolong Ma , Fatemeh Afghah

Despite attaining high empirical generalization, the sharpness of models trained with sharpness-aware minimization (SAM) do not always correlate with generalization error. Instead of viewing SAM as minimizing sharpness to improve…

Machine Learning · Computer Science 2024-06-12 Ankit Vani , Frederick Tung , Gabriel L. Oliveira , Hossein Sharifi-Noghabi

Sharpness-aware Minimization (SAM) improves generalization in large-scale model training by linking loss landscape geometry to generalization. However, challenges such as mislabeled noisy data and privacy concerns have emerged as…

Machine Learning · Computer Science 2025-07-08 Chenyang Ren , Yifan Jia , Huanyi Xie , Zhaobin Xu , Tianxing Wei , Liangyu Wang , Lijie Hu , Di Wang

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) was recently introduced as a regularization procedure for training deep neural networks. It simultaneously minimizes the fitness (or loss) function and the so-called fitness sharpness. The latter serves as…

Neural and Evolutionary Computing · Computer Science 2024-05-20 Illya Bakurov , Nathan Haut , Wolfgang Banzhaf

Sharpness-aware minimization (SAM) has received increasing attention in computer vision since it can effectively eliminate the sharp local minima from the training trajectory and mitigate generalization degradation. However, SAM requires…

Machine Learning · Computer Science 2024-06-21 Yili Wang , Kaixiong Zhou , Ninghao Liu , Ying Wang , Xin Wang

To fully uncover the great potential of deep neural networks (DNNs), various learning algorithms have been developed to improve the model's generalization ability. Recently, sharpness-aware minimization (SAM) establishes a generic scheme…

Computer Vision and Pattern Recognition · Computer Science 2022-11-22 Tao Li , Weihao Yan , Zehao Lei , Yingwen Wu , Kun Fang , Ming Yang , Xiaolin Huang

Sharpness-aware minimization (SAM) is a recently proposed training method that seeks to find flat minima in deep learning, resulting in state-of-the-art performance across various domains. Instead of minimizing the loss of the current…

Machine Learning · Computer Science 2023-01-18 Hoki Kim , Jinseong Park , Yujin Choi , Jaewook Lee

Sharpness-Aware Minimization (SAM) is an optimizer that takes a descent step based on the gradient at a perturbation $y_t = x_t + \rho \frac{\nabla f(x_t)}{\lVert \nabla f(x_t) \rVert}$ of the current point $x_t$. Existing studies prove…

Machine Learning · Computer Science 2023-10-30 Dongkuk Si , Chulhee Yun

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

Hybrid modeling, the combination of machine learning models and scientific mathematical models, enables flexible and robust data-driven prediction with partial interpretability. However, effectively the scientific models may be ignored in…

Machine Learning · Computer Science 2026-02-09 Naoya Takeishi

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) reports improving domain generalization by reducing the loss surface curvature in the parameter space. However, generalization during fine-tuning is often more dependent on the transferability of…

Machine Learning · Computer Science 2024-03-13 Tom Sherborne , Naomi Saphra , Pradeep Dasigi , Hao Peng

Fine-tuning large pretrained language models on a limited training corpus usually suffers from poor generalization. Prior works show that the recently-proposed sharpness-aware minimization (SAM) optimization method can improve the model…

Computation and Language · Computer Science 2022-10-12 Qihuang Zhong , Liang Ding , Li Shen , Peng Mi , Juhua Liu , Bo Du , Dacheng Tao

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

We consider Sharpness-Aware Minimization (SAM), a gradient-based optimization method for deep networks that has exhibited performance improvements on image and language prediction problems. We show that when SAM is applied with a convex…

Machine Learning · Computer Science 2023-04-12 Peter L. Bartlett , Philip M. Long , Olivier Bousquet

We study the implicit bias of Sharpness-Aware Minimization (SAM) when training $L$-layer linear diagonal networks on linearly separable binary classification. For linear models ($L=1$), both $\ell_\infty$- and $\ell_2$-SAM recover the…

Machine Learning · Computer Science 2026-05-19 Chaewon Moon , Dongkuk Si , Chulhee Yun

Sharpness-aware minimization (SAM) methods have gained increasing popularity by formulating the problem of minimizing both loss value and loss sharpness as a minimax objective. In this work, we increase the efficiency of the maximization…

Machine Learning · Computer Science 2024-05-17 Gonçalo Mordido , Pranshu Malviya , Aristide Baratin , Sarath Chandar