Surface defect classification of steels with a new semi-supervised learning method

•A semi-supervised learning method named CAE-SGAN is proposed to classify surface defects of steels.•CAE-SGAN improves the performance of SGAN with limited training samples.•When training the discriminator of SGAN, the decoder network of CAE is not truncated.•CAE-SGAN is tested with sample images co...

Celý popis

Uložené v:

Podrobná bibliografia
Vydané v:	Optics and lasers in engineering Ročník 117; s. 40 - 48
Hlavní autori:	Di, He, Ke, Xu, Peng, Zhou, Dongdong, Zhou
Médium:	Journal Article
Jazyk:	English
Vydavateľské údaje:	Elsevier Ltd 01.06.2019
Predmet:	Convolutional autoencoder Defect detection Generative adversarial networks Semi-supervised learning Surface inspection Generative adversarial networks Surface inspection Semi-supervised learning Convolutional autoencoder Defect detection
ISSN:	0143-8166, 1873-0302
On-line prístup:	Získať plný text
Tagy:	Pridať tag Žiadne tagy, Buďte prvý, kto otaguje tento záznam!

Popis
Shrnutí:	•A semi-supervised learning method named CAE-SGAN is proposed to classify surface defects of steels.•CAE-SGAN improves the performance of SGAN with limited training samples.•When training the discriminator of SGAN, the decoder network of CAE is not truncated.•CAE-SGAN is tested with sample images collected from three different steel production lines.•CAE-SGAN provides a better way to apply deep learning methods to some industrial scenes. Defect inspection is extremely crucial to ensure the quality of steel surface. It affects not only the subsequent production, but also the quality of the end-products. However, due to the rare occurrence and appearance variations of defects, surface defect identification of steels has always been a challenging task. Recently, deep learning methods have shown outstanding performance in image classification, especially when there are enough training samples. Since most sample images of steel surface are unlabeled, a new semi-supervised learning method is proposed to classify surface defects of steels. The new method is named CAE-SGAN, as it is based on Convolutional Autoencoder (CAE) and semi-supervised Generative Adversarial Networks (SGAN). CAE-SGAN first trains a stacked CAE through massive unlabeled data. Considering the appearance variations of defects, the passthrough layer is used to help CAE extract fine-grained features. After CAE is trained, the encoder network of CAE is reserved as the feature extractor and fed into a softmax layer to form a new classifier. SGAN is introduced for semi-supervised learning to further improve the generalization ability of the new method. The classifier is trained with images collected from real production lines and images randomly generated by SGAN. Extensive experiments are carried out with samples captured from different steel production lines, and the results indicate that CAE-SGAN had yielded best performances compared with traditional methods. Especially for hot rolled plates, the classification rate is improved by around 16%.
ISSN:	0143-8166 1873-0302
DOI:	10.1016/j.optlaseng.2019.01.011