IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0347765.html

LiteCrackSeg: A lightweight hybrid CNN–transformer for efficient crack segmentation

Author

Listed:
  • Kaleb Amsalu Gobena
  • MD Youshuf Khan Rakib
  • Fiseha Berhanu Tesema
  • Galana Fekadu Asafa
  • Shengbing Ren

Abstract

Infrastructure cracks are critical indicators of structural deterioration in pavements, bridges, and buildings. Automated crack segmentation has therefore become an important component of structural health monitoring systems. However, accurate pixel-level crack segmentation on resource-constrained devices remains challenging due to the thin, low-contrast, and curvilinear morphology of cracks, as well as severe foreground–background class imbalance. To address these challenges, we propose LiteCrackSeg, a lightweight hybrid CNN–transformer architecture designed for efficient and accurate crack segmentation. The proposed framework adopts a hybrid MobileViT encoder that captures both local spatial details and long-range contextual dependencies while maintaining a compact model size. To enhance morphological sensitivity to elongated crack structures, we introduce a Morphology-Aware MobileViT (MAM-ViT) bottleneck, which integrates dual-branch Dynamic Snake Convolutions (DSConv) to align receptive fields with crack trajectories. Furthermore, a transformer-based decoder with local self-attention progressively reconstructs spatial details, while an attention-guided multi-scale fusion strategy improves boundary precision and structural continuity. To mitigate severe class imbalance, the model is trained using the Tversky loss, which explicitly balances false positives and false negatives. Extensive experiments on three public crack segmentation datasets (DeepCrack, CrackMap, and TUT) demonstrate that LiteCrackSeg achieves state-of-the-art segmentation performance while maintaining high computational efficiency. The proposed model requires only 2.72M parameters and 3.23 GFLOPs, achieving real-time inference at 56 FPS on 512 × 512 images, making it suitable for deployment on resource-constrained edge devices for practical infrastructure inspection.

Suggested Citation

  • Kaleb Amsalu Gobena & MD Youshuf Khan Rakib & Fiseha Berhanu Tesema & Galana Fekadu Asafa & Shengbing Ren, 2026. "LiteCrackSeg: A lightweight hybrid CNN–transformer for efficient crack segmentation," PLOS ONE, Public Library of Science, vol. 21(4), pages 1-29, April.
  • Handle: RePEc:plo:pone00:0347765
    DOI: 10.1371/journal.pone.0347765
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0347765
    Download Restriction: no

    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0347765&type=printable
    Download Restriction: no

    File URL: https://libkey.io/10.1371/journal.pone.0347765?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0347765. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.