Additive Margin Softmax for Face Verification

论文地址:点击打开链接

源码地址:点击打开链接

1. Softmax loss

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

其中f为最后全连接层的输入(fi表示第i个样本),wj为全连接层的第j列。

2. A-Softmax loss 

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification


3. AMSoftmax loss

本文提出的方法将cos(mθ)更改为cosθ-m,将sphereface中的倍乘更改为加法。增加类间距,并减小类内距。为了提高收敛速度,引进了一个超参数s,这里s设置为固定值30。

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

几种loss函数的特征分布可视化结果如下:

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

4. 实验结果如下:

(本文采用的是CASIA-WebFace数据集作为训练数据,在lfw以及megaface数据集上进行验证。CASIA-WebFace数据集和lfw数据集有17个重复身份的人,CASIA-WebFace数据集和megaface数据集有42个重复身份的人。本文采用的是清洗过后的CASIA-WebFace数据集作为训练集。)

sphereface中只对w进行了归一化,本文对w和x都进行了归一化处理。其中w/o FN表示没有进行特征归一化处理。

人脸识别 -- AMSoftmax : Additive Margin Softmax for Face Verification

实验结果表明,在高质量图片集(lfw)中,不使用特征归一化结果更好。在很多低质量的图片集(megaface)中,使用特征归一化结果更好。


相关文章: