Add ArcFace loss.

System information.

TensorFlow version (you are using): 2.8 Are you willing to contribute it (Yes/No) : No (but willing to provide starter if required.)

Describe the feature and the current behavior/state.

ArcFace, or Additive Angular Margin Loss, is a loss function used mainly in face recognition tasks. It is used to obtain highly discriminative features. This loss function can be used in <del>both</del> vision task <del>and nlp task</del>. <del>So, I think its general for keras_cv and keras_nlp</del>.

Paper: https://arxiv.org/pdf/1801.07698.pdf, TL,DR, https://paperswithcode.com/method/arcface Citation: 2939 (current)

Will this change the current api? How?

tf.keras.losses.SparseArcFace
tf.keras.losses.CategoricalArcFace

Who will benefit from this feature?

keras practitioners.

Contributing

• Briefly describe your candidate solution(if contributing):
  • https://github.com/4uiiurz1/keras-arcface

[LukeWood]

When we do include this I do think it’s important to have a tutorial showing off how to use it, so let’s keep that in mind! Doesn’t have to be anything crazy, just some basics!

[sayakpaul]

I concur. Especially for retrieval tasks, ArcFace loss is greatly beneficial.