English

On Weak-to-Strong Generalization and f-Divergence

Machine Learning 2025-06-04 v1

Abstract

Weak-to-strong generalization (W2SG) has emerged as a promising paradigm for stimulating the capabilities of strong pre-trained models by leveraging supervision from weaker supervisors. To improve the performance of the strong model, existing methods often require additional weak models or complex procedures, leading to substantial computational and memory overhead. Motivated by the effectiveness of ff-divergence loss in various machine learning domains, we introduce ff-divergence as an information-theoretic loss function framework in W2SG. Our theoretical analysis reveals fundamental limitations and equivalence of different ff-divergence losses in W2SG, supported by sample complexity bounds and information-theoretic insights. We empirically demonstrate that ff-divergence loss, which generalizes widely-used metrics like KL divergence, effectively improves generalization and noise tolerance of the strong model in practice.

Keywords

Cite

@article{arxiv.2506.03109,
  title  = {On Weak-to-Strong Generalization and f-Divergence},
  author = {Wei Yao and Gengze Xu and Huayi Tang and Wenkai Yang and Donglin Di and Ziqiao Wang and Yong Liu},
  journal= {arXiv preprint arXiv:2506.03109},
  year   = {2025}
}
R2 v1 2026-07-01T02:57:26.786Z