Application of AlexNet, EfficientNetV2B0, and VGG19 with Explainable AI for Cataract and Glaucoma Image Classification

被引:1
作者
Fayyad, Muhammad Fauzi [1 ]
Mustakim [1 ]
机构
[1] Univ Islam Negeri Sultan Syarif Kasim Riau, Fac Sci & Technol, Dept Informat Syst, Pekanbaru, Indonesia
来源
2024 INTERNATIONAL ELECTRONICS SYMPOSIUM, IES 2024 | 2024年
关键词
cataract; deep learning; glaucoma; grad-cam; XAI;
D O I
10.1109/IES63037.2024.10665856
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The rapid integration of Artificial Intelligence (AI) has significantly improved healthcare outcomes, especially in ophthalmology. However, Deep learning algorithms are often called to as 'black-box' models, making interpreting decision processes difficult. The Explainable AI approach helps understand AI model predictions by explaining the decision-making process based on features. The research objective is to find the best model for classifying eye disease images through experiments using AlexNet, EfficientNetV2, and VGG19 architectures. The selected best model will be analyzed using Grad-CAM, enhancing transparency and interpretability in healthcare AI models. This study employs multi-class classification using AlexNet, EfficientNetV2B0, and VGG19 architectures to classify cataract, glaucoma, and normal fundus. Different holdout-validation techniques (70:30, 80:20, 90:10) and optimization strategies (Adam, AdamW, RMSProp, SGDM) are studied for effective disease classification. The experimental results found that the best-performing model, EfficientNetV2B0 with holdout 90:10 and RMSProp optimizer, outperformed other models in accurately classifying fundus images. The model demonstrated strong performance with high recall, precision, f1-score of 0.8969, 0.8970, 0.8969, and an accuracy rate of 89.77%. Grad-CAM analysis also identified unique features of each eye disease class, for example, cataracts. AI model is capable of detecting cataract characteristics through the level of opacity in the eye lens, underscoring its robust classification capabilities in eye disease diagnosis.
引用
收藏
页码:406 / 412
页数:7
相关论文
共 50 条
[1]   An Adoptive Threshold-Based Multi-Level Deep Convolutional Neural Network for Glaucoma Eye Disease Detection and Classification [J].
Aamir, Muhammad ;
Irfan, Muhammad ;
Ali, Tariq ;
Ali, Ghulam ;
Shaf, Ahmad ;
Saeed, Alqahtani S. ;
Al-Beshri, Ali ;
Alasbali, Tariq ;
Mahnashi, Mater H. .
DIAGNOSTICS, 2020, 10 (08)
[2]  
[Anonymous], WHO Rep
[3]   Comprehensive Review on the Use of Artificial Intelligence in Ophthalmology and Future Research Directions [J].
Anton, Nicoleta ;
Doroftei, Bogdan ;
Curteanu, Silvia ;
Catalin, Lisa ;
Ilie, Ovidiu-Dumitru ;
Tarcoveanu, Filip ;
Bogdanici, Camelia Margareta .
DIAGNOSTICS, 2023, 13 (01)
[4]  
Arias-Serrano Isaac, 2023, F1000Res, V12, P14, DOI 10.12688/f1000research.122288.2
[5]   Explainable deep learning for efficient and robust pattern recognition: A survey of recent developments [J].
Bai, Xiao ;
Wang, Xiang ;
Liu, Xianglong ;
Liu, Qiang ;
Song, Jingkuan ;
Sebe, Nicu ;
Kim, Been .
PATTERN RECOGNITION, 2021, 120
[6]   Hierarchical method for cataract grading based on retinal images using improved Haar wavelet [J].
Cao, Lvchen ;
Li, Huiqi ;
Zhang, Yanjun ;
Zhang, Li ;
Xu, Liang .
INFORMATION FUSION, 2020, 53 :196-208
[7]  
Choi D, 2020, Arxiv, DOI [arXiv:1910.05446, DOI 10.48550/ARXIV.1910.05446, 10.48550/arXiv.1910.05446]
[8]   Detection of glaucoma using three-stage training with EfficientNet [J].
de Zarza, I. ;
de Curto, J. ;
Calafate, Carlos T. .
INTELLIGENT SYSTEMS WITH APPLICATIONS, 2022, 16
[9]  
Doddi Guna Venkat, Kaggle
[10]   Deep Learning Approach for Stages of Severity Classification in Diabetic Retinopathy Using Color Fundus Retinal Images [J].
Goel, Silky ;
Gupta, Siddharth ;
Panwar, Avnish ;
Kumar, Sunil ;
Verma, Madhushi ;
Bourouis, Sami ;
Ullah, Mohammad Aman .
MATHEMATICAL PROBLEMS IN ENGINEERING, 2021, 2021