English

A Mechanism Study of Delayed Loss Spikes in Batch-Normalized Linear Models

Machine Learning 2026-04-21 v1 Machine Learning Optimization and Control

Abstract

Delayed loss spikes have been reported in neural-network training, but existing theory mainly explains earlier non-monotone behavior caused by overly large fixed learning rates. We study one stylized hypothesis: normalization can postpone instability by gradually increasing the effective learning rate during otherwise stable descent. To test this hypothesis at theorem level, we analyze batch-normalized linear models. Our flagship result concerns whitened square-loss linear regression, where we derive explicit no-rising-edge and delayed-onset conditions, bound the waiting time to directional onset, and show that the rising edge self-stabilizes within finitely many iterations. Combined with a square-loss decomposition, this yields a concrete delayed-spike mechanism in the whitened regime. For logistic regression, under highly restrictive active-margin assumptions, we prove only a supporting finite-horizon directional precursor in a knife-edge regime, with an optional appendix-only loss lower bound under an extra non-degeneracy condition. The paper should therefore be read as a stylized mechanism study rather than a general explanation of neural-network loss spikes. Within that scope, the results isolate one concrete delayed-instability pathway induced by batch normalization.

Keywords

Cite

@article{arxiv.2604.16809,
  title  = {A Mechanism Study of Delayed Loss Spikes in Batch-Normalized Linear Models},
  author = {Peifeng Gao and Wenyi Fang and Yang Zheng and Difan Zou},
  journal= {arXiv preprint arXiv:2604.16809},
  year   = {2026}
}
R2 v1 2026-07-01T12:15:41.913Z