基于自注意力机制和侧输出损失函数的视网膜血管分割网络

于振华 ,  闫本聪 ,  王迎美

科学技术与工程 ›› 2025, Vol. 25 ›› Issue (21) :8993-9001

PDF (11690KB)
科学技术与工程 ›› 2025, Vol. 25 ›› Issue (21) :8993-9001 DOI: 10.12404/j.issn.1671-1815.2406155
自动化技术、计算机技术

基于自注意力机制和侧输出损失函数的视网膜血管分割网络

作者信息 +

Retinal Vessel Segmentation Network Based on Self-attention Mechanism and Lateral Output Loss Function

Author information +
文章历史 +

摘要

图像分割是医学图像分析中的一个基本问题,基于深度学习的典型UNet架构(UNet architecture)分割网络模型及其变式被广泛应用于视网膜血管分割之中。但是UNet网络通过局部卷积模块提取图像的特征信息,难以关联图像的全局信息,无法有效捕捉像素之间的长距离依赖关系。考虑到UNet网络模型存在的问题和视网膜血管图像的特点,在UNet跳跃连接中加入注意力模块,可以捕捉血管之间的长距离依赖关系。此外,为增强网络的分割能力,使用群归一化(group normalization, GN)代替UNet网络模型原始的批归一化(batch normalization, BN),对不同的通道选择对应的分组。为更新参数和优化网络,利用侧输出层和最后的输出层设计了交叉熵损失函数。在DRIVE数据集和CHASEDB1数据集上进行了实验,实验结果表明所提出的模型有更好的图像分割效果。

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.

关键词

UNet / Swin-UNet / 群归一化 / 侧输出层 / 视网膜血管分割

Key words

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

引用本文

引用格式 ▾
于振华, 闫本聪, 王迎美. 基于自注意力机制和侧输出损失函数的视网膜血管分割网络[J]. 科学技术与工程, 2025, 25(21): 8993-9001 DOI:10.12404/j.issn.1671-1815.2406155
YU Zhen-hua, YAN Ben-cong, WANG Ying-mei. Retinal Vessel Segmentation Network Based on Self-attention Mechanism and Lateral Output Loss Function[J]. Science Technology and Engineering, 2025, 25(21): 8993-9001 DOI:10.12404/j.issn.1671-1815.2406155

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

山东省自然科学基金面上项目(ZR2022MA027)

PDF (11690KB)

41

访问

0

被引

详细

导航
相关文章

/

〈 〉