Skip to yearly menu bar Skip to main content


Poster

Theoretical Analysis of Weak-to-Strong Generalization

Hunter Lang · David Sontag · Aravindan Vijayaraghavan

West Ballroom A-D #6607
[ ]
Fri 13 Dec 11 a.m. PST — 2 p.m. PST

Abstract:

Strong student models can learn from weaker teachers: when trained on the predictions of a weaker model, a strong pretrained student can learn to correct the weak model’s errors and generalize to examples where the teacher is not confident, even when these examples are excluded from training. This enables learning from cheap, incomplete, and possibly incorrect label information, such as coarse logical rules or the generations of a language model. We show that existing weak supervision theory results fail to account for both of these effects, which we call pseudolabel correction and coverage expansion, respectively. We give a new bound based on expansion properties of the data distribution and student hypothesis class that directly accounts for pseudolabel correction and coverage expansion. Our bound generalizes results from the co-training and self-training literature and captures the intuition that weak-to-strong generalization occurs when the mistakes of the weak model are hard for the strong model to fit without incurring additional error. We show that these expansion properties can be checked from finite data and give empirical evidence that they hold in practice.

Live content is unavailable. Log in and register to view live content