MTPA_Unet: Multi-Scale Transformer-Position Attention Retinal Vessel Segmentation Network Joint Transformer and CNN

被引:15
作者
Jiang, Yun [1 ]
Liang, Jing [1 ]
Cheng, Tongtong [1 ]
Lin, Xin [1 ]
Zhang, Yuan [1 ]
Dong, Jinkun [1 ]
机构
[1] Northwest Normal Univ, Coll Comp Sci & Engn, Lanzhou 730070, Peoples R China
基金
中国国家自然科学基金;
关键词
retinal vessel segmentation; convolutional neural network; transformer; attention mechanism; IMAGES;
D O I
10.3390/s22124592
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Retinal vessel segmentation is extremely important for risk prediction and treatment of many major diseases. Therefore, accurate segmentation of blood vessel features from retinal images can help assist physicians in diagnosis and treatment. Convolutional neural networks are good at extracting local feature information, but the convolutional block receptive field is limited. Transformer, on the other hand, performs well in modeling long-distance dependencies. Therefore, in this paper, a new network model MTPA_Unet is designed from the perspective of extracting connections between local detailed features and making complements using long-distance dependency information, which is applied to the retinal vessel segmentation task. MTPA_Unet uses multi-resolution image input to enable the network to extract information at different levels. The proposed TPA module not only captures long-distance dependencies, but also focuses on the location information of the vessel pixels to facilitate capillary segmentation. The Transformer is combined with the convolutional neural network in a serial approach, and the original MSA module is replaced by the TPA module to achieve finer segmentation. Finally, the network model is evaluated and analyzed on three recognized retinal image datasets DRIVE, CHASE DB1, and STARE. The evaluation metrics were 0.9718, 0.9762, and 0.9773 for accuracy; 0.8410, 0.8437, and 0.8938 for sensitivity; and 0.8318, 0.8164, and 0.8557 for Dice coefficient. Compared with existing retinal image segmentation methods, the proposed method in this paper achieved better vessel segmentation in all of the publicly available fundus datasets tested performance and results.
引用
收藏
页数:22
相关论文
共 58 条
  • [1] Alom M ..., 2018, arXiv
  • [2] Detecting retinal vasculature as a key biomarker for deep Learning-based intelligent screening and analysis of diabetic and hypertensive retinopathy
    Arsalan, Muhammad
    Haider, Adnan
    Lee, Young Won
    Park, Kang Ryoung
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2022, 200
  • [3] Diabetic and Hypertensive Retinopathy Screening in Fundus Images Using Artificially Intelligent Shallow Architectures
    Arsalan, Muhammad
    Haider, Adnan
    Choi, Jiho
    Park, Kang Ryoung
    [J]. JOURNAL OF PERSONALIZED MEDICINE, 2022, 12 (01):
  • [4] Trainable COSFIRE filters for vessel delineation with application to retinal images
    Azzopardi, George
    Strisciuglio, Nicola
    Vento, Mario
    Petkov, Nicolai
    [J]. MEDICAL IMAGE ANALYSIS, 2015, 19 (01) : 46 - 57
  • [5] Cao H., 2021, ARXIV
  • [6] Changlu Guo, 2020, Neural Information Processing. 27th International Conference, ICONIP 2020. Proceedings. Lecture Notes in Computer Science (LNCS 12532), P509, DOI 10.1007/978-3-030-63830-6_43
  • [7] Chen B., 2021, ARXIV
  • [8] Retina Image Vessel Segmentation Using a Hybrid CGLI Level Set Method
    Chen, Guannan
    Chen, Meizhu
    Li, Jichun
    Zhang, Encai
    [J]. BIOMED RESEARCH INTERNATIONAL, 2017, 2017
  • [9] The Blood-Retinal Barrier in the Management of Retinal Disease: EURETINA Award Lecture
    Cunha-Vaz, Jose
    [J]. OPHTHALMOLOGICA, 2017, 237 (01) : 1 - 10
  • [10] ConViT: improving vision transformers with soft convolutional inductive biases
    d'Ascoli, Stephane
    Touvron, Hugo
    Leavitt, Matthew L.
    Morcos, Ari S.
    Biroli, Giulio
    Sagun, Levent
    [J]. JOURNAL OF STATISTICAL MECHANICS-THEORY AND EXPERIMENT, 2022, 2022 (11):