A feature extraction and classification algorithm based on improved sparse auto-encoder for round steel surface defects.

Abstract

Traditional feature dimensionality reduction (FDR) algorithms can extract features by reducing feature dimensions. However, it may lose some useful information and affect the accuracy of classification. Normally, in traditional defect feature extraction, it first obtain the defect area of the defect image by image preprocessing and defect segmentation, select the original feature set of defects by prior knowledge, and extract the optimal features by traditional FDR algorithms to solve the problem of "curse of dimensionality". In this paper, a feature extraction and classification algorithm based on improved sparse auto-encoder (AE) is proposed. We adopt three traditional FDR algorithms at the same time, combine the defect features obtained in pairs, take the merged defect features as the input of sparse AE, then use the "bottleneck" of sparse AE to conduct the defects classification by Softmax classifier. The experimental results show that the proposed algorithm can extract the optimal features of round steel surface defects with less network training time than individual sparse AE, finally get higher classification accuracy than individual FDR algorithm in the actual production line.