Skip to main navigation Skip to search Skip to main content

A Unified Lightweight Attention Module for Road Classification and Crack Segmentation

  • Mingwu Li
  • , Yunfei Yin*
  • , Wantong Li
  • , Yuanhao Liu
  • , Abaho G. Gershome
  • , Zejiao Dong
  • *Corresponding author for this work
  • Harbin Institute of Technology
  • University of Rwanda

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Automated health maintenance of transport infrastructure is pivotal for ensuring public safety and operational efficiency within intelligent transportation systems. While deep learning has revolutionized surface distress diagnosis, a critical trade-off persists between diagnostic accuracy and computational efficiency, particularly for deployment on resource-constrained edge devices. To bridge this gap, we propose the anchor gated self-attention (AGSA), a lightweight plug-and-play module. AGSA introduces sparse learnable anchors as semantic hubs to decouple global dependency modeling, reducing computational complexity from quadratic to linear. Furthermore, it incorporates a novel anchor-guided spatial gating mechanism that dynamically fuses global semantics with local spatial details, effectively highlighting fault regions while suppressing environmental noise. Extensive experiments demonstrate the superiority of AGSA across diverse tasks. In road surface classification, AGSA consistently enhances performance across multiple backbones on the large-scale road surface classification dataset benchmark and a challenging self-collected winter dataset, achieving state-of-the-art accuracy in extreme conditions such as ice and snow coverage. Moreover, in dense prediction tasks, AGSA significantly improves crack segmentation precision on the Crack500 dataset, yielding notable gains in IoU and Dice scores for subtle fracture detection. These results confirm AGSA as a scalable, high-performance solution for real-time, automated infrastructure inspection.

Original languageEnglish
Title of host publicationProceedings of 2026 IEEE 15th Data Driven Control and Learning Systems Conference, DDCLS 2026
EditorsMingxuan Sun, Ronghu Chi
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1517-1522
Number of pages6
ISBN (Electronic)9798319521910
DOIs
StatePublished - 2026
Event15th IEEE Data Driven Control and Learning Systems Conference, DDCLS 2026 - Jishou, China
Duration: 8 May 202611 May 2026

Publication series

NameProceedings of 2026 IEEE 15th Data Driven Control and Learning Systems Conference, DDCLS 2026

Conference

Conference15th IEEE Data Driven Control and Learning Systems Conference, DDCLS 2026
Country/TerritoryChina
CityJishou
Period8/05/2611/05/26

Keywords

  • Lightweight attention
  • crack segmentation
  • deep learning
  • road surface classification

Fingerprint

Dive into the research topics of 'A Unified Lightweight Attention Module for Road Classification and Crack Segmentation'. Together they form a unique fingerprint.

Cite this