Enhancing Neurodegenerative Disease Diagnosis Through Confidence-Driven Dynamic Spatio-Temporal Convolutional Network

Abstract

Dynamic brain networks are more effective than static networks in characterizing the evolving patterns of brain functional connectivity, making them a more promising tool for diagnosing neurodegenerative diseases. However, existing classification methods for dynamic brain networks often rely on sliding windows to extract multi-window features, leading to suboptimal performance due to the spatio-temporal coupling on these windows and limited ability to effectively integrate complex topological features. To address these limitations, we propose a novel method called Confidence-Driven Dynamic Spatio-Temporal Convolutional Network (CD-DSTCN). First, our proposed method employs a spatio-temporal convolutional network integrated with a temporal attention mechanism to extract spatio-temporal features within each window. By propagating information across temporal windows during spatial convolution, the method effectively captures and integrates complex temporal and spatial dependencies. Second, each window generates an output probability, which quantifies prediction confidence based on the true class probability (TCP). This confidence score serves as a weight to assess the relative importance of different time windows. Finally, the confidence-weighted fused features are passed through a multilayer perceptron (MLP) for final classification. Extensive experiments on Alzheimer’s and Parkinson’s datasets show that the proposed method outperforms the state-of-the-art algorithms and can provide valuable biomarkers for brain disease diagnosis. Our code is publicly available at: https://github.com/YNingCode/CD-DSTCN