arXiv Machine Learning

Bilateral Sharpness-Aware Minimization for Flatter Minima

The paper introduces Bilateral Sharpness-Aware Minimization (BSAM), a new optimization method that combines the traditional Max-Sharpness (MaxS) metric with a newly defined Min-Sharpness (MinS) metric. By using the difference between the training loss and the minimum loss in a neighborhood around the current weights, BSAM creates a more comprehensive Flatness Indicator (FI) that guides the search toward flatter minima. Experiments show that BSAM improves generalization and robustness over vanilla SAM in tasks such as classification, transfer learning, human pose estimation, and network quantization.

arXiv Machine Learning
Oct 9

Do Flatter Minima Drive Better Generalization? An Algorithmic Separation in Grokking

The paper investigates whether flatter loss landscapes drive better generalization in the context of Grokking. It shows that sharpness‑aware minimization (SAM) alone does not reliably induce the transition to generalization, but when combined with weight decay, SAM can accelerate this transition up to four times faster. The authors provide a theoretical analysis using a minimal two‑layer ReLU model, demonstrating that flatness alone cannot distinguish memorizing from generalizing solutions, while weight decay favors generalization and a local stability window explains SAM’s effect.

By Mohnish Harwani
arXiv Machine Learning
Oct 9

Explaining the Saliency Map Sparsity of Adversarially-Trained Neural Networks

The paper investigates why saliency maps of adversarially‑trained neural networks appear sparse. It provides a theoretical explanation for two‑layer ReLU networks, showing that as data size and neuron count grow, minimizers converge to a Bayes classifier with minimal gradient and Barron norm. The sparsity arises because α−−‑training with λ−1 attacks yields anisotropic gradient norms that favor axis‑aligned, sparse gradients, a claim supported by experimental evaluation of gradient α‑norms and sparsity thresholds.

By Yannick Lunk, Atell Yehor Krasnopolsky, Damien Garreau, Leon Bungert
arXiv Machine Learning
Jun 19

Fisher-Geometric Sharpness and the Implicit Bias of SGD toward Flat Minima

arXiv:2606. 20469v1 Announce Type: new Abstract: A widely held intuition in deep learning is that stochastic gradient descent (SGD) implicitly favors flat minima and that flat minima generalize better, but standard Euclidean measures of flatness such as the trace or maximum eigenvalue of the loss Hessian are not invariant under reparametrizations that preserve the network function, which undermines the theoretical foundations of this narrative.

By Md Sakir Ahmed, Kumaresh Sarmah, Hemen Dutta
arXiv Machine Learning
Oct 16

Pairwise AUC Optimization Needs Corrective Power: A Unified View

The paper investigates why direct pairwise AUC surrogate training, though aligned with the ranking metric, can fail to provide corrective updates during stochastic training. It introduces the concept of Corrective Power, showing that in certain regions severe ranking errors persist without generating strong gradient signals, a phenomenon linked to degeneracies in the U‑statistics of pairwise gradients. The authors explain empirical observations—such as the effectiveness of squared‑hinge loss, the benefit of cross‑entropy warm‑up, and the advantage of parameter‑efficient fine‑tuning—through this theoretical lens, and validate their claims with experiments on eight computer‑vision and eight NLP tasks using various models.

By Jia Chen, Zhiyong Yang, Shilong Bao, Qianqian Xu, Qingming Huang
Hugging Face Trending Papers
Jun 24

Pre-Warm: Input-Conditioned Weight Initialization for Convolutional Neural Networks

We introduce Pre-Warm, a simple yet effective zero-training-cost method for data-conditioned initialization of the first convolutional layer. Before the first forward pass, Pre-Warm extracts mean-centered local patches from a single training batch, clusters them with MiniBatchKMeans, applies inverse Manhattan spatial weighting, and uses the resulting centroids to initialize half of the first-layer filters (the remainder retain Kaiming initialization).