On the Mechanisms of Weak-to-Strong Generalization: A Theoretical Perspective

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Moniri, Behrad, Hassani, Hamed
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910966650765312
author Moniri, Behrad
Hassani, Hamed
author_facet Moniri, Behrad
Hassani, Hamed
contents Weak-to-strong generalization, where a student model trained on imperfect labels generated by a weaker teacher nonetheless surpasses that teacher, has been widely observed but the mechanisms that enable it have remained poorly understood. In this paper, through a theoretical analysis of simple models, we uncover three core mechanisms that can drive this phenomenon. First, by analyzing ridge regression, we study the interplay between the teacher and student regularization and prove that a student can compensate for a teacher's under-regularization and achieve lower test error. We also analyze the role of the parameterization regime of the models. Second, by analyzing weighted ridge regression, we show that a student model with a regularization structure more aligned to the target, can outperform its teacher. Third, in a nonlinear multi-index setting, we demonstrate that a student can learn easy, task-specific features from the teacher while leveraging its own broader pre-training to learn hard-to-learn features that the teacher cannot capture.
format Preprint
id arxiv_https___arxiv_org_abs_2505_18346
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle On the Mechanisms of Weak-to-Strong Generalization: A Theoretical Perspective
Moniri, Behrad
Hassani, Hamed
Machine Learning
Weak-to-strong generalization, where a student model trained on imperfect labels generated by a weaker teacher nonetheless surpasses that teacher, has been widely observed but the mechanisms that enable it have remained poorly understood. In this paper, through a theoretical analysis of simple models, we uncover three core mechanisms that can drive this phenomenon. First, by analyzing ridge regression, we study the interplay between the teacher and student regularization and prove that a student can compensate for a teacher's under-regularization and achieve lower test error. We also analyze the role of the parameterization regime of the models. Second, by analyzing weighted ridge regression, we show that a student model with a regularization structure more aligned to the target, can outperform its teacher. Third, in a nonlinear multi-index setting, we demonstrate that a student can learn easy, task-specific features from the teacher while leveraging its own broader pre-training to learn hard-to-learn features that the teacher cannot capture.
title On the Mechanisms of Weak-to-Strong Generalization: A Theoretical Perspective
topic Machine Learning
url https://arxiv.org/abs/2505.18346