Retinoblastoma Detection via Image Processing and Interpretable Artificial Intelligence Techniques

Abstract

Retinoblastoma (RB) is a treatable ocular melanoma that is diagnosed early and subsequently cured in the United States but has a poor prognosis in low- and middle-income countries (LMICs). This study outlines an approach to aid healthcare professionals in identifying RB in LMICs. Transfer learning methods were utilized for detection from fundus imaging. One hundred and forty RB+ and 140 RB- images were acquired from a previous deep-learning study. Next, five models were tested: VGG16, VGG19, Xception, Inception v3, and ResNet50, which were trained on the two-hundred-and-eighty image dataset. To evaluate these models, the Dice Similarity Coefficient (DSC) and Intersection-over-Union (IoU) were used. Explainable AI techniques such as SHAP and LIME were implemented into the best-performing models to increase the transparency of their decision-making frameworks, which is critical for the use of AI in medicine. We present that VGG16 is the best at identifying RB, though the other models achieved great levels of prediction. Transfer learning methods were effective at identifying RB, and explainable AI increased viability in clinical settings.

Competing Interest Statement

This work has been submitted to the IEEE for possible publication. Copyright may be transferred without notice, after which this version may no longer be accessible.

Funding Statement

This study did not receive any funding.

Author Declarations

I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.

Yes

The details of the IRB/oversight body that provided approval or exemption for the research described are given below:

All malignant image data is available at: https://www.mathworks.com/matlabcentral/fileexchange/99559-retinoblastoma-dataset Benign images were scraped from multiple publicly available sources and the full benign-malignant dataset is available upon reasonable request to the authors.

I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.

Yes

I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).

Yes

I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.

Yes

留言 (0)

沒有登入
gif