Download PDFOpen PDF in browser

AI-Driven Diagnostics in Ophthalmology: Tailored Deep Learning Models for Diabetic Retinopathy with XAI Insights

10 pagesPublished: July 12, 2024

Abstract

Diabetes Retinopathy, a leading cause of vision impairment, necessitates early and pre- cise detection. To address this, we developed a Convolutional Neural Network (CNN) model and tuned three popular pre-trained models, namely VGG16, Xception, and Mo- bileNetV2, to suit the specific characteristics of our dataset. To better understand the functioning of these deep learning algorithms, Explainable AI (XAI) techniques, such as CAM and Grad CAM++, were employed to highlight the crucial features influencing the model’s classifications. This study extends to the realm of imaging analysis, emphasiz- ing the critical importance of carefully selecting and customizing models to ensure precise and dependable diagnosis of complex conditions such as DR. Notably, the VGG16 model exhibited strong performance in identifying cases categorized as ’Moderate’ and ’No DR’, achieving accuracies of 0.90 and 0.98, respectively. Similarly, both Xception and Mo- bileNetV2 demonstrated promising results in the DR categories. Remarkably, our custom CNN model, tailored for our dataset, achieved an accuracy of 0.986 in identifying cases without DR (’No DR’). These results underscore the effectiveness of the trained deep learn- ing models in accurately diagnosing DR.

Keyphrases: Convolutional Neural Networks (CNN), deep learning, Diabetic Retinopathy, Explainable AI (XAI), Fundus Imaging, Grad-CAM, Grad-CAM++

In: Hisham Al-Mubaid, Tamer Aldwairi and Oliver Eulenstein (editors). Proceedings of the 16th International Conference on Bioinformatics and Computational Biology (BICOB-2024), vol 101, pages 73--82

Links:
BibTeX entry
@inproceedings{BICOB-2024:AI_Driven_Diagnostics_in_Ophthalmology,
  author    = {Krishna Mridha and Ming Wang and Lijun Zhang},
  title     = {AI-Driven Diagnostics in Ophthalmology: Tailored Deep Learning Models for Diabetic Retinopathy with XAI Insights},
  booktitle = {Proceedings of the 16th International Conference on Bioinformatics and Computational Biology (BICOB-2024)},
  editor    = {Hisham Al-Mubaid and Tamer Aldwairi and Oliver Eulenstein},
  series    = {EPiC Series in Computing},
  volume    = {101},
  pages     = {73--82},
  year      = {2024},
  publisher = {EasyChair},
  bibsource = {EasyChair, https://easychair.org},
  issn      = {2398-7340},
  url       = {https://easychair.org/publications/paper/G1Ff},
  doi       = {10.29007/gd9d}}
Download PDFOpen PDF in browser