Why Self-Training Helps and Hurts: Denoising vs. Signal Forgetting
Abstract
In high-dimensional linear regression, iterative self-training produces a U-shaped risk curve from competing signal forgetting and denoising, with iteration-dependent spectral filtering and a consistent cross-validation criterion for early stopping.
Iterative self-training (self-distillation) repeatedly refits a model on pseudo-labels generated by its own predictions. We study this procedure in overparameterized linear regression: an initial estimator is trained on noisy labels, and each subsequent iterate is trained on fresh covariates with noiseless pseudo-labels from the previous model. In the high-dimensional regime, we derive deterministic-equivalent recursions for the prediction risk and effective noise across iterations, and prove that the empirical quantities concentrate sharply around these limits. The recursion separates two competing forces: a systematic component that grows with iteration due to progressive signal forgetting, and a stochastic component that decays due to denoising via repeated data-dependent projections. Their interaction yields a U-shaped test-risk curve and an optimal early-stopping time. In spiked covariance models, iteration further acts as an iteration-dependent spectral filter that preserves strong eigendirections while suppressing weaker ones, inducing an implicit form of soft feature selection distinct from ridge regression. Finally, we propose an iterated generalized cross-validation criterion and prove its uniform consistency for estimating the risk along the self-training trajectory, enabling fully data-driven selection of the stopping time and regularization. Experiments on synthetic covariances validate the theory and illustrate the predicted denoising-forgetting trade-off.
Get this paper in your agent:
hf papers read 2602.14029 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 3
evalstate/cifar10-selftrain-degraded-iterate
Datasets citing this paper 1
evalstate/synthetic-selftrain-denoising-forgetting
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper