Saved in:
Bibliographic Details
Main Authors: Kricheli, Joshua Shay, Reid, Alexander Lawrence, Sarkar, Soumajyoti, Gandikota, Venkata, Shakarian, Paulo
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2605.08541
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909038249246720
author Kricheli, Joshua Shay
Reid, Alexander Lawrence
Sarkar, Soumajyoti
Gandikota, Venkata
Shakarian, Paulo
author_facet Kricheli, Joshua Shay
Reid, Alexander Lawrence
Sarkar, Soumajyoti
Gandikota, Venkata
Shakarian, Paulo
contents Neural scaling laws approximate a language model's loss as a power-law function of parameter count $N$ and token count $D$. Following Chinchilla-style compute-optimal training, many studies fit scaling laws from runs performed under a fixed tokens-per-parameter (TPP) ratio $k$ and set $D = kN$. We show that this collinear design, combined with the empirically common near-equality of the exponents governing $N$ and $D$, induces an inherent ill-conditioning in the Gauss-Newton least-squares problem: the condition number of the design grows as the inverse square of the gap between the $N$ and $D$-exponents. The scale coefficients become practically unidentifiable, with confidence intervals inflating by an order of magnitude or more, yielding a ``sloppy'' model whose extrapolations degrade sharply off the training ray. We prove this for four scaling-law formalisms and derive a closed-form TPP-diversity threshold that is necessary and sufficient for well-conditioned estimation. Empirically, non-collinear designs outperform collinear ones on held-out splits with a 97.3\% win rate across four laws, five corpora, multiple floating point precision modes. We further show the degeneracy is rooted in Jacobian geometry and is not an artifact of the loss function: any smooth estimation objective whose curvature involves the Jacobian inherits the same ill-conditioning.
format Preprint
id arxiv_https___arxiv_org_abs_2605_08541
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Tokens-per-Parameter Coverage Is Critical for Robust LLM Scaling Law Extrapolation
Kricheli, Joshua Shay
Reid, Alexander Lawrence
Sarkar, Soumajyoti
Gandikota, Venkata
Shakarian, Paulo
Machine Learning
Neural scaling laws approximate a language model's loss as a power-law function of parameter count $N$ and token count $D$. Following Chinchilla-style compute-optimal training, many studies fit scaling laws from runs performed under a fixed tokens-per-parameter (TPP) ratio $k$ and set $D = kN$. We show that this collinear design, combined with the empirically common near-equality of the exponents governing $N$ and $D$, induces an inherent ill-conditioning in the Gauss-Newton least-squares problem: the condition number of the design grows as the inverse square of the gap between the $N$ and $D$-exponents. The scale coefficients become practically unidentifiable, with confidence intervals inflating by an order of magnitude or more, yielding a ``sloppy'' model whose extrapolations degrade sharply off the training ray. We prove this for four scaling-law formalisms and derive a closed-form TPP-diversity threshold that is necessary and sufficient for well-conditioned estimation. Empirically, non-collinear designs outperform collinear ones on held-out splits with a 97.3\% win rate across four laws, five corpora, multiple floating point precision modes. We further show the degeneracy is rooted in Jacobian geometry and is not an artifact of the loss function: any smooth estimation objective whose curvature involves the Jacobian inherits the same ill-conditioning.
title Tokens-per-Parameter Coverage Is Critical for Robust LLM Scaling Law Extrapolation
topic Machine Learning
url https://arxiv.org/abs/2605.08541