Document Type : Original/Review Paper
Authors
1 Department of Computer Science, Shahid Bahonar University of Kerman, Kerman, Iran
2 Department of Computer Science, Shahid Bahonar University of Kerman, Kerman, Iran.
Abstract
The diagnosis of Alzheimer's Disease (AD) remains a significant challenge in medical research. To address the limitations of static models in capturing dynamic brain changes, this paper proposes a novel GNN-xLSTM model that integrates Graph Neural Networks (GNN) with an extended Long Short-Term Memory (xLSTM) architecture. The key innovation lies in combining GNN’s ability to model spatial relationships in brain imaging data with xLSTM’s enhanced sequential learning via matrix-based memory representation and exponential gate stabilization. In the proposed approach, brain images are divided into regions, with each region represented as a graph node connected in a grid structure, and feature vectors are extracted for each node. The proposed architecture incorporates Graph Convolutional Network (GCN) layers, xLSTM cells, residual connections, batch normalization, and dropout to jointly capture global, local, and temporal dependencies. Evaluated on the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset, the GNN-xLSTM model outperforms baseline models in terms of accuracy, precision, recall, and F1-score. These results demonstrate the model’s effectiveness in identifying critical brain regions and improving AD classification performance.
Keywords
- Alzheimer’s disease
- Magnetic resonance imaging
- Brain connectivity graphs
- Graph Convolutional Network
- Convolutional Neural Network
Main Subjects
[1] A. K. Mishra, M. K. Tripathi, D. Kumar, and S. P. Gupta, "Neurons specialize in presynaptic autophagy: A perspective to ameliorate neurodegeneration," Molecular Neurobiology, 2024, pp. 1–15.
[2] P. Scheltens, B. De Strooper, M. Kivipelto, H. Holstege, G. Chételat, C. E. Teunissen, J. Cummings, and W. M. Van der Flier, "Alzheimer’s disease," The Lancet, vol. 397, no. 10284, pp. 1577–1590, 2021.
[3] G. B. Frisoni, N. C. Fox, C. R. Jack Jr, P. Scheltens, and P. M. Thompson, "The clinical use of structural MRI in Alzheimer disease," Nature Reviews Neurology, vol. 6, no. 2, pp. 67–77, 2010.[4] N. Habib, C. McCabe, S. Medina, M. Varshavsky, D. Kitsberg, R. Dvir-Szternfeld, G. Green, D. Dionne, L. Nguyen, J. L. Marshall, et al., "Disease-associated astrocytes in Alzheimer’s disease and aging," Nature Neuroscience, vol. 23, no. 6, pp. 701–706, 2020.
[5] O. O. Olatunde, K. S. Oyetunde, J. Han, M. T. Khasawneh, H. Yoon, and A. D. Initiative, "Multiclass classification of Alzheimer’s disease prodromal stages using sequential feature embeddings and regularized multikernel support vector machine," NeuroImage, 2024, p. 120929.
[6] R. J. Bateman, C. Xiong, T. L. Benzinger, A. M. Fagan, A. Goate, N. C. Fox, D. S. Marcus, N. J. Cairns, X. Xie, T. M. Blazey, et al., "Clinical and biomarker changes in dominantly inherited Alzheimer’s disease," New England Journal of Medicine, vol. 367, no. 9, pp. 795–804, Sept. 2012.
[7] M. W. Weiner, D. P. Veitch, P. S. Aisen, L. A. Beckett, N. J. Cairns, R. C. Green, D. Harvey, C. R. Jack, W. Jagust, E. Liu, et al., "The Alzheimer’s disease neuroimaging initiative: A review of papers published since its inception," Alzheimer’s & Dementia, vol. 9, no. 5, pp. e111–e194, 2013.
[8] G. Mukhtar and S. Farhan, "Convolutional neural network based prediction of conversion from mild cognitive impairment to Alzheimer's disease: A technique using hippocampus extracted from MRI," Advances in Electrical and Computer Engineering, vol. 20, no. 2, pp. 113–122, 2020.
[9] Y. LeCun, Y. Bengio, and G. Hinton, "Deep learning," Nature, vol. 521, no. 7553, pp. 436–444, 2015.
[10] Z. Wu, S. Pan, F. Chen, G. Long, C. Zhang, and P. S. Yu, "A comprehensive survey on graph neural networks," IEEE Transactions on Neural Networks and Learning Systems, vol. 32, no. 1, pp. 4–24, 2021.
[11] M. Beck, K. Pöppel, M. Spanring, A. Auer, O. Prudnikova, M. K. Kopp, et al., "XLSTM: Extended long short-term memory," in First Workshop on Long-Context Foundation Models @ ICML 2024, 2024.
[12] A. Singh, P. van de Ven, C. Eising, and P. Denny, "Dynamic filter application in graph convolutional networks for enhanced spectral feature analysis and class discrimination in medical imaging," IEEE Access, 2024.
[13] M. Wei, Y. Li, M. Liang, M. Xi, and H. Tian, "Artificial intelligence approaches for early detection and diagnosis of Alzheimer’s disease: A review," Academic Journal of Science and Technology, vol. 5, no. 3, pp. 215–221, 2023.
[14] G. Liu, Y. Yan, J. Cai, E. Q. Wu, S. Fang, A. D. Cheok, and A. G. Song, "GCD: Graph contrastive denoising module for GNNs in EEG classification," Expert Systems with Applications, p. 126013, 2024.
[15] J. Tang, Y. Yang, W. Wei, L. Shi, S. Su, S. Cheng, D. Yin, and C. Huang, "GraphGPT: Graph instruction tuning for large language models," in Proceedings of the 47th International ACM SIGIR Conference on Research and Development in Information Retrieval, 2024, pp. 491–500.
[16] M. Ghorbani, A. Kazi, M. S. Baghshah, H. R. Rabiee, and N. Navab, "RA-GCN: Graph convolutional network for disease prediction problems with imbalanced data," Medical Image Analysis, vol. 75, p. 102272, 2022.
[17] H. Song, C. Liu, S. Li, and P. Zhang, "TS-GCN: A novel tumor segmentation method integrating transformer and GCN," Mathematical Biosciences and Engineering, vol. 20, no. 10, pp. 18173–18190, 2023.
[18] F. Behrad and M. S. Abadeh, "An overview of deep learning methods for multimodal medical data mining," Expert Systems with Applications, vol. 200, p. 117006, 2022.
[19] A. De and A. S. Chowdhury, "DTI based Alzheimer's disease classification with rank modulated fusion of CNNs and random forest," Expert Systems with Applications, vol. 169, p. 114338, 2021.
[20] E. M. Benyoussef, A. Elbyed, and H. El Hadiri, "3D MRI classification using KNN and deep neural network for Alzheimer's disease diagnosis," in Advanced Intelligent Systems for Sustainable Development (AI2SD'2018) Vol 4: Advanced Intelligent Systems Applied to Health, Springer, 2019, pp. 154–158.
[21] S. Park et al., “Prospective classification of Alzheimer’s disease conversion from mild cognitive impairment,” Neural Networks, vol. 164, pp. 335–344, 2023.
[22] N.-H. Ho et al., “Predicting progression of Alzheimer’s disease using forward-to-backward bi-directional network with integrative imputation,” Neural Networks, vol. 150, pp. 422–439, 2022.
[23] C. Liu et al., “Monte Carlo ensemble neural network for the diagnosis of Alzheimer’s disease,” Neural Networks, vol. 159, pp. 14–24, 2023.
[24] T.-E. Kam, H. Zhang, Z. Jiao, and D. Shen, "Deep learning of static and dynamic brain functional networks for early MCI detection," IEEE Transactions on Medical Imaging, vol. 39, no. 2, pp. 478–487, 2019.
[25] A. J. Chang, R. Roth, E. Bougioukli, T. Ruber, S. S. Keller, D. L. Drane, R. E. Gross, J. Welsh, A. Abrol, V. Calhoun, et al., "MRI-based deep learning can discriminate between temporal lobe epilepsy, Alzheimer's disease, and healthy controls," Communications Medicine, vol. 3, no. 1, p. 33, 2023.
[26] B. Yu, Y. Li, C. Wang, H. Zhang, and J. Chen, "LSTM-Fuzzy Logic for Fault Forecasting and Mitigation in Propulsion System," IEEE Transactions on Aerospace and Electronic Systems, 2024.
[27] S. Ebrahimi Mood, A. Rouhbakhsh, and A. Souri, "Evolutionary recurrent neural network based on equilibrium optimization method for cloud-edge resource management in internet of things," Neural Computing and Applications, vol. 37, no. 6, pp. 4957–4969, 2025.
[28] Y. Dong, "Application Research on Classification and Integration Model of Innovation and Entrepreneurship Education Resources Based on GNN-PSO Algorithm," Systems and Soft Computing, p. 200326, 2025.
[29] H. Zeng, H. Zhou, A. Srivastava, R. Kannan, and V. Prasanna, "Accurate, efficient and scalable graph embedding," in 2019 IEEE International Parallel and Distributed Processing Symposium (IPDPS), 2019, pp. 462–471.
[30] G. Ma, N. K. Ahmed, T. L. Willke, D. Sengupta, M. W. Cole, N. B. Turk-Browne, and P. S. Yu, "Deep graph similarity learning for brain data analysis," in Proceedings of the 28th ACM International Conference on Information and Knowledge Management, 2019, pp. 2743–2751.
[31] J. Guo, W. Qiu, X. Li, X. Zhao, N. Guo, and Q. Li, "Predicting Alzheimer's disease by hierarchical graph convolution from positron emission tomography imaging," in 2019 IEEE International Conference on Big Data (Big Data), IEEE, 2019, pp. 5359–5363.
[32] T.-A. Song, S. R. Chowdhury, F. Yang, H. Jacobs, G. El Fakhri, Q. Li, K. Johnson, and J. Dutta, "Graph convolutional neural networks for Alzheimer’s disease classification," in 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019), 2019, pp. 414–417.
[33] D. Yao, J. Sui, M. Wang, E. Yang, Y. Jiaerken, N. Luo, P.-T. Yap, M. Liu, and D. Shen, "A mutual multi-scale triplet graph convolutional network for classification of brain disorders using functional or structural connectivity," IEEE Transactions on Medical Imaging, vol. 40, no. 4, pp. 1279–1289, 2021.
[34] P. Rabiei and N. Ashrafi-Payaman, "Anomaly detection in dynamic graph using machine learning algorithms," Journal of AI and Data Mining, vol. 12, no. 3, pp. 359–367, 2024, doi: 10.22044/jadm.2024.14476.2551.
[35] I. Loshchilov and F. Hutter, "Decoupled weight decay regularization," arXiv preprint arXiv:1711.05101, 2017.
[36] S. Parisot, S. Ktena, E. Ferrante, M. Lee, R. Guerrero, B. Glocker, and D. Rueckert, "Disease prediction using graph convolutional networks: Application to autism spectrum disorder and Alzheimer's disease," Medical Image Analysis, vol. 48, pp. 117-130, 2018.
[37] S. Sarraf and G. Tofighi, "Classification of Alzheimer's disease using fMRI data and deep learning convolutional neural networks," arXiv preprint arXiv:1603.08631, 2016.
)