acceptodds
Under review as a conference paper at ICLR 2027

Curvature Controlled Mixing on the Interpolation Manifold: Sharp Cutoffs in Late-Phase Training Dynamics

Abstract

After a network interpolates its training data, weight decay drives a slow norm-minimising flow on the interpolation manifold. We study the stochastic version of this late phase and show that its approach to equilibrium is not gradual but abrupt. We analyse the effective diffusion on the interpolation manifold , whose invariant measure carries the Fixman correction produced by integrating out normal fluctuations. Our main result is a lower bound on the Bakry-Emery Ricci curvature of this measure, with for weight decay and noise scale , in which every correction — the intrinsic Ricci curvature via the Gauss equation, the extrinsic shape-operator coupling, and the Hessian of the term — is controlled by quantities measurable on a trained network: the model Hessian norm, its third derivative, the tangential Hessian rank, and the spectrum of the empirical NTK Gram matrix. We also record what the hypotheses force: strong convexity of the log-density makes complete, noncompact and contractible, so the theorem describes mixing within a single well-conditioned basin rather than across the permutation-symmetric zero-loss set. Positive curvature has two consequences. Existing results on cutoff for non-negatively curved diffusions then give a total-variation cutoff, and we locate it: with window , where is the initial displacement and the equilibrium correlation length. The clock is set by weight decay alone, independent of noise temperature at leading order. We identify the equilibrium as the minimum-norm interpolator measure, concentrated to in parameter norm, recovering deterministic norm-minimisation results as its zero-temperature mode. The resulting prediction — delay logarithmic in the ratio of initial displacement to equilibrium fluctuation scale — is distinct from existing norm-ratio delay laws. In simulations of constrained diffusions where every assumption holds by construction, the predicted location holds with regression slope () across independent variation of dimension, weight decay, temperature and initial displacement, and the total-variation profiles collapse under the predicted rescaling.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.