研究伪相关下弱到强泛化是否成立及如何改进
Does Weak-to-strong Generalization Happen under Spurious Correlations?
- 分析教师模型在不平衡数据下的伪标签对泛化的影响
- 理论证明当数据比例相等时泛化始终成立,否则性能随差异平方下降
- 提出无需标签的重训练策略,显著提升失败场景下的效果
我们首次对弱到强(W2S)泛化中的关键问题进行统一的理论与算法研究:当使用弱教师模型生成伪标签,对强学生模型进行微调以应对存在伪相关性的下游任务时,W2S 是否发生?如何在失败时改进?考虑由组不平衡引起的两类伪相关:(i) 弱教师在标签不平衡数据上微调,少数组占比为 $η_ ext{ℓ}$;(ii) 教师对未标记数据进行伪标注,少数组占比为 $η_ ext{u}$。理论上,在比例渐近极限下,当 $η_ ext{u} = η_ ext{ℓ}$ 时,只要有足够伪标签,W2S 总是成立;而当 $η_ ext{u} e η_ ext{ℓ}$ 时,可能失败,且 W2S 收益随 $(η_ ext{u} - η_ ext{ℓ})^2$ 增大而衰减。该理论在多个伪相关基准和教师-学生组合上得到广泛实验验证。为进一步提升失败情况下的 W2S 表现,我们提出一种简单有效的算法补救:在完成 W2S 微调后,用高置信度子集重新训练强学生模型。该方法无需组标签,且一致实现显著优于原始 W2S 微调的性能提升。
原文摘要 · Abstract (English)
We initiate a unified theoretical and algorithmic study of a key problem in weak-to-strong (W2S) generalization: when fine-tuning a strong pre-trained student with pseudolabels from a weaker teacher on a downstream task with spurious correlations, does W2S happen, and how to improve it upon failures? We consider two sources of spurious correlations caused by group imbalance: (i) a weak teacher fine-tuned on group-imbalanced labeled data with a minority group of fraction $η_\ell$, and (ii) a group-imbalanced unlabeled set pseudolabeled by the teacher with a minority group of fraction $η_u$. Theoretically, a precise characterization of W2S gain at the proportional asymptotic limit shows that W2S always happens with sufficient pseudolabels when $η_u = η_\ell$ but may fail when $η_u \ne η_\ell$, where W2S gain diminishes as $(η_u - η_\ell)^2$ increases. Our theory is corroborated by extensive experiments on various spurious correlation benchmarks and teacher-student pairs. To boost W2S performance upon failures, we further propose a simple, effective algorithmic remedy that retrains the strong student on its high-confidence data subset after W2S fine-tuning. Our algorithm is group-label-free and achieves consistent, substantial improvements over vanilla W2S fine-tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。