KappaFace: Adaptive Additive Angular Margin Loss for Deep Face Recognition
Chingis Oinar, Binh M. Le, Simon S. Woo
Abstract
Feature learning is a widely used method for large-scale face recognition tasks. Recently, large-margin softmax loss methods have demonstrated significant improvements in deep face recognition. However, these methods typically propose fixed positive margins to enforce intra-class compactness and inter-class diversity, without considering imbalanced learning issues that arise due to different learning difficulties or the number of training samples available in each class. This overlook not only compromises the efficiency of the learning process but, more critically, the generalization capability of the resultant models. To address this problem, we introduce a novel adaptive strategy called KappaFace, which modulates the relative importance of each class based on its learning difficulty and imbalance. Drawing inspiration from the von Mises-Fisher distribution, KappaFace increases the margin values for the challenging or underrepresented classes and decreases that of more well-represented classes. Comprehensive experiments across eight cutting-edge baselines and nine well-established facial benchmark datasets strongly confirm the advantage of our method. Notably, we observed an enhancement of up to 0.5% on the verification task when evaluated on the IJB-B/C datasets. In conclusion, KappaFace offers a novel solution that effectively tackles imbalanced learning in deep face recognition tasks and establishes a new baseline.