Research map / Label-efficient and robust learning / Recognition and long-tail
The Mechanism of Weak-to-Strong Generalization: Feature Elicitation from Latent Knowledge
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Theoretical analysis of weak-to-strong generalization in which a strong model trained by multi-step SGD on weak-teacher outputs learns task features (feature elicitation) while preserving pre-trained features of other tasks.
Problem
Whether multi-step SGD on a weak model's outputs can make a strong model learn task features (weak-to-strong generalization) while preserving diverse pre-trained capabilities, which prior theory left open by fixing representations or using restricted settings.
Approach
Shows that a nonlinear (exponential) transformation of the weak teacher's signal reduces its information exponent to 2, so multi-step SGD on two-layer networks elicits the target feature; task-weighted learning rates keep off-target neurons' pre-trained features nearly intact.
Key results
- Multi-step SGD reaches alignment theta_kappa^T w >= 1 - eps within T = Otilde(s^{3/2}) samples under eps = Otilde(s^{-1/2}) scaling.
- Off-target task features are preserved up to a drop of Otilde(||Sigma_kappa w_k^0||^2 + eps + chi_k s^{-1/2}).
- If the generative exponent of the link is 1 instead of 2, off-target neurons accumulate alignment with the target and forget pre-trained features (Remark).
Surprising findings
- A teacher signal with generative exponent 1 causes forgetting of pre-trained features for other tasks, so the transformation to exponent 2 is essential.
Limitations
- Analysis restricted to two-layer networks with polynomially-structured Hermite assumptions and polylog constants.
- Results hold with high probability under specific initialization and learning-rate schedules.
Experimental setup
Metrics: sample complexity T feature alignment theta^T w
Builds on: weak-to-strong generalization information exponent / generative exponent theory single-index feature learning with SGD
Closest papers by meaning
- Representations Shape Weak-to-Strong Generalization: Theoretical Insights and Empirical Predictions (ICML 2025 · similarity 0.93)
- From Linear to Nonlinear: Provable Weak-to-Strong Generalization through Feature Learning (NeurIPS 2025 · similarity 0.91)
- Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervision (ICML 2024 · similarity 0.91)
- Weak-to-Strong Generalization under Distribution Shifts (NeurIPS 2025 · similarity 0.90)
- Strong Empowered and Aligned Weak Mastered Annotation for Weak-to-Strong Generalization (AAAI 2025 · similarity 0.90)
- Relating Misfit to Gain in Weak-to-Strong Generalization Beyond the Squared Loss (ICML 2025 · similarity 0.90)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
