Improving Scattered Defect Grading in Castings Digital Radiographs via Smoothing the One-Hot Encoding

Han Yu,Xingjie Li,Xue Hao,Zhaowei Song,Shangyu Liu,Xinyue Li,Chunyu Hou,Huasheng Xie
DOI: https://doi.org/10.1007/s40962-024-01335-3
2024-04-24
International Journal of Metalcasting
Abstract:Ensuring the precise grading of discontinuities is imperative to guarantee the quality of castings and enhance profitability in casting production. Recent grading methods leveraging computer vision are advanced by performing a single-label image classification or regression, which loses the intrinsically ordinal relationship. Motivated by this observation, we propose a label smoothing technology for ordinal variables to convert the level of each defect instance into a discrete probability distribution, aiming to model the noise label and ordinal relationship. Furthermore, we design a convolutional neural network framework based on multi-task learning. This framework, by simultaneously learning the level label distribution and regressing the level directly, outperforms a single-task network in terms of overall performance. Finally, we construct a casting gas porosity defect grading dataset. Experimental results on this dataset highlight the significant advantages of our proposed method compared to traditional single-label image classification or regression algorithms.
metallurgy & metallurgical engineering
What problem does this paper attempt to address?