From Linear to Nonlinear: Provable Weak-to-Strong Generalization through Feature Learning
Abstract
Weak-to-strong generalization refers to the phenomenon where a stronger model trained under supervision from a weaker one can outperform its teacher. While prior studies aim to explain this effect, most theoretical insights are limited to abstract frameworks or linear/random feature models. In this paper, we provide a formal analysis of weak-to-strong generalization from a linear CNN (weak) to a two-layer ReLU CNN (strong). We consider structured data composed of label-dependent signals of varying difficulty and label-independent noise, and analyze gradient descent dynamics when the strong model is trained on data labeled by the pretrained weak model. Our analysis identifies two regimes -- data-scarce and data-abundant -- based on the signal-to-noise characteristics of the dataset, and reveals distinct mechanisms of weak-to-strong generalization. In the data-scarce regime, generalization occurs via benign overfitting or fails via harmful overfitting, depending on the amount of data, and we characterize the transition boundary. In the data-abundant regime, generalization emerges in the early phase through label correction, but we observe that overtraining can subsequently degrade performance.
Cite
@article{arxiv.2510.24812,
title = {From Linear to Nonlinear: Provable Weak-to-Strong Generalization through Feature Learning},
author = {Junsoo Oh and Jerry Song and Chulhee Yun},
journal= {arXiv preprint arXiv:2510.24812},
year = {2025}
}
Comments
NeurIPS 2025 camera-ready version, 70 pages