arXiv Machine Learning

Broken scale symmetries in undercomplete linear autoencoders

The paper investigates how finite‑stepsize stochastic gradient descent (SGD) breaks scale symmetry in undercomplete linear autoencoders. It shows that SGD induces a directed scale drift that favors large decoder weights on the PCA solution manifold, a process that is slow and analytically tractable but eventually halted by a stability boundary. The resulting solutions exhibit sharper loss Hessians compared to a balanced baseline, illustrating how loss geometry can transform residual gradient noise into directed motion along equivalent solution manifolds.

arXiv Machine Learning
6d ago

Symmetry Breaking at the Edge of Stability via Central Flow Analysis

The paper investigates how continuous symmetries in neural networks, which give rise to conserved quantities under gradient flow, behave when training with finite‑step gradient descent at the edge of stability (EOS). By applying central flow dynamics, the authors show that curvature corrections cause a finite‑dimensional drift driven by oscillation covariance along critical parameter directions. They analyze this effect in a scalar two‑layer model, two‑layer matrix factorisation, and a two‑layer ReLU network, demonstrating which conservation laws persist or break under EOS conditions.

By Naima Elosegui Borras, Georgios Arvanitidis
arXiv Machine Learning
Oct 3

Learning the identity: a case study of how SGD selects among functional decompositions

The paper investigates how stochastic gradient descent (SGD) selects specific functional decompositions when training a deep linear residual network to learn the identity function. Although many weight configurations minimize the population loss, SGD consistently prefers particular solutions, especially under anisotropic label noise or different parametrizations. The authors explain this bias using an entropic loss term that penalizes the expected squared norm of the minibatch gradient, analytically characterizing its minimizers and showing that trained networks align with these predictions.

By Andy Arditi, Weian Xie, David Bau, Liu Ziyin
arXiv Machine Learning
Sep 10

When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay

The paper derives an exact discrete‑time law that captures how learning‑rate schedules and weight decay interact in scale‑invariant neural networks, showing that a single scalar quantity governs the effective step size. It demonstrates that the balance point between contraction and expansion is intrinsically unstable, leading to recurrent dynamics when using constant learning rates with weight decay. The authors extend this analysis to various optimizers and datasets, confirming the law’s precision and showing that performance peaks sharply at the predicted boundary.

By Hasan Amin, Wei-Kai Chang, Rajiv Khanna
arXiv AI
Sep 3

Percolation Dynamics in Optimization : Variance Cascades and Discrete Scale Invariance

The paper models the dynamics of Stochastic Gradient Descent (SGD) as a percolation process, showing that architectural symmetries cause subnetworks to merge in discrete blocks rather than sequentially. These structural transitions produce variance spikes in a macroscopic order parameter, analogous to physical phase transitions. The authors also demonstrate that this trapping mechanism and its scaling cascade apply to Adam and AdamW under a heavy‑tailed noise model.

By Sai Niranjan Ramachandran, Suvrit Sra
Hugging Face Trending Papers
Sep 8

When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay

The paper investigates how normalization makes neural networks scale‑invariant, creating a feedback loop between learning‑rate schedules and weight decay that controls the effective step size of the optimizer. It derives an exact discrete‑time law showing that a single scalar quantity captures all schedule and decay effects, with norm growth providing a self‑quenching counter‑force that defines a sharp boundary between contraction‑ and expansion‑dominated regimes. Through exact analysis of a normalized regression model and experiments on MLPs, CNNs, GPT‑2, and various datasets, the authors demonstrate that constant learning rates with weight decay are intrinsically unstable, leading to recurrent dynamics, and that adaptive optimizers exhibit weaker stabilization under normalization. "whyItMatters":"The study provides a precise, actionable rule for controlling training dynamics and schedule design in modern deep learning by isolating a single governing quantity for scale‑invariant optimization."

arXiv Machine Learning
Oct 5

Stimulus symmetries can confound representational similarity analyses

The paper investigates how symmetries in network inputs can mislead representational similarity analyses (RSAs). It shows that functionally equivalent representations, arising from stimulus symmetries, can produce distinct RSMs that depict different representational geometries—from disentangled to maximally mixed. The authors demonstrate that training dynamics such as stochastic gradient descent or energetic regularization can yield sparse, drifting codes, causing RSMs to drift over time, and that these effects also appear in networks trained on image data with latent symmetries.

By Farhad Pashakhanloo, Jacob A. Zavatone-Veth