TD Swin-UNet: Texture-Driven Swin-UNet with Enhanced Boundary-Wise Perception for Retinal Vessel Segmentation

被引:1
作者
Li, Angran [1 ]
Sun, Mingzhu [1 ]
Wang, Zengshuo [1 ]
机构
[1] Nankai Univ, Coll Artificial Intelligence, Tianjin 300350, Peoples R China
来源
BIOENGINEERING-BASEL | 2024年 / 11卷 / 05期
基金
国家重点研发计划;
关键词
retinal vessel segmentation; transformer; cross-level texture complementary module; pixel-wise texture highlighting module; Hausdorff loss;
D O I
10.3390/bioengineering11050488
中图分类号
Q81 [生物工程学(生物技术)]; Q93 [微生物学];
学科分类号
071005 ; 0836 ; 090102 ; 100705 ;
摘要
Retinal vessel segmentation plays a crucial role in medical image analysis, aiding ophthalmologists in disease diagnosis, monitoring, and treatment guidance. However, due to the complex boundary structure and rich texture features in retinal blood vessel images, existing methods have challenges in the accurate segmentation of blood vessel boundaries. In this study, we propose the texture-driven Swin-UNet with enhanced boundary-wise perception. Firstly, we designed a Cross-level Texture Complementary Module (CTCM) to fuse feature maps at different scales during the encoding stage, thereby recovering detailed features lost in the downsampling process. Additionally, we introduced a Pixel-wise Texture Swin Block (PT Swin Block) to improve the model's ability to localize vessel boundary and contour information. Finally, we introduced an improved Hausdorff distance loss function to further enhance the accuracy of vessel boundary segmentation. The proposed method was evaluated on the DRIVE and CHASEDB1 datasets, and the experimental results demonstrate that our model obtained superior performance in terms of Accuracy (ACC), Sensitivity (SE), Specificity (SP), and F1 score (F1), and the accuracy of vessel boundary segmentation was significantly improved.
引用
收藏
页数:18
相关论文
共 41 条
[1]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[2]  
Cao H., 2021, arXiv
[3]   MFA-UNet: a vessel segmentation method based on multi-scale feature fusion and attention module [J].
Cao, Juan ;
Chen, Jiaran ;
Gu, Yuanyuan ;
Liu, Jinjia .
FRONTIERS IN NEUROSCIENCE, 2023, 17
[4]   Retinal Vessel Segmentation Using Deep Learning: A Review [J].
Chen, Chunhui ;
Chuah, Joon Huang ;
Ali, Raza ;
Wang, Yizhou .
IEEE ACCESS, 2021, 9 :111985-112004
[5]   LMSA-Net: A lightweight multi-scale aware network for retinal vessel segmentation [J].
Chen, Jian ;
Wan, Jiaze ;
Fang, Zhenghan ;
Wei, Lifang .
INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2023, 33 (05) :1515-1530
[6]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
[7]   UNet retinal blood vessel segmentation algorithm based on improved pyramid pooling method and attention mechanism [J].
Du, Xin-Feng ;
Wang, Jie-Sheng ;
Sun, Wei-zhen .
PHYSICS IN MEDICINE AND BIOLOGY, 2021, 66 (17)
[8]   CE-Net: Context Encoder Network for 2D Medical Image Segmentation [J].
Gu, Zaiwang ;
Cheng, Jun ;
Fu, Huazhu ;
Zhou, Kang ;
Hao, Huaying ;
Zhao, Yitian ;
Zhang, Tianyang ;
Gao, Shenghua ;
Liu, Jiang .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (10) :2281-2292
[9]   Fundus Retinal Vessels Image Segmentation Method Based on Improved U-Net [J].
Han, J. ;
Wang, Y. ;
Gong, H. .
IRBM, 2022, 43 (06) :628-639
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778