C3BAM-XAI: Convolutional Block Attention Module Enhanced Explainable Artificial Intelligence-Based Parkinson’s Disease Stage Classification
摘要
Parkinson’s disease (PD) is a progressive neurodegenerative disorder that causes significant impairment in neurons, physiological structures, and behavior of people. However, these changes are very subtle in the early stages of PD, making diagnosis and treatment challenging. To overcome such challenges, we proposed a novel convolutional neural network (CNN) architecture called C3BAM-XAI that is based on the convolutional block attention module (CBAM) enhanced CNN and explainable artificial intelligence (XAI). In the proposed architecture, the data imbalance issue was resolved at the initial stage by employing augmentation techniques for the training data. After that, the proposed CNN architecture is designed with CBAM. The CBAM architecture is based on two sub-modules—the channel attention module and the spatial attention module. The input of the initial convolutional layer is passed to the channel attention module, which is further passed to the spatial attention module for more critical information extraction. For smooth training, hyperparameters are selected using Nadam Optimization (NO) instead of manual assignment. In the testing phase, features are extracted from the dense layer and passed to several neural network classifiers. In addition to this, the designed model is interpreted using the explainable AI technique to see the predicted label and insight decision features. The experimental process was performed on the publically available PD Kaggle dataset and obtained an improved accuracy of 93.33%. In addition, several ablation studies were conducted, and it is concluded that the CBAM not only improved accuracy but also made this model a reliable tool for clinical diagnosis, both in terms of accuracy and interpretability. The results thus open new avenues toward applying deep learning models practically for applications in the medical domain and early-stage detection of Parkinson’s disease, thereby helping to improve patient outcomes.