Contrastive Graph Pooling for Explainable Classification of Brain Networks

التفاصيل البيبلوغرافية
العنوان: Contrastive Graph Pooling for Explainable Classification of Brain Networks
المؤلفون: Xu, Jiaxing, Bian, Qingtian, Li, Xinhang, Zhang, Aihu, Ke, Yiping, Qiao, Miao, Zhang, Wei, Sim, Wei Khang Jeremy, Gulyás, Balázs
المصدر: IEEE Transactions on Medical Imaging, vol. 43, no. 9, pp. 3292-3305, Sept. 2024
سنة النشر: 2023
المجموعة: Computer Science
Quantitative Biology
مصطلحات موضوعية: Quantitative Biology - Neurons and Cognition, Computer Science - Artificial Intelligence, Computer Science - Machine Learning
الوصف: Functional magnetic resonance imaging (fMRI) is a commonly used technique to measure neural activation. Its application has been particularly important in identifying underlying neurodegenerative conditions such as Parkinson's, Alzheimer's, and Autism. Recent analysis of fMRI data models the brain as a graph and extracts features by graph neural networks (GNNs). However, the unique characteristics of fMRI data require a special design of GNN. Tailoring GNN to generate effective and domain-explainable features remains challenging. In this paper, we propose a contrastive dual-attention block and a differentiable graph pooling method called ContrastPool to better utilize GNN for brain networks, meeting fMRI-specific requirements. We apply our method to 5 resting-state fMRI brain network datasets of 3 diseases and demonstrate its superiority over state-of-the-art baselines. Our case study confirms that the patterns extracted by our method match the domain knowledge in neuroscience literature, and disclose direct and interesting insights. Our contributions underscore the potential of ContrastPool for advancing the understanding of brain networks and neurodegenerative conditions. The source code is available at https://github.com/AngusMonroe/ContrastPool.
نوع الوثيقة: Working Paper
DOI: 10.1109/TMI.2024.3392988
URL الوصول: http://arxiv.org/abs/2307.11133
رقم الأكسشن: edsarx.2307.11133
قاعدة البيانات: arXiv
الوصف
DOI:10.1109/TMI.2024.3392988