arXiv · 2510.16074
Early-stopping for Transformer model training
Abstract
This work, based on Random Matrix Theory (RMT), introduces a novel early-stopping strategy for Transformer training dynamics. Utilizing the Power Law (PL) fit to tansformer attention matrices as a probe, we demarcate training into three stages: structural exploration, heavy-tailed structure stabilization, and convergence saturation. Empirically, we observe that the spectral density of the shallow self-attention matrix $V$ consistently evolves into a heavy-tailed distribution. Crucially, we propose two consistent and validation-set-free criteria: a quantitative metric for heavy-tailed dynamics and a novel spectral signature indicative of convergence. The strong alignment between these criteria highlights the utility of RMT for monitoring and diagnosing the progression of Transformer model training.
Explore related subjects
Keep this discovery
Jing He, Hua Jiang, Cheng Li, Siqian Xin, Shuzhen Yang. 2025-10-17. Early-stopping for Transformer model training. https://arxiv.org/abs/2510.16074
Cite the original work for its findings. Save a collection to share your selection of sources.