A Novel Face Recognition Algorithm for Imbalanced Small Samples

Received: 28 January 2020 Accepted: 10 April 2020 Deep learning (DL) has become a hotspot in the research of image recognition. However, the DL strategy must be trained with lots of samples that are distributed evenly across classes, i.e. subjected to balanced distribution. Therefore, this paper attempts to design a method to satisfactorily recognize faces in imbalanced small samples. Firstly, the deep convolutional generative adversarial network (DCGAN) was improved to generate data samples with similar distribution as the original training data, creating a balanced training set of sufficient labelled samples. Then, transfer learning was performed to transform the AlexNet, which is pretrained on big dataset, to the balanced target dataset of small samples. Next, the previous convolutional layer was frozen as a feature extractor, and the truncated normal distribution was reinitialized for the next fully-connected layer. Simulations on face recognition show that our method achieved higher recognition rate and less serious overfitting than ordinary CNNs.

[1]  Mark Braverman,et al.  Poly-logarithmic independence fools bounded-depth boolean circuits , 2011, Commun. ACM.

[2]  Vytautas Perlibakas,et al.  Distance measures for PCA-based face recognition , 2004, Pattern Recognit. Lett..

[3]  Christian Igel,et al.  Big Universe, Big Data: Machine Learning and Image Analysis for Astronomy , 2017, IEEE Intelligent Systems.

[4]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[5]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[6]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[7]  Song Gao,et al.  A New Hybrid Method in Global Dynamic Path Planning of Mobile Robot , 2018, Int. J. Comput. Commun. Control.

[8]  Yoshua Bengio,et al.  Generative Adversarial Networks , 2014, ArXiv.

[9]  Satyanarayana Teki,et al.  Brain Tumour Segmentation Using U-net Based Adversarial Networks , 2019, Traitement du Signal.

[10]  Paulo Lobato Correia,et al.  Face Recognition: A Novel Multi-Level Taxonomy based Survey , 2019, IET Biom..

[11]  Xiaoru Song,et al.  Stabilization precision control methods of photoelectric aim-stabilized system , 2015 .

[12]  Alberto Del Bimbo,et al.  Investigating Nuisance Factors in Face Recognition with DCNN Representation , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[13]  Shu Zhan,et al.  Robust face detection using local CNN and SVM based on kernel combination , 2016, Neurocomputing.

[14]  Steven C. H. Hoi,et al.  Face Detection using Deep Learning: An Improved Faster RCNN Approach , 2017, Neurocomputing.

[15]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[16]  Zhengming Ma,et al.  Face recognition based on the fusion of global and local HOG features of face images , 2014, IET Comput. Vis..

[17]  Zhang Jin.  Research on Face Recognition Technology Based on Sub-Image Feature Combination , 2007 .

[18]  Xiaogang Wang,et al.  Deep Learning Face Representation by Joint Identification-Verification , 2014, NIPS.

[19]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[20]  Stephen Balaban,et al.  Deep learning and face recognition: the state of the art , 2015, Defense + Security Symposium.

[21]  Tara N. Sainath,et al.  Deep Neural Networks for Acoustic Modeling in Speech Recognition: The Shared Views of Four Research Groups , 2012, IEEE Signal Processing Magazine.

[22]  Mohammed Bennamoun,et al.  Cost-Sensitive Learning of Deep Feature Representations From Imbalanced Data , 2015, IEEE Transactions on Neural Networks and Learning Systems.

[23]  Tetsuya Ogata,et al.  Audio-visual speech recognition using deep learning , 2014, Applied Intelligence.

[24]  Yun Zhang,et al.  Face recognition based on wavelet transform and SVM , 2005, 2005 IEEE International Conference on Information Acquisition.