AI News HubLIVE
站内改写2 分钟阅读

ForensicNet: 轻量级注意力增强MobileNetV2用于自动人脸识别

本文提出ForensicNet,一种轻量级深度学习框架,用于法医环境中的自动人脸识别。该模型结合MobileNetV2主干与CBAM注意力模块,采用两阶段迁移学习,在LFW和SCFace数据集上达到92.4%准确率,每推理仅需2.1 GFLOPs,适合实时法医监控。

来源arXiv Computer Vision作者: Savitha N J, Lata B T

在法医鉴定领域,自动识别嫌疑人面临诸多挑战,如姿态变化、光照条件不佳、面部遮挡以及标记数据的缺乏。针对这些问题,研究人员提出了ForensicNet——一种轻量级的深度学习框架,专门用于法医人脸识别,通过增强注意力机制来提高识别性能。

ForensicNet的核心架构基于MobileNetV2骨干网络,并集成了卷积块注意力模块(CBAM)。CBAM能够同时关注通道和空间维度,帮助网络更有效地学习判别特征,同时保持较低的计算成本。MobileNetV2本身以其深度可分离卷积和线性瓶颈结构著称,能够在边缘设备上高效运行。结合CBAM后,模型能够在关键区域分配更多权重,从而提升在复杂场景下的识别准确率。

为了进一步提升模型在法医场景中的适应性,研究团队采用了两阶段迁移学习策略。第一阶段,在大型通用人脸数据集上预训练模型;第二阶段,在法医特定数据集上微调,并采用自适应层解冻技术——即逐步解冻网络层,以避免灾难性遗忘并减少过拟合风险。这种策略有效改善了领域转移问题,使模型更好地适应法医环境中的低质量监控图像。

实验部分,研究人员使用了两个公开数据集:LFW(Labeled Faces in the Wild)和SCFace(Surveillance Camera Face Database),共包含15000张面部图像,覆盖68个身份类别。这些数据模拟了真实监控场景中的挑战,包括不同角度、光照不足和部分遮挡。与AlexNet、ResNet-50以及原始的MobileNetV2等基线架构相比,ForensicNet表现出更优越的性能,取得了92.4%的准确率、90.8%的精确率和89.5%的召回率。此外,该框架每次推理仅需2.1 GFLOPs的计算量,使其能够部署在实时法医监控系统中,满足实际执法的时效性要求。

这项研究的意义在于,它提供了一种平衡精度与效率的解决方案,有望在实际执法和司法鉴定中发挥重要作用。未来工作可以进一步探索在更大规模数据集上的表现,以及针对更多样化的环境条件进行优化,例如夜间监控或极端遮挡情况。研究者还计划将模型集成到嵌入式系统中,以实现便携式法医识别设备。