激光与光电子学进展, 2020, 57 (14): 141501, 网络出版: 2020-07-28
基于集成卷积神经网络的面部表情分类 下载: 1799次
Facial Expression Classification Based on Ensemble Convolutional Neural Network
机器视觉 人脸表情识别 卷积神经网络 集成学习 machine vision facial expression recognition convolution neural network ensemble learning
摘要
针对传统机器学习中人工提取特征复杂度高,以及单卷积网络提取特征不充分导致识别率不高的问题,提出了一种基于集成卷积神经网络的面部表情识别新方法。该方法是将VGGNet-19改进后的VGGNet-19GP模型和ResNet-18模型进行集成,构建了集成网络(EnsembleNet)模型。该模型首先在训练集上对单模型进行训练,使单模型达到实验最优,然后在测试集上进行集成测试。在FER2013和CK+数据集上分别获得了73.854%和97.611%的平均准确率。与VGGNet-19GP和ResNet-18模型以及现有方法进行对比,结果表明,基于集成的面部表情分类方法具有分类更加准确和泛化能力更强的优点。
Abstract
In view of the high complexity of artificial feature extraction in traditional machine learning and the low recognition rate caused by inadequate feature extraction in single convolutional network, a new facial expression recognition method based on ensemble convolutional neural network is proposed. The method is to construct an ensemble network (EnsembleNet) model based on integrating an improved VGGNet-19GP model after VGGNet-19 with a ResNet-18 model. The model first trains a single model on the training set to make the single model reach the optimal experiment. Then the ensemble test is performed on the testing set. The average accuracy of 73.854% and 97.611% are obtained on FER2013 and CK+ datasets, respectively. By comparison with the VGGNet-19GP and ResNet-18 models and other existing methods, it is shown that the ensemble-based facial expression classification method has the advantages of more accurate classification and stronger generalization ability.
周涛, 吕晓琪, 任国印, 谷宇, 张明, 李菁. 基于集成卷积神经网络的面部表情分类[J]. 激光与光电子学进展, 2020, 57(14): 141501. Tao Zhou, Xiaoqi Lü, Guoyin Ren, Yu Gu, Ming Zhang, Jing Li. Facial Expression Classification Based on Ensemble Convolutional Neural Network[J]. Laser & Optoelectronics Progress, 2020, 57(14): 141501.