4 个月前

PocketNet:基于神经架构搜索和多阶段知识蒸馏的极轻量级人脸识别网络

PocketNet:基于神经架构搜索和多阶段知识蒸馏的极轻量级人脸识别网络

摘要

深度神经网络已迅速成为面部识别(FR)的主流方法。然而,这限制了包含极大量参数的此类模型在嵌入式和低端设备上的部署。在本研究中,我们提出了一种极其轻量且准确的面部识别解决方案,即PocketNet。我们利用神经架构搜索(NAS)开发了一种新的轻量级面向面部的架构系列。此外,我们提出了一种基于知识蒸馏(KD)的新颖训练范式——多阶段知识蒸馏,其中知识在训练的不同成熟阶段从教师模型逐步传递到学生模型。我们进行了详细的消融研究,证明了使用NAS专门针对面部识别任务而非通用对象分类的有效性,以及所提出的多阶段KD的优势。我们展示了广泛的实验评估,并与九个不同基准测试中的现有最先进(SOTA)紧凑型面部识别模型进行了比较,包括大规模评估基准如IJB-B、IJB-C和MegaFace。当考虑相同水平的模型紧凑性时,PocketNet在九个主流基准测试中均显著提升了SOTA面部识别性能。我们的最小网络PocketNetS-128仅包含0.92M参数,在与最近包含高达4M参数的SOTA紧凑型模型相比时,取得了非常有竞争力的结果。

代码仓库

fdbtrs/pocketnet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
lightweight-face-recognition-on-agedb-30PocketNetS
Accuracy: 0.9635
lightweight-face-recognition-on-calfwPocketNetS
Accuracy: 0.955
MParams: 0.99
lightweight-face-recognition-on-cfp-fpPocketNetS
Accuracy: 0.9334
lightweight-face-recognition-on-cplfwPocketNetS
Accuracy: 0.8893
lightweight-face-recognition-on-lfwPocketNetS
Accuracy: 0.9966
MFLOPs: 587.24
MParams: 0.99

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PocketNet:基于神经架构搜索和多阶段知识蒸馏的极轻量级人脸识别网络 | 论文 | HyperAI超神经