arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.04462cs.CV
收藏

面向深度人脸识别的分组知识蒸馏

Grouped Knowledge Distillation for Deep Face Recognition

Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);School of Cyber Security, University of Chinese Academy of Sciences(中国科学院大学网络空间安全学院);School of Artificial Intelligence, University of Chinese Academy of Sciences(中国科学院大学人工智能学院);Centre for Artificial Intelligence and Robotics, Hong Kong Institute of Science & Innovation, Chinese Academy of Sciences(中国科学院香港创新研究院人工智能与机器人中心)

Weisong Zhao, Xiangyu Zhu, Kaiwen Guo, Xiao-Yu Zhang, Zhen Lei

AI总结:

针对人脸识别中轻量级学生网络难以拟合目标logits的问题,提出分组知识蒸馏方法GKD,通过将logits分组并保留主要知识蒸馏与分布一致性损失、省略次要知识蒸馏,显著提升了性能。

摘要:

与基于特征的知识蒸馏方法相比,logits蒸馏可以放宽对教师网络和学生网络特征维度一致性的要求,但在人脸识别中的性能被认为是较差的。一个主要挑战是,由于人脸识别中身份数量众多,轻量级学生网络因其模型容量低而难以拟合目标logits。因此,我们试图探测目标logits以提取与人脸身份相关的主要知识,并丢弃其他知识,从而使蒸馏对学生网络更具可行性。具体而言,预测中存在一个包含近零值的尾部组,其包含的次要知识对蒸馏作用微小。为了清晰地说明其影响,我们首先根据软化预测的累积概率将logits分为两组,即Primary Group(主要组)和Secondary Group(次要组)。然后,我们将分组logits的知识蒸馏(KD)损失重组为三部分,即Primary-KD、Secondary-KD和Binary-KD。Primary-KD指从教师网络蒸馏主要知识,Secondary-KD旨在细化次要知识但会增加蒸馏难度,Binary-KD确保教师和学生网络间知识分布的一致性。我们通过实验发现:(1) Primary-KD和Binary-KD对KD不可或缺,(2) Secondary-KD是限制KD达到瓶颈的主因。因此,我们提出了分组知识蒸馏(GKD),在最终KD损失计算中保留Primary-KD和Binary-KD而省略Secondary-KD。在流行人脸识别基准上的大量实验结果证明了所提GKD优于现有先进方法。

英文摘要:

Compared with the feature-based distillation methods, logits distillation can liberalize the requirements of consistent feature dimension between teacher and student networks, while the performance is deemed inferior in face recognition. One major challenge is that the light-weight student network has difficulty fitting the target logits due to its low model capacity, which is attributed to the significant number of identities in face recognition. Therefore, we seek to probe the target logits to extract the primary knowledge related to face identity, and discard the others, to make the distillation more achievable for the student network. Specifically, there is a tail group with near-zero values in the prediction, containing minor knowledge for distillation. To provide a clear perspective of its impact, we first partition the logits into two groups, i.e., Primary Group and Secondary Group, according to the cumulative probability of the softened prediction. Then, we reorganize the Knowledge Distillation (KD) loss of grouped logits into three parts, i.e., Primary-KD, Secondary-KD, and Binary-KD. Primary-KD refers to distilling the primary knowledge from the teacher, Secondary-KD aims to refine minor knowledge but increases the difficulty of distillation, and Binary-KD ensures the consistency of knowledge distribution between teacher and student. We experimentally found that (1) Primary-KD and Binary-KD are indispensable for KD, and (2) Secondary-KD is the culprit restricting KD at the bottleneck. Therefore, we propose a Grouped Knowledge Distillation (GKD) that retains the Primary-KD and Binary-KD but omits Secondary-KD in the ultimate KD loss calculation. Extensive experimental results on popular face recognition benchmarks demonstrate the superiority of proposed GKD over state-of-the-art methods.

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

↑