Efficient retinal artery/vein classification with dense color-invariant feature learning

Xiaojuan Chen,Luyu Niu,Song Guo
DOI: https://doi.org/10.1007/s00521-024-10696-z
2024-11-25
Neural Computing and Applications
Abstract:Automatic classification of retinal arteries and veins (A/V) is crucial for assisting clinicians in diagnosing cardiovascular and eye diseases. Deep learning models have been widely employed for A/V classification, achieving remarkable performance. However, two primary challenges persist: vessel discontinuity and A/V confusion. To address the vessel discontinuity challenge, we have designed a multi-scale convolution block (MCB) that combines square and strip convolutions. This design enhances the segmentation of tiny vessels by capturing more discriminative vessel features, effectively resolving the issue of vessel discontinuity. Meanwhile, to tackle the A/V confusion challenge, we propose a dense color-invariant feature learning (DCIFL) method to improve the model's robustness to color variations in retinal images. Specifically, DCIFL uses KL divergences to align the distributions of both latent representations and prediction maps of the raw input image and its color-transformed counterpart. By integrating MCB and DCIFL, we introduce EAV-Net, an efficient model for retinal A/V classification. The proposed method has been tested on three publicly available datasets, achieving accuracies of 96.59%, 97.28%, and 99.34% on the AV-DRIVE, WIDE, and HRF datasets, respectively. These results demonstrate the superiority of our approach in outperforming state-of-the-art methods. Our code will be released at https://github.com/guomugong/EAV-Net.
computer science, artificial intelligence
What problem does this paper attempt to address?