Hyperparameter optimization of convolutional neural network using grey wolf optimization for facial emotion recognition

Abstract

Facial emotion recognition (FER) is a challenging task in computer vision with wide applications in areas such as human-computer interaction, security, and healthcare. To improve the performance of convolutional neural networks (CNN) in FER, a novel approach combining CNN with grey wolf optimization (GWO) was proposed to optimize key hyperparameters. The CNN-GWO model was fine-tuned by adjusting hyperparameters such as the number of convolutional layers, kernel size, number of filters, and learning rate. This model was evaluated using the CK+ dataset and achieved an accuracy of 90.97%, demonstrating its competitive performance compared to existing methods. The optimized hyperparameters included three convolutional layers, 35 filters, a kernel size of 5, a learning rate of 0.045990, a dropout rate of 0.4988, and a max pooling size of 3. These results confirm that GWO is effective in optimizing CNN for FER tasks, providing an efficient solution to enhance model accuracy. This approach shows promising potential for future FER applications, highlighting GWO as a valuable optimization technique for CNN architectures.