Diabetic Retinopathy Lesion Segmentation Method Based on Multi-Scale Attention and Lesion Perception

被引:1
作者
Bian, Ye [1 ]
Si, Chengyong [1 ,2 ]
Wang, Lei [3 ]
机构
[1] Univ Shanghai Sci & Technol, Sch Opt Elect & Comp Engn, Shanghai 200093, Peoples R China
[2] Univ Shanghai Sci & Technol, Sino German Coll, Shanghai 200093, Peoples R China
[3] Tongji Univ, Coll Elect & Informat Engn, Shanghai 201804, Peoples R China
基金
中国国家自然科学基金;
关键词
diabetic retinopathy (DR); image segmentation; attention; deep learning;
D O I
10.3390/a17040164
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The early diagnosis of diabetic retinopathy (DR) can effectively prevent irreversible vision loss and assist ophthalmologists in providing timely and accurate treatment plans. However, the existing methods based on deep learning have a weak perception ability of different scale information in retinal fundus images, and the segmentation capability of subtle lesions is also insufficient. This paper aims to address these issues and proposes MLNet for DR lesion segmentation, which mainly consists of the Multi-Scale Attention Block (MSAB) and the Lesion Perception Block (LPB). The MSAB is designed to capture multi-scale lesion features in fundus images, while the LPB perceives subtle lesions in depth. In addition, a novel loss function with tailored lesion weight is designed to reduce the influence of imbalanced datasets on the algorithm. The performance comparison between MLNet and other state-of-the-art methods is carried out in the DDR dataset and DIARETDB1 dataset, and MLNet achieves the best results of 51.81% mAUPR, 49.85% mDice, and 37.19% mIoU in the DDR dataset, and 67.16% mAUPR and 61.82% mDice in the DIARETDB1 dataset. The generalization experiment of MLNet in the IDRiD dataset achieves 59.54% mAUPR, which is the best among other methods. The results show that MLNet has outstanding DR lesion segmentation ability.
引用
收藏
页数:16
相关论文
共 40 条
[1]   LezioSeg: Multi-Scale Attention Affine-Based CNN for Segmenting Diabetic Retinopathy Lesions in Images [J].
Ali, Mohammed Yousef Salem ;
Jabreel, Mohammed ;
Valls, Aida ;
Baget, Marc ;
Abdel-Nasser, Mohamed .
ELECTRONICS, 2023, 12 (24)
[2]   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
[3]   An interpretable dual attention network for diabetic retinopathy grading: IDANet [J].
Bhati, Amit ;
Gour, Neha ;
Khanna, Pritee ;
Ojha, Aparajita ;
Werghi, Naoufel .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2024, 149
[4]  
Cao Hu, 2023, Computer Vision - ECCV 2022 Workshops: Proceedings. Lecture Notes in Computer Science (13803), P205, DOI 10.1007/978-3-031-25066-8_9
[5]   L-Seg: An end-to-end unified framework for multi-lesion segmentation of fundus images [J].
Guo, Song ;
Li, Tao ;
Kang, Hong ;
Li, Ning ;
Zhang, Yujun ;
Wang, Kai .
NEUROCOMPUTING, 2019, 349 :52-63
[6]   Diabetic retinopathy lesion segmentation using deep multi-scale framework [J].
Guo, Tianjiao ;
Yang, Jie ;
Yu, Qi .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 88
[7]   H2Former: An Efficient Hierarchical Hybrid Transformer for Medical Image Segmentation [J].
He, Along ;
Wang, Kai ;
Li, Tao ;
Du, Chengkun ;
Xia, Shuang ;
Fu, Huazhu .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (09) :2763-2775
[8]   Progressive Multiscale Consistent Network for Multiclass Fundus Lesion Segmentation [J].
He, Along ;
Wang, Kai ;
Li, Tao ;
Bo, Wang ;
Kang, Hong ;
Fu, Huazhu .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2022, 41 (11) :3146-3157
[9]   SaB-Net: Self-attention backward network for gastric tumor segmentation in CT images [J].
He, Junjie ;
Zhang, Mudan ;
Li, Wuchao ;
Peng, Yunsong ;
Fu, Bangkang ;
Liu, Chen ;
Wang, Jian ;
Wang, Rongpin .
COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 169
[10]   RTNet: Relation Transformer Network for Diabetic Retinopathy Multi-Lesion Segmentation [J].
Huang, Shiqi ;
Li, Jianan ;
Xiao, Yuze ;
Shen, Ning ;
Xu, Tingfa .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2022, 41 (06) :1596-1607