Multi-Level Graph Neural Network With Sparsity Pooling for Recognizing Parkinsons Disease

Xiaobo Zhang,Yuxin Zhou,Zhijie Lu,Donghai Zhai,Haonan Luo,Tianrui Li,Yang Li
DOI: https://doi.org/10.1109/TNSRE.2023.3330643
IF: 4.9
2023-01-01
IEEE Transactions on Neural Systems and Rehabilitation Engineering
Abstract:Parkinson's disease (PD) is a neurodegenerative disease of the brain associated with motor symptoms. With the maturation of machine learning (ML), especially deep learning, ML has been used to assist in the diagnosis of PD. In this paper, we explore graph neural networks (GNNs) to implement PD prediction using MRI data. However, most existing GNN models suffer from the efficiency of graph construction on MRI data and the problem of overfitting on small data. This paper proposes a novel multi-layer GNN model that incorporates a fast graph construction method and a sparsity-based pooling layer with an attention mechanism. In addition, graph structure sparsity is plugged into the graph pooling layer as prior knowledge to mitigate overfitting in model training. Experimental results on real-world datasets demonstrate the effectiveness of the proposed model and its superiority over baseline methods.
What problem does this paper attempt to address?