Papers for

machine learning optimization teams

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Neural network size must grow to find complex patterns early in learning

Representation Complexity Along Learning Trajectories: Trading Width for Precision and Time

Abstract: What computational role does overparameterisation play during learning? Information-plane analyses suggest that neural networks can traverse representations substantially more complex than those required by their final predictors. We explore when this transient complexity serves a computational purpose by comparing the representation complexity needed for prediction with that needed to make an unknown target statistically visible through gradient feedback. We introduce an oracle model of neural network learning and use Kolmogorov \(n\)-widths to quantify effective representation complexity. Our main technical result transfers complexity bounds from the forward representation to the corresponding gradient-query features. Combined with statistical-query lower bounds, this gives a task-agnostic lower bound on the initial effective width required for target-dependent gradient information to become detectable. For \(k\)-sparse parity in \(D\) dimensions, a known target admits an \(O(k)\)-width representation, while the general bound already implies a polynomial separation between representation and discovery. We then exploit parity's invariance under coordinatewise sign reflections to identify a sharper unsigned effective width, which removes sign distinctions invisible to the parity information channel. At a transfer-compatible measurement resolution, its critical scale is \(\widetildeΘ_k\left( D^{k-1}τ^2 \right)\), up to logarithmic factors. Thus, the representation resources required to discover a predictor can substantially exceed those required to represent it once learned. Interpreting precision through accumulation over fresh samples yields a corresponding width-time frontier. Experiments support the predicted dimension dependence and examine how optimisation and generalisation depart from this idealised frontier.

Wed 28 OctMachine LearningArtificial Intelligence
The gist
This paper studies why big neural networks use more complicated internal steps when learning than needed for their final task. The authors show that finding complex patterns can require much larger network capacity than just representing them after learning. Using mathematical tools, they prove limits on how small these networks can be at the start before they can detect a signal through learning. Their experiments confirm these predictions and show how the tradeoff between network size, precision, and time affects learning.
Open → 2610.39027v1

Speech language models show collapse repair and drift during fine tuning

Collapse, Repair, and Drift: Dissecting Fine Tuning Dynamics in Speech Language Models

Abstract: Supervised fine tuning (SFT) is commonly monitored through training and evaluation loss, but loss does not fully characterize how model functionality evolves along the same trajectory. We study this gap through dialect adaptation of Qwen3-ASR, MiMo-Audio, and FireRed-LLM across learning rates from $10^{-5}$ to $10^{-3}$. Character error rate (CER) reveals pronounced transient collapse and recovery whose severity and temporal structure are only weakly reflected by loss. Bidirectional layer patching reveals strong cross checkpoint functional incompatibility during recoverable collapse, with model dependent depth profiles that reorganize during recovery. Pairwise gradient similarity alone does not reliably indicate optimization stability, whereas raw gradient spectral analysis provides a clearer structural distinction between stable and unstable optimization: successfully converged runs approach a broad and temporally stable spectral energy distribution, while failed runs remain spectrally concentrated or exhibit persistent fluctuations. Finally, AdamW update transfer bypasses pronounced early collapse, yet substantial Mandarin degradation still emerges later in training. These results motivate three separable SFT dynamics: collapse, repair, and drift, and show that transient collapse is not necessary for persistent source domain degradation.

Tue 13 OctSound
The gist
When speech language models are fine-tuned to understand different dialects better, their performance doesn't always improve steadily. The authors found that these models often go through phases where they temporarily get worse (collapse), then recover (repair), and later change slowly (drift). They discovered that traditional loss measurements don't fully capture these changes, but looking at error rates and model behavior over time reveals much more. This research helps explain how and why speech models change during fine-tuning and shows that early problems aren't always needed for lasting issues to appear.
Open → 2610.18192v1