Application of an improved sparrow search algorithm in BP network classification of strip steel surface defect images
Guoyuan Ma,Xiaofeng Yue,Xueliang Gao,Fuqiuxuan Liu
DOI: https://doi.org/10.1007/s11042-022-13757-4
IF: 2.577
2022-09-22
Multimedia Tools and Applications
Abstract:In this paper, the BP neural network optimized by an improved sparrow search algorithm is proposed (WLT-SSA-BP), which performs excellently in the classification task of strip steel defect images. In the current work, an adaptive weight is introduced into the WLT-SSA algorithm, and the weight is affected by the number of population iterations and the size of the fitness value, which enhances the global search ability of the algorithm; A Levy flight strategy is introduced into WLT-SSA, which helps the algorithm jump out of the local optimal solution through a short-distance local search and occasional longer-distance walking; At the same time, the population t distribution strategy is introduced into the population initialization, which effectively prevents the aggregation of the population. Finally, the proposed WLT-SSA and BP neural network are modeled. The experimental results show that the accuracy rate of WLT-SSA-BP in classifying defect images can reach 96.62%, which is comparable to some other typical heuristic algorithm-optimized BP networks. Its accuracy, precision, recall, specificity, and F1 score were increased by 0.5%–4.66%, 2.62%–8.1%, 1.71%–7.77%, 0.72%–1.33%, 1.59%–8.73%, respectively.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering