Affiliation:
1. Shanghai Key Laboratory of Multidimensional Information Processing, School of Communication Electronic Engineering East China Normal University Shanghai China
2. Department of Biochemistry National University of Singapore Singapore
3. Naval Medical Department Naval Medical University Shanghai China
4. College of Acupuncture‐Moxibustion and Tuina Shanghai University of Traditional Chinese Medicine Shanghai China
5. Artificial Intelligence Innovation Center (AIIC) Midea Group Shanghai China
6. School of Aeronautics and Astronautics Shanghai Jiao Tong University Shanghai China
Abstract
ABSTRACTAccurate segmentation of retinal vessels is crucial for the early diagnosis and treatment of eye diseases, for example, diabetic retinopathy, glaucoma, and macular degeneration. Due to the intricate structure of retinal vessels, it is essential to extract their features with precision for the semantic segmentation of medical images. In this study, an improved deep learning neural network was developed with a focus on feature extraction based on the U‐Net structure. The enhanced U‐Net combines the architecture of convolutional neural networks (CNNs) with SE blocks (squeeze‐and‐excitation blocks) to adaptively extract image features after each U‐Net encoder's convolution. This approach aids in suppressing nonvascular regions and highlighting features for specific segmentation tasks. The proposed method was trained and tested on the DRIVECHASE_DB1 and STARE datasets. As a result, the proposed model had an algorithmic accuracy, sensitivity, specificity, Dice coefficient (Dc), and Matthews correlation coefficient (MCC) of 95.62/0.9853/0.9652, 0.7751/0.7976/0.7773, 0.9832/0.8567/0.9865, 82.53/87.23/83.42, and 0.7823/0.7987/0.8345, respectively, outperforming previous methods, including UNet++, attention U‐Net, and ResUNet. The experimental results demonstrated that the proposed method improved the retinal vessel segmentation performance.
Funder
National Natural Science Foundation of China