CNN Image Classification Algorithm Based on Fusion of Mask and Attention Mechanism

LI Wei,HUANG He-ming
DOI: https://doi.org/10.3969/j.issn.1006-9348.2023.04.070
2023-01-01
Abstract:In order to suppress the over-fitting phenomenon of convolution neural network in the training process of image classification and improve the accuracy of image classification and the stability of algorithm, a CNN image classification algorithm based on fusion of mask andattention mechanism is proposed. The algorithm was used to train the network model based on mask pre-training method, and the denoising neural network was used to extract the denoising deep features as the input of the overall framework of deep CNN network model. Then, experiments were carried out on the standard image database Corel-1000 using different attention activation functions.The experimental results show that using the pre-training method based on mask not only improves the image classification effect, but also makes the performance of the whole model more stable and robust; at the same time, after adding the mixed attention mechanism, the experimental results are better. By reducing the interference of image background noise and deep feature noise, the algorithm can effectively suppress the over-fitting phenomenon in the training process; And different attention activation functions have different effects, among which the mixed attention activation function has the best effect.
What problem does this paper attempt to address?