论文标题

夫妇:关系识别蒸馏的关系很重要

CoupleFace: Relation Matters for Face Recognition Distillation

论文作者

Liu, Jiaheng, Qin, Haoyu, Wu, Yichao, Guo, Jinyang, Liang, Ding, Xu, Ke

论文摘要

知识蒸馏是一种有效的方法,可以通过转移良好表现的神经网络(即教师模型)的知识来提高轻型神经网络(即学生模型)的性能,该方法已广泛应用于许多计算机视觉任务,包括面部识别。然而,当前的面部识别蒸馏方法通常会利用特征一致性蒸馏(例如FCD)(例如,L2距离)在教师和学生模型中为每个样本提取的学习嵌入,这无法将知识从老师中完全传递到学生的面部识别。在这项工作中,我们观察到,样本之间的相互关系知识对于提高学生模型的判别能力也很重要,并通过将相互关系蒸馏(MRD)引入现有的蒸馏框架中,提出了一种称为助教的面部识别蒸馏方法。具体而言,在MRD中,我们首先建议挖掘信息丰富的相互关系,然后引入关系感知的蒸馏(RAD)损失(RAD),以将教师模型的相互关系知识转移到学生模型中。在多个基准数据集上进行的广泛实验结果证明了我们提出的夫妻面对面部识别的有效性。此外,根据我们提出的夫妻表面,我们赢得了ICCV21蒙面面部识别挑战赛(MS1M轨道)的第一名。

Knowledge distillation is an effective method to improve the performance of a lightweight neural network (i.e., student model) by transferring the knowledge of a well-performed neural network (i.e., teacher model), which has been widely applied in many computer vision tasks, including face recognition. Nevertheless, the current face recognition distillation methods usually utilize the Feature Consistency Distillation (FCD) (e.g., L2 distance) on the learned embeddings extracted by the teacher and student models for each sample, which is not able to fully transfer the knowledge from the teacher to the student for face recognition. In this work, we observe that mutual relation knowledge between samples is also important to improve the discriminative ability of the learned representation of the student model, and propose an effective face recognition distillation method called CoupleFace by additionally introducing the Mutual Relation Distillation (MRD) into existing distillation framework. Specifically, in MRD, we first propose to mine the informative mutual relations, and then introduce the Relation-Aware Distillation (RAD) loss to transfer the mutual relation knowledge of the teacher model to the student model. Extensive experimental results on multiple benchmark datasets demonstrate the effectiveness of our proposed CoupleFace for face recognition. Moreover, based on our proposed CoupleFace, we have won the first place in the ICCV21 Masked Face Recognition Challenge (MS1M track).

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源