2017的一篇cvpr,SphereFace: Deep Hypersphere Embedding for Face Recognition,继centerloss之后又一大作。

文章主要提出了归一化权值(normalize weights and zero biases)和角度间距(angular margin),基于这2个点,对传统的softmax进行了改进,从而实现了,最大类内距离小于最小类间距离的识别标准。

 

 

原始的softmax的loss损失是一个互熵损失,

 

其中,

 

将其代入第一个式子,得出了损失函数如下,

 

进行归一化操作,将其都映射到一个单位球面上,令||W||=1,b=0,并且引入夹角,得出Modified Softmax Loss公式如下,

 

在此基础上,再引入angular margin,用m表示,最终产生A-softmax的loss公式

 

在LFW和YTF数据集上的识别率分别如下图所示,可以看出LFW已经达到了99.42的识别率。

 

这里分析一下,作者A-softmax 的参数,

 

在margin_inner_product_layer.cpp的Forward_cpu函数中,

也就是说,

 

计算出来也就是说,lambda=5(iteration>1658)lambda=大于5的值(iteration<1658)

 

训练过程loss下降如下,

 

自己在LFW上10折测试的结果如下,

 

 

reference:

https://github.com/wy1iu/sphereface

 

 

 

 

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐