Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper investigates how learning-rate and batch-size schedules influence observed power-law learning curves in LLM pre-training by analyzing noisy online SGD with linear random features. It derives an exact Volterra equation revealing that power-law behavior depends on the cumulative weighted spectral mass of the representation, not individual eigenvalues, and that intrinsic time and noise injection rates dictate schedule effects. Experiments with nanoGPT confirm that matched $B/\eta$ paths are equivalent in intrinsic time, a forcing-memory surrogate predicts loss accurately, and LLM pre-training regimes align with the proposed $3+3(+2)$ framework.
LLM pre-training power laws are not fixed but dynamically shaped by learning-rate and batch-size schedules, with a novel forcing-memory surrogate accurately predicting loss across these regimes.
Power-law learning curves are often treated as fixed properties of a model and its data, although learning-rate and batch-size schedules can change the observed loss. We study this dependence in noisy online SGD with linear random features. Conditional on the representation, an exact Volterra equation separates two response components: a forcing term that propagates unresolved target error and a memory kernel that propagates stochastic-error injections. We prove that either component follows a power law if and only if its cumulative weighted spectral mass has the corresponding low-spectrum scaling; individual eigenvalues and target coefficients need not obey coordinatewise power laws. Under a joint schedule, intrinsic time $T_t=\sum_{s<t}η_s$ controls optimization progress, while $r_t=B_t/η_t$ controls noise injection. Their interaction yields sharp conditions under which a schedule preserves, changes, or destroys the clean power law, together with a memory ceiling on noise reduction. The power-law random-feature model realizes this mechanism in $3+3(+2)$ propagation regimes with phase-dependent compute rates. Controlled nanoGPT experiments show that (1) learning-rate and batch-size schedules with matched $B/η$ paths are nearly equivalent in intrinsic time, (2) a forcing-memory surrogate accurately predicts loss across schedules, and (3) its fitted exponents across real-world datasets identify the regime of LLMs in $3+3(+2)$ map.