用图神经网络和信息瓶颈法,自动找精神分裂症的脑功能生物标志物。
BrainIB++: Leveraging Graph Neural Networks and Information Bottleneck for Functional Brain Biomarkers in Schizophrenia
- 基于信息瓶颈原理,端到端识别关键脑区子图。
- 在三个数据集上诊断准确率优于9种现有方法。
- 发现视觉、感觉运动和高级认知网络异常,可解释性强。
精神疾病诊断模型研究日益受到关注。近年来,基于静息态功能磁共振成像(rs-fMRI)的机器学习分类器被用于识别区分精神障碍与健康对照的脑生物标志物。然而,传统机器学习模型常依赖大量特征工程,易引入人工偏差。尽管深度学习模型理论上可免于人工干预,但其可解释性差,难以获得可信且可解释的脑生物标志物,限制了临床应用。本文提出一种名为BrainIB++的端到端图神经网络框架,结合信息瓶颈(IB)原则,在训练过程中自动识别最具信息量的数据驱动脑区子图以实现解释。我们在三个多队列精神分裂症数据集上评估该模型性能,对比九种成熟脑网络分类方法,结果表明其诊断准确率始终领先,并展现出对未见数据的良好泛化能力。此外,模型识别出的子图与已知的精神分裂症临床生物标志物高度一致,尤其强调了视觉、感觉运动及高级认知功能网络的异常,显著提升模型可解释性,凸显其在真实诊疗中的应用价值。
原文摘要 · Abstract (English)
The development of diagnostic models is gaining traction in the field of psychiatric disorders. Recently, machine learning classifiers based on resting-state functional magnetic resonance imaging (rs-fMRI) have been developed to identify brain biomarkers that differentiate psychiatric disorders from healthy controls. However, conventional machine learning-based diagnostic models often depend on extensive feature engineering, which introduces bias through manual intervention. While deep learning models are expected to operate without manual involvement, their lack of interpretability poses significant challenges in obtaining explainable and reliable brain biomarkers to support diagnostic decisions, ultimately limiting their clinical applicability. In this study, we introduce an end-to-end innovative graph neural network framework named BrainIB++, which applies the information bottleneck (IB) principle to identify the most informative data-driven brain regions as subgraphs during model training for interpretation. We evaluate the performance of our model against nine established brain network classification methods across three multi-cohort schizophrenia datasets. It consistently demonstrates superior diagnostic accuracy and exhibits generalizability to unseen data. Furthermore, the subgraphs identified by our model also correspond with established clinical biomarkers in schizophrenia, particularly emphasizing abnormalities in the visual, sensorimotor, and higher cognition brain functional network. This alignment enhances the model's interpretability and underscores its relevance for real-world diagnostic applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。