
科学技术与工程

中国技术经济学会
Retinal Vessel Segmentation Network Based on Self-attention Mechanism and Lateral Output Loss Function
Zhen-hua YU , Ben-cong YAN , Ying-mei WANG
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (21) : 8993 -9001.
Image segmentation is a fundamental problem in medical image analysis, the typical deep learning based UNet architecture (UNet) and its variants are widely used in retinal vessel segmentation. However, the UNet network extracts feature information from images through local convolution modules, which makes the global information of the images difficult to be correlated and the long-distance dependencies between pixels difficult to be effectively captured. Considering the problems with the UNet network model and the characteristics of retinal vascular images, an attention module was added to the skip connections of UNet to capture long-distance dependencies between blood vessels. In addition, to enhance the segmentation ability of the network, the group normalization(GN) was used instead of the original batch normalization (BN) of the UNet network model, and the corresponding groups were selected for different channels. To update parameters and optimize the network, the final cross entropy loss function was designed using the side output layer and the final output layer. Experiments are implemented on the DRIVE dataset and CHASEDB1 dataset, and the experimental results show that the proposed model has better image segmentation performance.
UNet / Swin-UNet / group normalization / lateral output layers / retinal blood vessel segmentation
/
| 〈 |
|
〉 |