3 个月前

DupNet:面向极小量化CNN的改进准确率人脸检测方法

DupNet:面向极小量化CNN的改进准确率人脸检测方法

摘要

在边缘设备上部署基于深度学习的人脸检测模型是一项极具挑战性的任务,主要受限于设备的计算资源。尽管对极小网络的权重进行二值化处理可显著减小模型体积(例如,IFQ-Tinier-YOLO的模型大小仅为240.9 KB),但该尺寸仍不足以满足内存严格受限的嵌入式设备的需求。为此,本文提出了一种名为DupNet的新架构,其包含两个核心部分:首先,针对权重密集型层,采用通道复制(duplicated channels)的权重设计,以进一步压缩模型规模;其次,针对对量化敏感、量化后导致显著精度下降的层,我们复制其输入特征图(feature maps),从而允许使用更多权重通道进行卷积操作,以生成更具代表性的输出。基于上述设计,我们构建了一个极小的人脸检测模型——DupNet-Tinier-YOLO,其模型尺寸比IFQ-Tinier-YOLO缩小了6.5倍,计算复杂度降低42.0%,同时检测精度提升2.4%。与全精度Tiny-YOLO相比,DupNet-Tinier-YOLO在模型尺寸上实现了1,694.2倍、计算复杂度上实现389.9倍的显著压缩,仅带来4.0%的检测率下降(0.880 vs. 0.920)。此外,我们的模型大小仅为36.9 KB,据我们所知,这是目前已知最小的深度学习人脸检测模型。

基准测试

基准方法指标
face-detection-on-wider-faceIFQ-Tinier-YOLO
GFLOPs: 0.1079
face-detection-on-wider-face-hardDupNet-L+PACT
AP: 0.906

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DupNet:面向极小量化CNN的改进准确率人脸检测方法 | 论文 | HyperAI超神经