6
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

顔認証におけるマージンベースな損失関数

Last updated at Posted at 2019-10-10

概要

これまで、顔認証における様々な損失関数が提案された。最適化手法における損失関数としては現在、Softmax,Contrastive Loss,Triplet Loss,Center Loss,Norm Face,Large-Margin Loss,A-Softmax Lossなどの手法がある。近年の手法の遷移は下記となる。その中、2017以降の主流となっているのはマージンベースな手法である。
1.JPG

前提知識

  • softmax loss

定義:Softmax Loss = FC + Softmax + Cross-Entropy
softmax loss 1.JPG

式:
softmax loss.JPG

「Large-Margin Softmax Loss for Convolutional Neural Networks」は初めて、softmax lossをcross entropy、softmax関数、最後の全結合層の組み合わせとして定義した。それ以後の論文はこの式を利用して、様々な亜種を提案した。

マージンベースな手法

下記、いくつかのロスを簡単に整理した。

  • Large-Margin Softmax(L-Softmax)
    論文:Large-Margin Softmax Loss for Convolutional Neural Networks
    定義:
    1.JPG
    ここでは、
    2.JPG
    3.JPG
    この図は、softmax loss(m=1)またはL-Softmax loss(m=2,3,4)で学習した特徴を可視化した結果である。クラス内のコンパクトさとクラス間の分離性を高める。直接マージンを求める訳ではなく、あるクラスに属するデータが分離平面上でできるだけ狭い角度に収まるようにする。

  • Sphereface(A-Softmax)
    論文:SphereFace: Deep Hypersphere Embedding for Face Recognition
    定義:
    1.JPG
    ここでは、
    3.JPG
    2.JPG

これはL-softmaxで出てくる重み$W$を正規化する。$m=1$のとき、A-Softmax lossはmodified softmax lossになる。

特徴量を l2-normalize してそれと列ごとに l2-normalize した重み$W$を掛けて、それぞれの要素が $cosθ_i$ として表せる特徴量を作る。この要素数は最終的な予測クラス数と同じになっていて、GroundTruth のラベルに対応する要素には $cos(θ + m)$ とマージンを入れることでクラス間をより分離するようにして、この特徴量を softmax につなぐということをする。

まとめ

Softmax、SphereFace、CosFace、ArcFaceが違いを比較する。
3.png
ここで、m1 が SphereFace で m2 が ArcFace で m3 が CosFace の寄与となっている。

マージンがどのように取られるかは手法によって振る舞いが異なり、ArcFace は角度そのものにマージンを加えているので、下図のように線形にマージンが確保されている。
2.JPG

参考

[1] Deep Face Recognition: A Survey

6
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
6
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?