HCM: A Hybrid CNN-Mamba Architecture for Semi-Supervised Retinal Vessel Segmentation
Retinal vessel segmentation plays a critical role in the diagnosis and monitoring of ocular and systemic diseases. However, the task remains challenging due to low contrast between vessels and background, complex morphological structures, and the scarcity of annotated data. To address these challenges, we propose HCM, a novel Hybrid CNN-Mamba Architecture for semi-supervised retinal vessel segmentation. HCM integrates the strengths of Convolutional Neural Networks (CNNs) and the Mamba architecture, leveraging CNN's local feature extraction capabilities and Mamba's global dependency modeling. Furthermore, we propose Multi-Scale Contrastive Learning (MSCL), which combines image-level, region-level, and pixel-level contrastive losses to optimize feature learning across different scales. Extensive experiments on the DRIVE, CHASE_DB1, and STARE datasets demonstrate that HCM outperforms existing semi-supervised methods, achieving superior performance in terms of Dice coefficient, sensitivity, and intersection-over-union (IoU). Ablation studies validate the effectiveness of the hybrid architecture and the MSCL, highlighting their contributions to robust and accurate retinal vessel segmentation. Our work provides a promising solution for leveraging limited annotated data and improving segmentation performance in retinal vessel segmentation.
Authors
- Yuefeng Liu
- Yanli Yang (ORCID: https://orcid.org/0009-0003-1377-8217)
- G. Y. Zhang (ORCID: https://orcid.org/0000-0002-0564-3684)
- Weilong Gong
- Xin Liu (ORCID: https://orcid.org/0009-0003-1631-9921)
Institutions
- Shenzhen University (CN)
Publication Details
- Published
- 2025-04-25
- DOI
- https://doi.org/10.1145/3744367.3744431
- Primary Topic
- Retinal Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00