Multi-level joint distributed alignment-based domain adaptation for cross-scenario strip defect recognition
Kun Liu,Ying Yang,Xiaosong Yang,Jingkai Wang,Weipeng Liu,Haiyong Chen
DOI: https://doi.org/10.1007/s10845-024-02344-z
IF: 8.3
2024-03-29
Journal of Intelligent Manufacturing
Abstract:Deep learning-based methods for object recognition under closed situation have achieved remarkable progress. However, the data distribution of cross-scenario images of strip steel defects varies considerably due to imaging factors such as camera type, parameters, and unstable illumination in different scenarios. Large domain gap and limited diversity may jeopardize the generalization of model over the cross-scenario target domain. In this article, we propose a novel Multi-Level Joint Distribution Alignment Domain Adaptation model (MLJDA). The model unites different granularity perspectives from pixel-, feature-, and category-level to align cross-domain data. First, we propose a parameter sharing style-expansion module based on pixel-level feature mapping, which synthesizes multi-source domain data with different distributions compared to source domain to improve the generalization performance of model towards unknown scenarios and synthesizes similar multi-target domain data. Second, the generated multi-source domain and multi-target domain data are subjected to feature-level adversarial learning to obtain domain-invariant representations. Finally, we design multi-domain collaborative pseudo-label verification based on a voting mechanism to purify cross-scenario target domain category labels. Experimental results on the Handan Steel Defect Dataset and the public Severstal Steel Defect Dataset show that MLJDA can effectively solve the problem of recognizing strip steel defects across scenarios. The MLJDA achieve a F1 score of 98%$$\%$$ and 90%$$\%$$ on the Handan Steel Defect Dataset and Severstal Steel Defect, respectively.
engineering, manufacturing,computer science, artificial intelligence