Proto-memformer: deformable memory transformer for Parkinson's MRI classification.
Authors
Affiliations (4)
Affiliations (4)
- Xiangya School of Nursing, Central South University, Changsha, China.
- School of Nursing and Midwifery, University of Galway, Galway, Ireland.
- Key Laboratory of Digital-Intelligent Disease Surveillance and Health Governance, North Sichuan Medical College, Nanchong, China.
- Xiangya School of Nursing, Central South University, Changsha, China. [email protected].
Abstract
This paper proposes a Prototype-Guided Deformable Memory Transformer (Proto-MemFormer) model for Parkinson's Disease (PD) MRI classification. In the encoding stage, the model integrates a prototype-guided memory mechanism with a deformable attention structure to dynamically aggregate local morphological features and global semantic information. In the decoding stage, a position-calibrated retrieval module is introduced to enhance cross-sample feature alignment and discriminative representation. Experiments conducted on two public datasets, NTUA-Parkinson and PPMI, demonstrate that the proposed model achieves Accuracy, Precision, Recall, F1-Score, and AUC of 93.45%, 93.72%, 93.21%, 93.46%, and 94.81%, respectively, on the NTUA-Parkinson dataset, outperforming current state-of-the-art deep learning methods. Moreover, in the hyperparameter and training set scaling experiments, the model exhibits performance fluctuations of less than 3%, verifying its stability and robustness under different data and environmental conditions.