Skip to main content
Article
100%
HD

Traditional tunnel lining inspections rely on human experts, making them slow and subjective. To overcome these limits, researchers designed CBAM‑TransUNet, an AI model that merges two cutting‑edge technologies: Vision Transformer (ViT) and UNet, enhanced with an attention module called CBAM. ViT scans the whole image as a puzzle of small patches, capturing global relationships, while UNet reconstructs pixel‑wise maps to locate leak spots. CBAM, placed in the encoder‑decoder skip connections, helps the network focus on the most relevant details, ignoring shadows, stains, or faint traces.
Tests showed impressive results: IoU 0.81, Dice 0.84, recall 0.95, precision 0.85, and an almost perfect accuracy of 0.98. Even in challenging conditions like deep shadows or dirty stains, the model kept high performance. Ablation studies proved that both ViT and CBAM are crucial: removing them dropped IoU by more than 6 %. Score‑CAM heatmaps also revealed that CBAM broadens the network’s attention area, making detection more robust.
This approach could make tunnel inspections faster, objective, and reliable, paving the way for automated monitoring systems in complex environments.

Bonus
100%
HD

The CBAM‑TransUNet model relies on a hybrid fusion of convolutional and transformer architectures. The UNet encoder uses convolutional blocks to extract local features at multiple scales, while the ViT, placed in the bottleneck, processes sequences of linear patches via self‑attention mechanisms, enabling the model to capture long‑range dependencies between distant image regions. The CBAM (Convolutional Block Attention Module) combines two sub‑modules: Channel Attention, which weights activation maps per channel, and Spatial Attention, which highlights the most informative spatial positions. These mechanisms are inserted into the skip‑connections, enhancing the propagation of relevant information from the encoding to the decoding stage.
During training, the model was optimized with a combined Dice and Cross‑Entropy loss, promoting both small‑region segmentation and numerical stability. Using Score‑CAM for attention visualization confirmed that CBAM broadens the area of interest, reducing false negatives amid background noise. Ablation studies showed that removing the ViT lowers IoU by about 7 %, highlighting the importance of modeling global context, while omitting CBAM reduces IoU by 6 %, confirming the added value of channel and spatial attention.
The model achieved an AUC of 0.9746, indicating excellent discriminative ability, and an MCC of 0.8568, measuring classification quality even with imbalanced classes. These results suggest that CBAM‑TransUNet can be integrated into real‑time monitoring systems, leveraging edge GPUs for fast inference, and potentially extended to other complex segmentation tasks such as crack detection in civil infrastructure or anomaly diagnosis in medical imaging.

Ask AI

Explore key aspects of this story with 3 quick insights

Source

Authors:
Zelong LiLi WanYimin WuRenjie SongShuai ShaoHaiping Wu
Date Published:
Original Article www.mdpi.com
Fact-check verified on 13/04/2026
Read our Editorial Guidelines & AI Ethics Policy
Creative Commons License CC BY-NC-SA 4.0

Keep exploring

Discover the full archive of science bites or head back to home

Riconnessione al server... Aggiornamento della sessione in corso
Connessione interrotta Impossibile stabilire il collegamento
Sessione aggiornata Ricarica per visualizzare i contenuti