Explainable AI for Autism Diagnosis: Identifying Critical Brain Regions Using fMRI Data

Suryansh Vidya,Kush Gupta,Amir Aly,Andy Wills,Emmanuel Ifeachor,Rohit Shankar
2024-09-20
Abstract:Early diagnosis and intervention for Autism Spectrum Disorder (ASD) has been shown to significantly improve the quality of life of autistic individuals. However, diagnostics methods for ASD rely on assessments based on clinical presentation that are prone to bias and can be challenging to arrive at an early diagnosis. There is a need for objective biomarkers of ASD which can help improve diagnostic accuracy. Deep learning (DL) has achieved outstanding performance in diagnosing diseases and conditions from medical imaging data. Extensive research has been conducted on creating models that classify ASD using resting-state functional Magnetic Resonance Imaging (fMRI) data. However, existing models lack interpretability. This research aims to improve the accuracy and interpretability of ASD diagnosis by creating a DL model that can not only accurately classify ASD but also provide explainable insights into its working. The dataset used is a preprocessed version of the Autism Brain Imaging Data Exchange (ABIDE) with 884 samples. Our findings show a model that can accurately classify ASD and highlight critical brain regions differing between ASD and typical controls, with potential implications for early diagnosis and understanding of the neural basis of ASD. These findings are validated by studies in the literature that use different datasets and modalities, confirming that the model actually learned characteristics of ASD and not just the dataset. This study advances the field of explainable AI in medical imaging by providing a robust and interpretable model, thereby contributing to a future with objective and reliable ASD diagnostics.
Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The paper aims to address several key issues in the early diagnosis of Autism Spectrum Disorder (ASD): 1. **Improving Diagnostic Accuracy**: Current clinical diagnostic methods mainly rely on behavioral analysis and clinical history assessment, which are susceptible to subjective bias and are not effective enough for early diagnosis. Therefore, an objective and reliable biomarker is needed to assist in diagnosis. 2. **Identifying Biomarkers**: By analyzing resting-state functional magnetic resonance imaging (fMRI) data, researchers hope to identify brain functional connectivity patterns in autism patients and use them as potential biomarkers. These biomarkers can help doctors identify ASD patients earlier. 3. **Enhancing Model Interpretability**: Existing deep learning-based ASD classification models, although performing well in terms of accuracy, lack interpretability. This reduces the trust in the model's application in practice. The study aims to develop a model that is both accurate and interpretable, making its decision-making process more transparent, thereby gaining the trust and support of clinicians. 4. **Validating Model Effectiveness**: The research not only focuses on the accuracy of the model but also uses various interpretability techniques (such as Integrated Gradients, LIME, SHAP, etc.) to analyze the features of the model. This ensures that the model has indeed learned features related to ASD, rather than just memorizing patterns from a specific dataset. In summary, the goal of this study is to develop a deep learning model that can accurately classify ASD and provide interpretable results, in order to better understand the neurobiological basis of ASD and support future early diagnostic tools.