用稀疏图提升脑影像模型泛化能力,仅保留1%关键连接仍保持高准确率。
SpaRG: Sparsely Reconstructed Graphs for Generalizable fMRI Analysis
- 训练时自动筛选重要脑区连接,其余屏蔽以增强可解释性。
- 在64区域粗划分下仅用1%连接,跨域分类准确率仍提升。
- 适合关注fMRI模型可解释性与跨中心泛化的研究者。
深度学习有助于揭示静息态功能磁共振成像(rs-fMRI)中与精神疾病和个性特征相关的模式,但其结果难以解释,因数据易受扫描差异影响且难以可视化。本文提出一种基于稀疏化与自监督的学习方法——SpaRG:不依赖事后特征归因,而是在训练过程中识别少量高信息量的脑区连接并屏蔽其余部分。通过端到端联合训练稀疏输入掩码、变分自编码器(VAE)和下游分类器,利用来自多个采集站点的无标签数据优化掩码与VAE,仅保留泛化能力强的特征。在公开数据集ABIDE上,针对性别分类任务,使用18个站点的标注样本训练,适应两个分布外站点的少量无标签样本。在64个脑区粗划分下,SpaRG仅使用原连接的1%,仍实现跨域性能提升。代码已开源。
原文摘要 · Abstract (English)
Deep learning can help uncover patterns in resting-state functional Magnetic Resonance Imaging (rs-fMRI) associated with psychiatric disorders and personal traits. Yet the problem of interpreting deep learning findings is rarely more evident than in fMRI analyses, as the data is sensitive to scanning effects and inherently difficult to visualize. We propose a simple approach to mitigate these challenges grounded on sparsification and self-supervision. Instead of extracting post-hoc feature attributions to uncover functional connections that are important to the target task, we identify a small subset of highly informative connections during training and occlude the rest. To this end, we jointly train a (1) sparse input mask, (2) variational autoencoder (VAE), and (3) downstream classifier in an end-to-end fashion. While we need a portion of labeled samples to train the classifier, we optimize the sparse mask and VAE with unlabeled data from additional acquisition sites, retaining only the input features that generalize well. We evaluate our method - Sparsely Reconstructed Graphs (SpaRG) - on the public ABIDE dataset for the task of sex classification, training with labeled cases from 18 sites and adapting the model to two additional out-of-distribution sites with a portion of unlabeled samples. For a relatively coarse parcellation (64 regions), SpaRG utilizes only 1% of the original connections while improving the classification accuracy across domains. Our code can be found at github.com/yanismiraoui/SpaRG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。