Strategic SA-UNet: Integrating self-attention blocks into U-Net for efficient crack segmentation

Ryota Kobayashi, Munehiro Kimura, Ryosuke Harakawa, Norrima Mokhtar, Yang Zhou, Muhammad Amirul Aiman Asri, Raza Ali and Masahiro Iwahashi

PLOS ONE, 2026, vol. 21, issue 3, 1-15

Abstract: Accurate crack segmentation plays a crucial role in ensuring safety and mitigating disaster risks during road inspections and structural health monitoring. However, traditional image processing techniques often struggle with low detection accuracy and poor generalization performance due to the diverse morphology of cracks and the presence of background noise. To address these challenges, MixSegNet, a model that combines the strengths of convolutional neural networks (CNNs) and Transformers, has been proposed and demonstrated to achieve high segmentation performance. However, this enhanced precision comes at the cost of prolonged training cycles, which limits its applicability in operational environments such as infrastructure inspection, where new data must be acquired and processed continuously and rapidly. In this paper, to address this limitation, we propose Strategic SA-UNet (Strategically Integrated Self-Attention U-Net), a novel crack segmentation network. The model strategically integrates a computationally efficient U-Net based CNN with a Self-Attention Block between the encoder and decoder to effectively fuse local features with global context, thereby maintaining high segmentation accuracy while reducing training time and computational cost. Experimental evaluations on publicly available datasets demonstrate that Strategic SA-UNet achieves segmentation accuracy comparable to MixSegNet, while reducing training time by 83%, Floating Point Operations (FLOPs) by 63%, and Model Parameters by 96%. Furthermore, Strategic SA-UNet achieves a high mean Intersection over Union (mIoU) even with a small number of epochs, highlighting its superior training efficiency. These results suggest that Strategic SA-UNet is an efficient segmentation model, especially suitable for real-time infrastructure inspection and structural monitoring applications.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0343162 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 43162&type=printable (application/pdf)

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:plo:pone00:0343162

DOI: 10.1371/journal.pone.0343162

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().