arXiv:2409.20449cs.LGcs.AI2024-09被引 1

通过线性投影教师嵌入,提升小样本分类知识蒸馏效果

Linear Projections of Teacher Embeddings for Few-Class Distillation

  • 从教师模型嵌入空间中提取有信息量的线性子空间
  • 在亚马逊评论和Sentiment140上优于主流蒸馏方法
  • 特别适合情感分析等少类别场景

知识蒸馏(KD)已成为将大型教师模型的知识迁移到小型学生模型的有效方法。传统方法让学生模仿教师输出概率,更先进方法则引导学生学习教师内部表示。然而,在二分类和少类别问题中,KD表现仍不理想,因为教师泛化模式的信息量随类别数增加而增长。此外,许多复杂蒸馏方法对计算机视觉以外的数据类型效果有限。为此,我们提出一种新方法——学习嵌入线性投影(LELP),受最终层表示结构启发,通过识别教师嵌入空间中有信息的线性子空间并划分为伪子类,让学生学习这些伪子类。在Amazon Reviews和Sentiment140等大规模NLP基准测试中,LELP在二分类和少类别任务上始终优于或媲美现有最先进蒸馏算法,显著改善了传统方法的性能瓶颈。

原文摘要 · Abstract (English)

Knowledge Distillation (KD) has emerged as a promising approach for transferring knowledge from a larger, more complex teacher model to a smaller student model. Traditionally, KD involves training the student to mimic the teacher's output probabilities, while more advanced techniques have explored guiding the student to adopt the teacher's internal representations. Despite its widespread success, the performance of KD in binary classification and few-class problems has been less satisfactory. This is because the information about the teacher model's generalization patterns scales directly with the number of classes. Moreover, several sophisticated distillation methods may not be universally applicable or effective for data types beyond Computer Vision. Consequently, effective distillation techniques remain elusive for a range of key real-world applications, such as sentiment analysis, search query understanding, and advertisement-query relevance assessment. Taking these observations into account, we introduce a novel method for distilling knowledge from the teacher's model representations, which we term Learning Embedding Linear Projections (LELP). Inspired by recent findings about the structure of final-layer representations, LELP works by identifying informative linear subspaces in the teacher's embedding space, and splitting them into pseudo-subclasses. The student model is then trained to replicate these pseudo-classes. Our experimental evaluation on large-scale NLP benchmarks like Amazon Reviews and Sentiment140 demonstrate the LELP is consistently competitive with, and typically superior to, existing state-of-the-art distillation algorithms for binary and few-class problems, where most KD methods suffer.

知识蒸馏少样本学习NLP嵌入投影

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