Alzheimer's disease prediction via an explainable CNN using genetic algorithm and SHAP values

利用遗传算法和SHAP值,通过可解释卷积神经网络预测阿尔茨海默病

阅读:1

Abstract

Convolutional neural networks (CNNs) are widely recognized for their high precision in image classification. Nevertheless, the lack of transparency in these black-box models raises concerns in sensitive domains such as healthcare, where understanding the knowledge acquired to derive outcomes can be challenging. To address this concern, several strategies within the field of explainable AI (XAI) have been developed to enhance model interpretability. This study introduces a novel XAI technique, GASHAP, which integrates a genetic algorithm (GA) with SHapley Additive exPlanations (SHAP) to improve the explainability of our 3D convolutional neural network (3D-CNN) model. The model is designed to classify magnetic resonance imaging (MRI) brain scans of individuals with Alzheimer's disease and cognitively normal controls. Deep SHAP, a widely used XAI technique, facilitates the understanding of the influence exerted by various voxels on the final classification outcome (Lundberg SM, Lee SI. A unified approach to interpreting model predictions. In: Advances in Neural Information Processing Systems, 2017. 4765-74. https://doi.org/10.5555/3295222.3295230). However, voxel-level representation alone lacks interpretive clarity. Therefore, the objective of this study is to provide findings at the level of anatomically defined brain regions. Critical regions are identified by leveraging their SHAP values, followed by the application of a genetic algorithm to generate a definitive mask highlighting the most significant regions for Alzheimer's disease diagnosis (Shahamat H, Saniee Abadeh M. Brain MRI analysis using a deep learning based evolutionary approach. Neural Netw. 2020;126:218-34. https://doi.org/10.1016/j.neunet.2020.03.017 PMID: 32259762). The research commenced by implementing a 3D-CNN for MRI image classification. Subsequently, the GASHAP technique was applied to enhance model transparency. The final result is a brain mask that delineates the pertinent regions crucial for Alzheimer's disease diagnosis. Finally, a comparative analysis is conducted between our findings and those of previous studies.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。