Functional Scaling Laws in Kernel Regression: Loss Dynamics and Learning Rate Schedules
Not on the map. Why?
Abstract (arXiv)
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 loss dynamics obey similar laws and, crucially, how the 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 intrinsic-time viewpoint, which captures the training progress more faithfully than iteration count. We then establish a 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 -- constant, exponential decay, and warmup-stable-decay (WSD) -- and derive explicit scaling relations in both data- and compute-limited regimes. These comparisons explain key empirical phenomena: (i) higher-capacity models are more data- and compute-efficient; (ii) learning-rate decay improves training efficiency; and (iii) WSD-type schedules outperform pure decay. Finally, experiments on LLMs ranging from 0.1B to 1B parameters demonstrate the practical relevance of FSL as a surrogate model for fitting and predicting loss trajectories in large-scale pre-training.
Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). First posted on arXiv 23 Sep 2025.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2025: 1; 2026: 16.
17 papers.
Sort:
By area:
- 2026Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averagingcited by 10
- 2026Optimal Learning Rate Schedules under Functional Scaling Laws: Power Decay and Warmup-Stable-Decaycited by 8
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Lawscited by 6
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretrainingcited by 3
- 2026Configuration-to-Performance Scaling Law with Neural Ansatzcited by 2
- 2026Scaling Domain Data Repetition in LLM Pretrainingcited by 2
- 2026Effective Frontiers: A Unification of Neural Scaling Lawscited by 1
- 2026Asymmetric Scaling Laws from Sparse Featurescited by 1
- 2026Explaining Data Mixing Scaling Lawscited by 1
- 2026Smooth Scaling Laws Hide Stepwise Token Learningcited by 1
- 2025When Does Learning Renormalize? Sufficient Conditions for Power Law Spectral Dynamicscited by 0
- 2026How Should LLMs Consume High-Quality Data? Optimal Data Scheduling via Quality-Aware Functional Scaling Lawscited by 0
5 more
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretrainingcited by 0
- 2026Learning with Synthetic Data via SGD in High-Dimensional Linear Regressioncited by 0
- 2026Compute-Optimal Pretrain--Fine-tune in Ridge Gradient Descentcited by 0
- 2026Understanding the Weight Averaging Mechanism in LLM Training for Post-Training Quantizationcited by 0
- 2026Asymptotic Risk of Learning-Rate Schedules for Least-Squares Regression with Discretely Distributed Featurescited by 0
- 2026Asymptotic Risk of Learning-Rate Schedules for Least-Squares Regression with Discretely Distributed Features12 Oct 2026
- 2026Understanding the Weight Averaging Mechanism in LLM Training for Post-Training Quantization4 Oct 2026
- 2026Compute-Optimal Pretrain--Fine-tune in Ridge Gradient Descent14 Sep 2026
- 2026Learning with Synthetic Data via SGD in High-Dimensional Linear Regression9 Sep 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining25 Aug 2026
- 2026Scaling Domain Data Repetition in LLM Pretraining14 Aug 2026
- 2026Smooth Scaling Laws Hide Stepwise Token Learning29 Jun 2026
- 2026Explaining Data Mixing Scaling Laws6 Jun 2026
- 2026How Should LLMs Consume High-Quality Data? Optimal Data Scheduling via Quality-Aware Functional Scaling Laws25 May 2026
- 2026Asymmetric Scaling Laws from Sparse Features22 May 2026
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Laws15 Feb 2026
5 more
- 2026Configuration-to-Performance Scaling Law with Neural Ansatz10 Feb 2026
- 2026Optimal Learning Rate Schedules under Functional Scaling Laws: Power Decay and Warmup-Stable-Decay6 Feb 2026
- 2026Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averaging3 Feb 2026
- 2026Effective Frontiers: A Unification of Neural Scaling Laws1 Feb 2026
- 2025When Does Learning Renormalize? Sufficient Conditions for Power Law Spectral Dynamics20 Dec 2025
- 2026Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averagingcited by 10
- 2026Optimal Learning Rate Schedules under Functional Scaling Laws: Power Decay and Warmup-Stable-Decaycited by 8
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Lawscited by 6
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretrainingcited by 3
- 2026Configuration-to-Performance Scaling Law with Neural Ansatzcited by 2
- 2026Scaling Domain Data Repetition in LLM Pretrainingcited by 2
- 2026Effective Frontiers: A Unification of Neural Scaling Lawscited by 1
- 2026Asymmetric Scaling Laws from Sparse Featurescited by 1
- 2026Explaining Data Mixing Scaling Lawscited by 1
- 2026Smooth Scaling Laws Hide Stepwise Token Learningcited by 1
- 2025When Does Learning Renormalize? Sufficient Conditions for Power Law Spectral Dynamicscited by 0
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretrainingcited by 0
12 most cited of 15 All 15 (JSON)
- 2026Asymptotic Risk of Learning-Rate Schedules for Least-Squares Regression with Discretely Distributed Features12 Oct 2026
- 2026Understanding the Weight Averaging Mechanism in LLM Training for Post-Training Quantization4 Oct 2026
- 2026Compute-Optimal Pretrain--Fine-tune in Ridge Gradient Descent14 Sep 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining25 Aug 2026
- 2026Scaling Domain Data Repetition in LLM Pretraining14 Aug 2026
- 2026Smooth Scaling Laws Hide Stepwise Token Learning29 Jun 2026
- 2026Explaining Data Mixing Scaling Laws6 Jun 2026
- 2026Asymmetric Scaling Laws from Sparse Features22 May 2026
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Laws15 Feb 2026
- 2026Configuration-to-Performance Scaling Law with Neural Ansatz10 Feb 2026
- 2026Optimal Learning Rate Schedules under Functional Scaling Laws: Power Decay and Warmup-Stable-Decay6 Feb 2026
12 newest of 15 All 15 (JSON)