arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2204.08922cs.CLcs.AIcs.LG

BERT迁移中基于中心化核对齐的特征结构蒸馏

Feature Structure Distillation with Centered Kernel Alignment in BERT Transferring

AI graduate school of Gwangju Institute of Science and Technology (GIST)(光州科学技术院AI研究生院);Jeonbook National University(全北国立大学)

Hee-Jun Jung, Doyeon Kim, Seung-Hoon Na, Kangil Kim

AI总结:

针对BERT知识蒸馏中学生表示灵活性受限的问题,提出基于中心化核对齐的特征结构蒸馏方法,迁移三类表示结构并在GLUE九项任务上优于现有蒸馏方法。

摘要:

知识蒸馏是一种通过缩小教师模型与学生模型之间表示差异,将表示信息从教师迁移到学生的方法。该方法的一个挑战是会降低学生表示的灵活性,从而导致对教师知识的不准确学习。为在迁移过程中解决这一问题,我们研究了针对三类表示结构的蒸馏:特征内结构、局部特征间结构和全局特征间结构。为迁移这些结构,我们提出了基于中心化核对齐(Centered Kernel Alignment)的特征结构蒸馏方法,该方法为相似的特征结构赋予一致的值,并揭示更具信息量的关系。特别地,针对全局结构,我们实现了一种结合聚类的记忆增强迁移方法。我们使用具有代表性的神经语言模型——来自Transformer的双向编码器表示(Bidirectional Encoder Representations from Transformers,BERT),在GLUE数据集的九项语言理解任务上对这些方法进行了实证分析。结果表明,所提出的方法能够有效迁移这三类结构,并且与最先进的蒸馏方法相比提升了性能。此外,这些方法的代码可在https://github.com/maroo-sky/FSD获取。

英文摘要:

Knowledge distillation is an approach to transfer information on representations from a teacher to a student by reducing their difference. A challenge of this approach is to reduce the flexibility of the student's representations inducing inaccurate learning of the teacher's knowledge. To resolve it in transferring, we investigate distillation of structures of representations specified to three types: intra-feature, local inter-feature, global inter-feature structures. To transfer them, we introduce feature structure distillation methods based on the Centered Kernel Alignment, which assigns a consistent value to similar features structures and reveals more informative relations. In particular, a memory-augmented transfer method with clustering is implemented for the global structures. The methods are empirically analyzed on the nine tasks for language understanding of the GLUE dataset with Bidirectional Encoder Representations from Transformers (BERT), which is a representative neural language model. In the results, the proposed methods effectively transfer the three types of structures and improve performance compared to state-of-the-art distillation methods. Indeed, the code for the methods is available in https://github.com/maroo-sky/FSD.

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

↑