A Coarse-Grained Regularization Method of Convolutional Kernel for Molten Pool Defect Identification

Abstract

Machine vision has a wide range of applications in the field of welding. The rise of convolutional neural network (CNN) provides a new way to extract visual features of welding. Due to the limitation of the small size of our molten pool dataset, the regularization of the CNN model is necessary to prevent overfitting. We propose a coarse-grained regularization method for convolution kernels (CGRCKs), which is designed to maximize the difference between convolution kernels in the same layer. The algorithm performance was tested on our self-made dataset and other public datasets. The results show that the CGRCK method can extract multi-faceted features. Compared with L1 or L2 regularization, the proposed method works great on CNNs and introduces little overhead cost to the training.