
摘要
人脸识别(FR)的目标可以被视为一个配对相似度优化问题,即最大化正样本对的相似度集合 $\mathcal{S}^p$,同时最小化负样本对的相似度集合 $\mathcal{S}^n$。理想情况下,期望人脸识别模型能够形成一个具有良好区分性的特征空间(WDFS),满足条件 $\inf{\mathcal{S}^p} > \sup{\mathcal{S}^n}$。关于 WDFS,现有的深度特征学习范式(即度量损失和分类损失)可以从不同的配对生成(PG)策略的角度进行统一表述。然而,在度量损失(ML)中,由于小批量大小的限制,每次迭代时考虑所有类别的负样本对生成是不可行的。相比之下,在分类损失(CL)中,由于类别权重向量收敛到其中心点,生成极其困难的负样本对也变得困难。这导致了采样负样本对与所有负样本对之间的两个相似度分布不匹配。因此,本文提出了一种统一的负样本对生成方法(UNPG),通过从统一视角结合两种 PG 策略(即 MLPG 和 CLPG)来缓解这种不匹配。UNPG 利用 MLPG 提供的有关负样本对的有用信息来克服 CLPG 的不足。此外,它还包括过滤噪声负样本对的相似度,以保证可靠的收敛性和性能提升。大量实验表明,UNPG 在公共基准数据集上通过实现最先进的性能超越了近期的各种损失函数。我们的代码和预训练模型已公开发布。
代码仓库
tomas-gajarsky/facetorch
pytorch
GitHub 中提及
jung-jun-uk/unpg
官方
pytorch
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| face-identification-on-megaface | Cos+UNPG | Accuracy: 99.27% |
| face-identification-on-megaface | Mag+UNPG | Accuracy: 98.03% |
| face-identification-on-megaface | Arc+UNPG | Accuracy: 98.82% |
| face-verification-on-ijb-b | Arc+UNPG | TAR @ FAR=0.001: 96.6 TAR @ FAR=0.01: 97.7% TAR@FAR=0.0001: 95.04 |
| face-verification-on-ijb-b | Cos+UNPG | TAR @ FAR=0.001: 96.5 TAR @ FAR=0.01: 97.36% TAR@FAR=0.0001: 94.99 |
| face-verification-on-ijb-b | Mag+UNPG | TAR @ FAR=0.001: 96.5 TAR @ FAR=0.01: 97.63% TAR@FAR=0.0001: 95.21 |
| face-verification-on-ijb-c | Mag+UNPG | TAR @ FAR=1e-5: 94.7% |
| face-verification-on-ijb-c | Cos+UNPG | TAR @ FAR=1e-3: 97.57 TAR @ FAR=1e-4: 96.38% TAR @ FAR=1e-5: 94.47% model: R100 training dataset: MS1MV2 |
| face-verification-on-ijb-c | Arc+UNPG | TAR @ FAR=1e-3: 97.51 TAR @ FAR=1e-4: 96.33% |