Functional Scaling Laws in Kernel Regression: Loss Dynamics and Learning Rate Schedules

Binghui Li (Peking University) · Fengling Chen (Peking University) · Zixun Huang (Peking University) · Lean Wang (Peking University) · Lei Wu (Peking University)
compute-limited regimesconvolutional functionaldata-limited regimesempirical phenomenafunctional scaling lawhigher-capacity modelsintrinsic-timelarge-scale pre-traininglearning rate schedulelearning-rate decayloss dynamicspower-law kernel regressionscaling lawsstochastic gradient descentwarmup-stable-decay

Scaling laws have emerged as a unifying lens for understanding and guiding the training of large language models (LLMs). However, existing studies predominantly focus on the final-step loss, leaving open whether the entire $\textit{loss dynamics}$ obey similar laws and, crucially, how the $\textit{learning rate schedule}$ (LRS) shapes them. We address these gaps in a controlled theoretical setting by analyzing stochastic gradient descent (SGD) on a power-law kernel regression model. The key insight is a novel $\textbf{intrinsic-time}$ viewpoint, which captures the training progress more faithfully than iteration count. We then establish a $\textbf{Functional Scaling Law (FSL)}$ that captures the full loss trajectory under arbitrary LRSs, with the schedule’s influence entering through a simple convolutional functional. We further instantiate the theory for three representative LRSs