SelectSeg: Uncertainty-based selective training and prediction for accurate crack segmentation under limited data and noisy annotations

Chen Zhang, Mahdi Bahrami, Dhanada K. Mishra, Matthew M.F. Yuen, Yantao Yu and Jize Zhang

Reliability Engineering and System Safety, 2025, vol. 259, issue C

Abstract: The performance of deep learning models in crack segmentation heavily depends on the availability of large-scale, pixel-wise annotated datasets. However, such annotation is costly to acquire, and can be noisy due to the complexity of crack patterns and the subjectivity of human annotators. To obtain accurate crack segmentation models under noisy annotations, we propose SelectSeg – a four-stage uncertainty-based framework. First, we start with training a deep ensemble of segmentation models to capture the crack prediction uncertainties. Secondly, an uncertainty-based filtering mechanism identifies possibly noisy annotations. Thirdly, semi-supervised learning leverages the information from both reliably annotated data (labeled) and unreliably annotated data (unlabeled) to retrain the segmentation model. Finally, a selective prediction mechanism allows the model to abstain from making predictions on challenging cases, enhancing the overall workflow reliability. Experimental results on real-world crack datasets demonstrate that SelectSeg outperforms existing methods in noisy annotation scenarios. Both selective training and prediction bring significant accuracy improvement.

Keywords: Crack segmentation; Noisy label; Deep ensembles; Uncertainty quantification; Semi-supervised learning; Selective prediction (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832025001127
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:259:y:2025:i:c:s0951832025001127

DOI: 10.1016/j.ress.2025.110909

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().