Asymptotic Training-Horizon Controller
Implementation & benchmark of arXiv:2607.23287 — Learning Asymptotics with Convergence-Rate Guarantees using Linear Least Squares
Source paper: Learning Asymptotics with Convergence-Rate Guarantees using Linear Least Squares arXiv:2607.23287 ⓘ · analyzed Aug 30, 2026
AI-generated research hypothesis, automatically tested. Not peer-reviewed.
Idea description
Model a checkpointed validation metric as a finite asymptotic expansion in known decay features, such as powers of training step, and estimate its limiting value using sliding least squares. Use a ridge-stabilized fit and require agreement across multiple windows before stopping, preventing the controller from reacting to transient non-asymptotic behavior.
Formulas
Mathematical statement
The paper studies sequences with an asymptotic form whose unknown coefficients are recovered by linear least squares over a sliding window. For a scalar metric y(t), use the adapted expansion y(t)=theta_0+sum_{j=1}^m theta_j phi_j(t)+epsilon(t), where t is the checkpoint index, theta_0 is the limiting metric, phi_j(t) are fixed decay features such as t^{-p_j}, and epsilon(t) is the unmodeled remainder. Given the window W_t={t-w+1,...,t}, sliding Tikhonov least squares is hat(theta)_t=argmin_theta sum_{i in W_t}(y_i-phi(i)^T theta)^2+lambda||D theta||_2^2. Here phi(i)=[1,phi_1(i),...,phi_m(i)]^T, D is a diagonal matrix that can leave the intercept unpenalized, and lambda>0 controls ill-conditioning. The fitted asymptotic limit is hat(y)_infinity=hat(theta)_{0,t}. The paper's rate example gives lim_{n to infinity}|r_n-alpha_3|sqrt(n)=2|alpha_3|, equivalently r_n-alpha_3=Theta(n^{-1/2}); in the neural adaptation, analogous residual improvement and decay-rate estimates are used only when several windows agree, because the paper explicitly identifies slow convergence and divergence as possible failure modes.
Implementation notes
Integrate this at the training loop's checkpoint and evaluation path, not inside the forward or backward pass. Every K optimizer steps, record a smoothed validation loss y_i, the checkpoint index t_i, and optionally training loss and gradient norm as separate series. Choose a small candidate basis, for example phi(t)=[1,t^{-1/2},t^{-1},t^{-3/2}], after rescaling t by the first checkpoint to avoid numerical magnitude problems. For each new checkpoint and each window size w in {8,16,32}, solve the ridge system (X^T X+lambda D^T D)theta=X^T y, with D=diag(0,1,1,1), lambda between 1e-4 and 1e-2, and compute y_infinity=theta[0]. Estimate expected improvement over H future checkpoints with theta^T(phi(t)-phi(t+H)). Compute a stability score from the standard deviation of y_infinity across window sizes and from the fit residual; do not stop unless both are below thresholds for three consecutive evaluations. Pseudocode: record y; for each w construct X and solve the ridge system; obtain y_infinity,w and delta_w; if median(delta_w)<epsilon and std(y_infinity,w)<tau, terminate training, otherwise continue. The mathematics supplies the estimator and extrapolated limit; epsilon, tau, basis exponents, and H must be calibrated empirically. First test on CIFAR-10 with a small ResNet-18 and on a 100M-parameter language-model fine-tuning run, comparing fixed-budget training and ordinary patience-based early stopping. Measure compute saved at equal final validation loss, premature-stop rate, and correlation between predicted and realized future improvement. A useful signal is at least 15% fewer optimizer steps with no statistically significant degradation in final validation loss; instability or large disagreement across windows should correctly disable stopping.
Verification
This idea has not been verified yet.
Verification happens in two stages: Stage 1 — a mechanism check on a toy system confirms the claimed mathematical phenomenon reproduces; Stage 2 — a benchmark implements the idea on a real (small) neural network task and compares it against a tuned baseline over 8 paired seeds with a permutation test.
Artifacts
Artifacts unavailable.