Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP
Retinoblastoma is a rare and aggressive form of childhood eye cancer that requires prompt diagnosis and treatment to prevent vision loss and even death. Deep learning models have shown promising results in detecting retinoblastoma from fundus images, but their decision-making process is often consid...
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2023-06-01
|
| Series: | Diagnostics |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2075-4418/13/11/1932 |
| _version_ | 1797597765997428736 |
|---|---|
| author | Bader Aldughayfiq Farzeen Ashfaq N. Z. Jhanjhi Mamoona Humayun |
| author_facet | Bader Aldughayfiq Farzeen Ashfaq N. Z. Jhanjhi Mamoona Humayun |
| author_sort | Bader Aldughayfiq |
| collection | DOAJ |
| description | Retinoblastoma is a rare and aggressive form of childhood eye cancer that requires prompt diagnosis and treatment to prevent vision loss and even death. Deep learning models have shown promising results in detecting retinoblastoma from fundus images, but their decision-making process is often considered a “black box” that lacks transparency and interpretability. In this project, we explore the use of LIME and SHAP, two popular explainable AI techniques, to generate local and global explanations for a deep learning model based on InceptionV3 architecture trained on retinoblastoma and non-retinoblastoma fundus images. We collected and labeled a dataset of 400 retinoblastoma and 400 non-retinoblastoma images, split it into training, validation, and test sets, and trained the model using transfer learning from the pre-trained InceptionV3 model. We then applied LIME and SHAP to generate explanations for the model’s predictions on the validation and test sets. Our results demonstrate that LIME and SHAP can effectively identify the regions and features in the input images that contribute the most to the model’s predictions, providing valuable insights into the decision-making process of the deep learning model. In addition, the use of InceptionV3 architecture with spatial attention mechanism achieved high accuracy of 97% on the test set, indicating the potential of combining deep learning and explainable AI for improving retinoblastoma diagnosis and treatment. |
| first_indexed | 2024-03-11T03:10:06Z |
| format | Article |
| id | doaj.art-f8305326a6d546c480a64920afffdec8 |
| institution | Directory Open Access Journal |
| issn | 2075-4418 |
| language | English |
| last_indexed | 2024-03-11T03:10:06Z |
| publishDate | 2023-06-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Diagnostics |
| spelling | doaj.art-f8305326a6d546c480a64920afffdec82023-11-18T07:43:02ZengMDPI AGDiagnostics2075-44182023-06-011311193210.3390/diagnostics13111932Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAPBader Aldughayfiq0Farzeen Ashfaq1N. Z. Jhanjhi2Mamoona Humayun3Department of Information Systems, College of Computer and Information Sciences, Jouf University, Sakaka 72388, Saudi ArabiaSchool of Computer Science, SCS, Taylor’s University, Subang Jaya 47500, MalaysiaSchool of Computer Science, SCS, Taylor’s University, Subang Jaya 47500, MalaysiaDepartment of Information Systems, College of Computer and Information Sciences, Jouf University, Sakaka 72388, Saudi ArabiaRetinoblastoma is a rare and aggressive form of childhood eye cancer that requires prompt diagnosis and treatment to prevent vision loss and even death. Deep learning models have shown promising results in detecting retinoblastoma from fundus images, but their decision-making process is often considered a “black box” that lacks transparency and interpretability. In this project, we explore the use of LIME and SHAP, two popular explainable AI techniques, to generate local and global explanations for a deep learning model based on InceptionV3 architecture trained on retinoblastoma and non-retinoblastoma fundus images. We collected and labeled a dataset of 400 retinoblastoma and 400 non-retinoblastoma images, split it into training, validation, and test sets, and trained the model using transfer learning from the pre-trained InceptionV3 model. We then applied LIME and SHAP to generate explanations for the model’s predictions on the validation and test sets. Our results demonstrate that LIME and SHAP can effectively identify the regions and features in the input images that contribute the most to the model’s predictions, providing valuable insights into the decision-making process of the deep learning model. In addition, the use of InceptionV3 architecture with spatial attention mechanism achieved high accuracy of 97% on the test set, indicating the potential of combining deep learning and explainable AI for improving retinoblastoma diagnosis and treatment.https://www.mdpi.com/2075-4418/13/11/1932retinoblastomaexplainable AIdeep learningLIMESHAPmedical image analysis |
| spellingShingle | Bader Aldughayfiq Farzeen Ashfaq N. Z. Jhanjhi Mamoona Humayun Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP Diagnostics retinoblastoma explainable AI deep learning LIME SHAP medical image analysis |
| title | Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP |
| title_full | Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP |
| title_fullStr | Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP |
| title_full_unstemmed | Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP |
| title_short | Explainable AI for Retinoblastoma Diagnosis: Interpreting Deep Learning Models with LIME and SHAP |
| title_sort | explainable ai for retinoblastoma diagnosis interpreting deep learning models with lime and shap |
| topic | retinoblastoma explainable AI deep learning LIME SHAP medical image analysis |
| url | https://www.mdpi.com/2075-4418/13/11/1932 |
| work_keys_str_mv | AT baderaldughayfiq explainableaiforretinoblastomadiagnosisinterpretingdeeplearningmodelswithlimeandshap AT farzeenashfaq explainableaiforretinoblastomadiagnosisinterpretingdeeplearningmodelswithlimeandshap AT nzjhanjhi explainableaiforretinoblastomadiagnosisinterpretingdeeplearningmodelswithlimeandshap AT mamoonahumayun explainableaiforretinoblastomadiagnosisinterpretingdeeplearningmodelswithlimeandshap |