Cost-effective Image Recognition of Water Leakage in Metro Tunnels Using Self-Supervised Learning

Yining Gu,Qing Ai,Zhengwen Xu,Lianbi Yao,Hui Wang,Xingchun Huang,Yong Yuan
DOI: https://doi.org/10.1016/j.autcon.2024.105678
IF: 10.3
2024-01-01
Automation in Construction
Abstract:Supervised learning (SL) methods have achieved good performance in image recognition of water leakage in tunnels, but they require a large amount of labeled data. This paper adopts a self-supervised learning (SSL) approach to realize cost-effective image recognition of water leakage in metro tunnels to reduce the necessary labeling effort. The Bootstrap Your Own Latent (BYOL) for contrastive learning, a framework of SSL method, is used to classify tunnel images with only 10% labeled data, and the model is transferred to two other tunnel datasets to analyze its ability for practical application. The results show that the proposed SSL method outperforms SL method with the most cost-effective data labeling proportion of approximately 10%, and the transferability of the SSL model is also significantly better than that of the SL model. This method is promising for reducing the labeling effort and improving the transferability of tunnel image recognition.
What problem does this paper attempt to address?