Understanding the effect of transfer learning on the automatic welding defect detection

Xiaopeng Wang, Xinghua Yu*

*此作品的通讯作者

科研成果: 期刊稿件文章同行评审

11 引用 (Scopus)

摘要

The welding defect dataset is difficult to collect due to its cost and time-consuming, which is overcome by the transfer learning method in this work. In detail, the models are initialized with the weights and biases in the pre-trained model and then compared with the model trained from scratch. The results show that the transferred weight improves the model's accuracy from 92.76% to 96.70%, but the transferred bias reduces the model's accuracy from 92.76% to 91.0%. The analysis of the feature map suggests that the transferred weight increases the variance of the feature map and improves the contrast between each channel of feature maps. As a method to validate whether the transfer learning could help detect the pixels of welding defects more accurately, the gradient class activation map (Grad-CAM) is used to track the important pixels in radiographic images for the model's predicted results. The results show that the transfer learning method enhances the model's attention to the pixels of the welding defect.

源语言英语
文章编号102784
期刊NDT and E International
134
DOI
出版状态已出版 - 3月 2023

指纹

探究 'Understanding the effect of transfer learning on the automatic welding defect detection' 的科研主题。它们共同构成独一无二的指纹。

引用此