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

协作知识蒸馏:一种学习者无关的方法

Cooperative Knowledge Distillation: A Learner Agnostic Approach

University of California, Davis(加利福尼亚大学戴维斯分校)

Michael Livanos, Ian Davidson, Stephen Wong

AI总结:

提出协作蒸馏方法,让多模型互为师生,通过反事实实例生成针对性传递知识,实现跨架构算法蒸馏,优于现有基线。

摘要:

知识蒸馏是一种简单但强大的方法,用于在教师模型和学生模型之间传递知识。现有工作在传递方向和范围方面至少存在以下关键局限性之一,这限制了其使用:所有知识都从教师传递给学生,无论该知识是否有用;学生是这种交换中唯一学习的一方;并且通常蒸馏仅从单个教师向单个学生传递知识。我们提出了一种新颖的知识蒸馏形式,其中多个模型可以同时充当学生和教师,我们称之为协作蒸馏。这些模型按如下方式协作:一个模型(学生)识别其性能中的具体缺陷,并搜索另一个模型(教师),该教师通过反事实实例生成将所学知识编码为教学性虚拟实例。由于不同模型可能具有不同的优势和劣势,所有模型在适当时都可以充当学生或教师(协作),并且仅在其优势相关的特定领域进行蒸馏(聚焦)。由于反事实作为一种范式不依赖于任何特定算法,我们可以使用这种方法在不同架构、算法甚至特征空间的学习者之间蒸馏知识。我们证明,我们的方法不仅在多个数据集上优于迁移学习、自监督学习和多种知识蒸馏算法等基线,而且还可以在上述技术无法应用的场景中使用。

英文摘要:

Knowledge distillation is a simple but powerful way to transfer knowledge between a teacher model to a student model. Existing work suffers from at least one of the following key limitations in terms of direction and scope of transfer which restrict its use: all knowledge is transferred from teacher to student regardless of whether or not that knowledge is useful, the student is the only one learning in this exchange, and typically distillation transfers knowledge only from a single teacher to a single student. We formulate a novel form of knowledge distillation in which many models can act as both students and teachers which we call cooperative distillation. The models cooperate as follows: a model (the student) identifies specific deficiencies in it's performance and searches for another model (the teacher) who encodes learned knowledge into instructional virtual instances via counterfactual instance generation. Because different models may have different strengths and weaknesses, all models can act as either students or teachers (cooperation) when appropriate and only distill knowledge in areas specific to their strengths (focus). Since counterfactuals as a paradigm are not tied to any specific algorithm, we can use this method to distill knowledge between learners of different architectures, algorithms, and even feature spaces. We demonstrate that our approach not only outperforms baselines such as transfer learning, self-supervised learning, and multiple knowledge distillation algorithms on several datasets, but it can also be used in settings where the aforementioned techniques cannot.

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

↑