Improved softmax loss for deep learning‐based face and expression recognition
Author(s) -
Zhou Jiancan,
Jia Xi,
Shen Linlin,
Wen Zhenkun,
Ming Zhong
Publication year - 2019
Publication title -
cognitive computation and systems
Language(s) - English
Resource type - Journals
ISSN - 2517-7567
DOI - 10.1049/ccs.2019.0010
Subject(s) - softmax function , china , software , people's republic , computer software , engineering , library science , artificial intelligence , computer science , geography , deep learning , software engineering , programming language , archaeology
In recent years, deep convolutional neural networks (CNN) have been widely used in computer vision and significantly improved the performance of image recognition tasks. Most works use softmax loss to supervise the training of CNN and then adopt the output of last layer as features. However, the discriminative capability of the softmax loss is limited. Here, the authors analyse and improve the softmax loss by manipulating the cosine value and input feature length. As the approach does not change the principle of the softmax loss, the network can easily be optimised by typical stochastic gradient descent. The MNIST handwritten digits dataset is employed to visualise the features learned by the improved softmax loss. The CASIA‐WebFace and FER2013 training set are adopted to train deep CNN for face and expression recognition, respectively. Results on both the LFW dataset and FER2013 test set show that the proposed softmax loss can learn more discriminative features and achieve better performance.
Accelerating Research
Robert Robinson Avenue,
Oxford Science Park, Oxford
OX4 4GP, United Kingdom
Address
John Eccles HouseRobert Robinson Avenue,
Oxford Science Park, Oxford
OX4 4GP, United Kingdom