arXiv · 1707.07391
Contrastive-center loss for deep neural networks
Abstract
The deep convolutional neural network(CNN) has significantly raised the performance of image classification and face recognition. Softmax is usually used as supervision, but it only penalizes the classification loss. In this paper, we propose a novel auxiliary supervision signal called contrastivecenter loss, which can further enhance the discriminative power of the features, for it learns a class center for each class. The proposed contrastive-center loss simultaneously considers intra-class compactness and inter-class separability, by penalizing the contrastive values between: (1)the distances of training samples to their corresponding class centers, and (2)the sum of the distances of training samples to their non-corresponding class centers. Experiments on different datasets demonstrate the effectiveness of contrastive-center loss.
Explore related subjects
Keep this discovery
Ce Qi, Fei Su. 2017-07-24. Contrastive-center loss for deep neural networks. https://arxiv.org/abs/1707.07391
Cite the original work for its findings. Save a collection to share your selection of sources.