Algorithm of overfitting avoidance in CNN based on maximum pooled and weight decay
Published Online: Aug 22, 2022
Page range: 965 - 974
Received: Oct 25, 2021
Accepted: May 15, 2022
DOI: https://doi.org/10.2478/amns.2022.1.00011
Keywords
© 2022 Guanzhan Li et al., published by Sciendo
This work is licensed under the Creative Commons Attribution 4.0 International License.
This paper aims to eradicate the poor performance of the convolutional neural network (CNN) for intelligent analysis and detection in samples. Moreover, to avoid overfitting of the CNN model during the training process, an algorithm is proposed for the fusion of maximum pooled and weight decay. Firstly, the maximum pooled method for the pooling layer is explored after mask processing to reduce the number of irrelevant neurons. Secondly, when updating the neuron weight parameters, the weight decay is introduced to further cut down complexity in model training. The experimental comparison shows that the overfitting avoidance algorithm can reduce the detection error rate by more than 10% in image detection than other methods, and it has better generalisation.