loading page

A Classification Algorithm Based on Improved Attention Mechanism and Residual Network
  • +2
  • Hai Tan,
  • Qiang Shang,
  • Minghua Wan,
  • Guowei Yang,
  • Ying Fang
Hai Tan
Nanjing Audit University
Author Profile
Qiang Shang
Nanjing Audit University

Corresponding Author:[email protected]

Author Profile
Minghua Wan
Nanjing Audit University
Author Profile
Guowei Yang
Nanjing Audit University
Author Profile
Ying Fang
Shangqiu Normal University
Author Profile

Abstract

In recent years, advancements in attention mechanisms and residual networks have significantly increased their application in facial expression classification. However, challenges such as poor key feature extraction and complex model training still exist. To tackle these challenges, this paper introduces a classification algorithm based on improved attention mechanism and residual network. Initially, ResNet50 serves as the backbone network for feature extraction, while the Convolutional Block Attention Module is incorporated to automatically learn and selectively emphasize crucial local features of the input data. Secondly, the residual modules of the backbone network are innovatively constructed to enhance the overall feature extraction effect. Finally, the improved CBAM-ERF, which includes enhancements to the CAM, is incorporated to address the issue of neuron suppression within intervals, thereby accelerating the network’s convergence speed and improving classification efficiency. We conducted experiments using three publicly available facial expression datasets: FER2013, CK+, and RAFDB. Compared to basic methods, the average accuracy increased by 13.04%, 25.67%, and 7.53%, respectively. This method can produce competitive recognition results, demonstrating its effectiveness in facial expression recognition tasks.