arXiv:2502.05075cs.LGcs.NA2025-02ICML被引 9

弱教师生成伪标签,强学生模型反而表现更好,原因在低维特征空间的方差减少。

Discrepancies are Virtue: Weak-to-Strong Generalization through Lens of Intrinsic Dimension

  • 利用特征子空间的低内在维数特性,从方差减少角度分析弱到强微调机制。
  • 在伪标签数量为N时,差异区域的方差降低至原始水平的dim(V_s)/N倍。
  • 适用于理解大模型微调中性能提升的本质,适合研究泛化与模型压缩者阅读。

弱到强(W2S)微调是一种强学生模型在弱教师生成的伪标签上进行训练的方法。令人惊讶的是,这种微调常优于弱教师本身。本文通过观察到微调通常发生在内在低维空间的现象,借助无正则化回归框架,从方差缩减视角分析了W2S。对于具有足够表达能力的低维特征子空间 $\mathcal{V}_s$(学生)和 $\mathcal{V}_w$(教师),我们精确刻画了主导W2S泛化误差的方差。这揭示了强弱模型差异的积极作用:在交集区域 $\mathcal{V}_s \cap \mathcal{V}_w$,弱教师的方差被学生继承;而在差异区域 $\mathcal{V}_w \setminus \mathcal{V}_s$,方差被降低至 $\mathrm{dim}(\mathcal{V}_s)/N$ 倍,其中 $N$ 为伪标签数量。分析进一步阐明了样本复杂度及性能差距恢复的缩放规律。实验验证涵盖合成回归任务以及真实视觉与自然语言处理任务。

原文摘要 · Abstract (English)

Weak-to-strong (W2S) generalization is a type of finetuning (FT) where a strong (large) student model is trained on pseudo-labels generated by a weak teacher. Surprisingly, W2S FT often outperforms the weak teacher. We seek to understand this phenomenon through the observation that FT often occurs in intrinsically low-dimensional spaces. Leveraging the low intrinsic dimensionality of FT, we analyze W2S in the ridgeless regression setting from a variance reduction perspective. For a strong student-weak teacher pair with sufficiently expressive low-dimensional feature subspaces $\mathcal{V}_s, \mathcal{V}_w$, we provide an exact characterization of the variance that dominates the generalization error of W2S. This unveils a virtue of discrepancy between the strong and weak models in W2S: the variance of the weak teacher is inherited by the strong student in $\mathcal{V}_s \cap \mathcal{V}_w$, while reduced by a factor of $\mathrm{dim}(\mathcal{V}_s)/N$ in the subspace of discrepancy $\mathcal{V}_w \setminus \mathcal{V}_s$ with $N$ pseudo-labels for W2S. Our analysis further casts light on the sample complexities and the scaling of performance gap recovery in W2S. The analysis is supported by experiments on synthetic regression problems, as well as real vision and NLP tasks.

泛化分析微调机制低维结构方差缩减

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。