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.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (21) : 8993-9001 DOI: 10.12404/j.issn.1671-1815.2406155
research-article
Retinal Vessel Segmentation Network Based on Self-attention Mechanism and Lateral Output Loss Function
Author information +
History +
PDF

Abstract

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.

Keywords

UNet / Swin-UNet / group normalization / lateral output layers / retinal blood vessel segmentation

Cite this article

Download citation ▾
Zhen-hua YU, Ben-cong YAN, Ying-mei WANG. Retinal Vessel Segmentation Network Based on Self-attention Mechanism and Lateral Output Loss Function. Science Technology and Engineering, 2025, 25(21): 8993-9001 DOI:10.12404/j.issn.1671-1815.2406155

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (11690KB)

20

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉