C2AM-Unet: Coordinate and Channel Attention Mixing Flexible Architecture for Retinal Vessel Segmentation

Jingling Zhang,Shuting Zheng,Wen Zhang,Jia Gu
DOI: https://doi.org/10.1101/2024.11.27.625635
2024-12-03
Abstract:Recently, CNN-based and Transformer-based network have become the de-facto standard for vessel segmentation, due to their strong feature representation capabilities. However, CNNs fall short in explicitly capturing global dependencies between different spatial locations, while Transformers require quadratic computation by virtue of the long sequences and lack fine-grained local correlations. Seeking to model both local correlations and long-range dependencies, our research revisits this issue and introduces a novel retinal vessel segmentation network based on U-Net, the Coordinate and Channel Attention Mixing Flexible Architecture (C2AM-Unet), a new lightweight network that combines the U-Net model with a multi-attention mechanism. Specifically, we present a core Dual Global Coordinate Attention (DGCA) module, which embeds two types of serial global spatial-specific property information: one with a direction-aware mechanism, responsible for encoding long-range dependencies and coordinate information in both horizontal and vertical directions, and another with a transformer-like coordinate module that focuses on the connections and interactions between each spatial location. Next, the C2AM feature maps are aggregated with the direction-aware feature maps to enhance their ability to represent the objects of interest. To further strengthen the long-distance semantic dependencies, we utilize the transformer-like channel attention model to capture the cross-channel communication of global information. To verify the effectiveness of our method, we also construct other similar variant components that can be flexibly applied to other networks. The experimental results on three retinal vessel and two coronary angiography datasets, including DRIVE, STARE, CHASE_DB1, DCA1 and CHUAC, demonstrate that C2AM-Unet achieves competitive scores compared to the state-of-the-art methods.
Bioinformatics
What problem does this paper attempt to address?