解决多中心脑影像数据差异问题,提升精神疾病识别准确率
A Confounding Factors-Inhibition Adversarial Learning Framework for Multi-site fMRI Mental Disorder Identification
- 设计节点信息融合机制,更有效提取功能连接特征
- 在ABIDE和ADHD-200数据集上分别达到75.56%和68.92%准确率
- 可减少站点差异,揭示具有生物学意义的高判别性脑区
在功能磁共振成像(fMRI)开放数据集中,数据异质性通常由扫描流程差异、混杂效应及多中心人群多样性共同导致。这些因素削弱了表征学习效果,进而影响分类性能。为此,提出一种新型多中心对抗学习网络MSalNET用于fMRI精神疾病检测。首先引入节点信息组装(NIA)机制的表征学习模块,从功能连接(FC)中聚合水平与垂直方向的边信息,增强节点特征表达;其次设计站点级特征提取模块,基于个体FC数据学习,无需额外先验信息;最后构建对抗学习网络,通过新损失函数平衡个体分类与站点回归任务。在ABIDE与ADHD-200两个多中心数据集上评估,准确率分别达75.56%和68.92%,优于现有算法。站点回归结果表明该方法从数据驱动角度降低了站点间差异。NIA揭示的最判别性脑区与统计研究一致,部分破解了深度学习‘黑箱’问题。
原文摘要 · Abstract (English)
In open data sets of functional magnetic resonance imaging (fMRI), the heterogeneity of the data is typically attributed to a combination of factors, including differences in scanning procedures, the presence of confounding effects, and population diversities between multiple sites. These factors contribute to the diminished effectiveness of representation learning, which in turn affects the overall efficacy of subsequent classification procedures. To address these limitations, we propose a novel multi-site adversarial learning network (MSalNET) for fMRI-based mental disorder detection. Firstly, a representation learning module is introduced with a node information assembly (NIA) mechanism to better extract features from functional connectivity (FC). This mechanism aggregates edge information from both horizontal and vertical directions, effectively assembling node information. Secondly, to generalize the feature across sites, we proposed a site-level feature extraction module that can learn from individual FC data, which circumvents additional prior information. Lastly, an adversarial learning network is proposed as a means of balancing the trade-off between individual classification and site regression tasks, with the introduction of a novel loss function. The proposed method was evaluated on two multi-site fMRI datasets, i.e., Autism Brain Imaging Data Exchange (ABIDE) and ADHD-200. The results indicate that the proposed method achieves a better performance than other related algorithms with the accuracy of 75.56 and 68.92 in ABIDE and ADHD-200 datasets, respectively. Furthermore, the result of the site regression indicates that the proposed method reduces site variability from a data-driven perspective. The most discriminative brain regions revealed by NIA are consistent with statistical findings, uncovering the "black box" of deep learning to a certain extent.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。