arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.10482cs.LGstat.ML
收藏

重新思考自蒸馏:标签平均与利用部分标签的增强软标签精炼

Rethinking Self-Distillation: Label Averaging and Enhanced Soft Label Refinement with Partial Labels

Hyeonsu Jeong, Hye Won Chung

AI总结:

本文揭示固定特征线性探测下自蒸馏的标签平均机制,并提出利用教师前两个 softmax 输出的单轮 PLL student 模型,在噪声场景中以更低成本取得相当或更优性能。

摘要:

我们研究多类分类中自蒸馏的机制,特别是在使用固定特征提取器进行线性探测的情境下,此时传统的特征学习解释并不适用。我们的理论分析表明,多轮自蒸馏实际上是在具有高特征相关性的实例之间执行标签平均,该过程由输入特征导出的 Gram 矩阵的特征向量所支配。这一过程产生聚类式预测并改善泛化能力,通过降低模型对可能已损坏标签的依赖来缓解标签噪声的影响。我们建立了多轮自蒸馏在存在标签噪声的情况下仍能达到100%总体准确率的条件。此外,我们提出一种新颖、高效的单轮自蒸馏方法,该方法使用来自教师前两个 softmax 输出的精炼部分标签,称为 PLL student 模型。该方法在单轮中复现了多轮蒸馏的收益,取得相当或更优的性能——尤其是在高噪声场景下——同时显著降低计算成本。

英文摘要:

We investigate the mechanisms of self-distillation in multi-class classification, particularly in the context of linear probing with fixed feature extractors where traditional feature learning explanations do not apply. Our theoretical analysis reveals that multi-round self-distillation effectively performs label averaging among instances with high feature correlations, governed by the eigenvectors of the Gram matrix derived from input features. This process leads to clustered predictions and improved generalization, mitigating the impact of label noise by reducing the model's reliance on potentially corrupted labels. We establish conditions under which multi-round self-distillation achieves 100% population accuracy despite label noise. Furthermore, we introduce a novel, efficient single-round self-distillation method using refined partial labels from the teacher's top two softmax outputs, referred to as the PLL student model. This approach replicates the benefits of multi-round distillation in a single round, achieving comparable or superior performance--especially in high-noise scenarios--while significantly reducing computational cost.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑