Learning Shrinks the Hard Tail: Training-Dependent Inference Scaling in a Solvable Linear Model
This provides insights into compute allocation strategies for machine learning practitioners, though it is incremental as it builds on existing scaling law frameworks.
The paper tackles the problem of understanding how training affects inference scaling in neural networks by analyzing a solvable linear model with latent instance difficulty, showing that learning reduces the heavy tail of error distributions, leading to a training-dependent power-law decay in failure rates that saturates at an intrinsic limit.
We analyze neural scaling laws in a solvable model of last-layer fine-tuning where targets have intrinsic, instance-heterogeneous difficulty. In our Latent Instance Difficulty (LID) model, each input's target variance is governed by a latent ``precision'' drawn from a heavy-tailed distribution. While generalization loss recovers standard scaling laws, our main contribution connects this to inference. The pass@$k$ failure rate exhibits a power-law decay, $k^{-β_\text{eff}}$, but the observed exponent $β_\text{eff}$ is training-dependent. It grows with sample size $N$ before saturating at an intrinsic limit $β$ set by the difficulty distribution's tail. This coupling reveals that learning shrinks the ``hard tail'' of the error distribution: improvements in the model's generalization error steepen the pass@$k$ curve until irreducible target variance dominates. The LID model yields testable, closed-form predictions for this behavior, including a compute-allocation rule that favors training before saturation and inference attempts after. We validate these predictions in simulations and in two real-data proxies: CIFAR-10H (human-label variance) and a maths teacher-student distillation task.