A hybrid evolutionary weighted ensemble of deep transfer learning models for retinal vessel segmentation and diabetic retinopathy detection

被引:16
作者
Vij, Richa [1 ]
Arora, Sakshi [1 ]
机构
[1] Shri Mata Vaishno Devi Univ, Sch Comp Sci & Engn, Katra 182320, Jammu And Kashm, India
关键词
Diabetic retinopathy; Automatic retinal blood vessel segmentation; Hybrid evolutionary weighted ensemble; U; -Net; Deep Transfer Learning; NEURAL-NETWORK; CLASSIFICATION; ARCHITECTURE;
D O I
10.1016/j.compeleceng.2024.109107
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Segmentation of retinal blood vessels in fundus images is critical for early detection and treatment of diabetic retinopathy(DR). Due to the complex distribution of blood vessels, variations in noise, illumination, and vessel orientation in fundus images, the segmentation process becomes extremely challenging and time-consuming. In recent years, deep learning(DL)-based methods have been recognized as promising strategies for automatic retinal blood vessel segmentation that aids in early treatment, but most earlier DL algorithms prioritized accuracy over the complexity of the model for segmenting retinal vessels, making them challenging to adapt to medical devices. To the best of our knowledge, this is the first work to present a novel hybrid evolutionary weighted ensemble of three deep transfer learning(DTL) models: ResNet34, Inception V3, and VGG16 utilizing two public retinal fundus databases: DRIVE, and HRF for retinal blood vessel segmentation and latter detecting DR using Resnet34+Unet. Among comparing single models, Resnet34+U-Net achieved impressive accuracy of 0.9575, F1 and IOU scores of 0.8421 and 0.8159, and AUC values of 0.9868 in the DRIVE dataset. In comparison to the state-of-the-art results for segmenting retinal blood vessels using DRIVE and HRF datasets, our proposed approach achieves 0.9853 and 0.9816 accuracy, 0.9881 and 0.9833 precision, 0.9853 and 0.9828 recall, 0.8791 and 0.8497 F1 scores, 0.8066 and 0.7754 IOU scores, and AUC of 0.9996 and 0.9928 respectively, with a rise of nearly 7-8 % and for classification, Resnet34+Unet achieves an accuracy= 0.996, precision= 0.993, recall= 1, and AUC= 0.999 respectively. Hence, the proposed model shows good potential for real-time diagnosis.
引用
收藏
页数:30
相关论文
共 52 条
[1]   DR-IIXRN : Detection Algorithm of Diabetic Retinopathy Based on Deep Ensemble Learning and Attention Mechanism [J].
Ai, Zhuang ;
Huang, Xuan ;
Fan, Yuan ;
Feng, Jing ;
Zeng, Fanxin ;
Lu, Yaping .
FRONTIERS IN NEUROINFORMATICS, 2021, 15
[2]   Automatic segmentation of melanoma skin cancer using transfer learning and fine-tuning [J].
Araujo, Rafael Luz ;
de Araujo, Flavio H. D. ;
e Silva, Romuere R., V .
MULTIMEDIA SYSTEMS, 2022, 28 (04) :1239-1250
[3]   Detection of Diabetic Retinopathy in Retinal Fundus Images Using CNN Classification Models [J].
Asia, Al-Omaisi ;
Zhu, Cheng-Zhang ;
Althubiti, Sara A. ;
Al-Alimi, Dalal ;
Xiao, Ya-Long ;
Ouyang, Ping-Bo ;
Al-Qaness, Mohammed A. A. .
ELECTRONICS, 2022, 11 (17)
[4]   Diabetic Retinopathy Improved Detection Using Deep Learning [J].
Ayala, Angel ;
Ortiz Figueroa, Tomas ;
Fernandes, Bruno ;
Cruz, Francisco .
APPLIED SCIENCES-BASEL, 2021, 11 (24)
[5]   Defining a standard for particle swarm optimization [J].
Bratton, Daniel ;
Kennedy, James .
2007 IEEE SWARM INTELLIGENCE SYMPOSIUM, 2007, :120-+
[6]   Robust Vessel Segmentation in Fundus Images [J].
Budai, A. ;
Bock, R. ;
Maier, A. ;
Hornegger, J. ;
Michelson, G. .
INTERNATIONAL JOURNAL OF BIOMEDICAL IMAGING, 2013, 2013 (2013)
[7]   A retinal blood vessel segmentation based on improved D-MNet and pulse-coupled neural network [J].
Deng, Xiangyu ;
Ye, Jinhong .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 73
[8]   A New Hybrid Algorithm for Retinal Vessels Segmentation on Fundus Images [J].
Dharmawan, Dhimas Arief ;
Li, Di ;
Ng, Boon Poh ;
Rahardja, Susanto .
IEEE ACCESS, 2019, 7 :41885-41896
[9]   SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation [J].
Guo, Changlu ;
Szemenyei, Marton ;
Yi, Yugen ;
Wang, Wenle ;
Chen, Buer ;
Fan, Changqi .
2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, :1236-1242
[10]   A retinal vessel detection approach using convolution neural network with reinforcement sample learning strategy [J].
Guo, Yanhui ;
Budak, Umit ;
Vespa, Lucas J. ;
Khorasani, Elham ;
Sengur, Abdulkadir .
MEASUREMENT, 2018, 125 :586-591