用信息瓶颈原理提升脑图谱与多模态数据融合的可解释性诊断能力
Information Bottleneck-Guided Heterogeneous Graph Learning for Interpretable Neurodevelopmental Disorder Diagnosis
- 基于信息瓶颈思想设计双组件图神经网络,统一建模功能连接与跨模态融合
- 在多个NDD数据集上实现超90%分类准确率,同时识别出关键生物标志物
- 适合关注脑疾病可解释诊断的临床研究者与医学人工智能开发者
构建可解释的神经发育障碍(NDDs)诊断模型面临如何有效编码、解码和整合多模态神经影像数据的挑战。现有机器学习方法虽在脑网络分析中表现良好,但通常缺乏可解释性,尤其难以从功能磁共振成像(fMRI)数据中提取有意义的生物标志物,并建立影像特征与人口统计学特征间的清晰关联。此外,当前图神经网络在捕捉局部与全局功能连接模式的同时实现理论严谨的多模态融合方面存在局限。为此,我们提出可解释的信息瓶颈异构图神经网络(I2B-HGNN),通过信息瓶颈原则指导脑连接建模与跨模态特征融合。该框架包含两个互补组件:一是信息瓶颈图变换器(IBGraphFormer),结合基于Transformer的全局注意力与图神经网络,通过信息瓶颈引导的池化机制识别充分生物标志物;二是信息瓶颈异构图注意力网络(IB-HGAN),采用基于元路径的异构图学习并施加结构一致性约束,实现神经影像与人口统计学数据的可解释融合。实验结果表明,I2B-HGNN在诊断NDDs方面表现优异,兼具高分类准确率与可解释的生物标志物识别能力,并能有效分析非影像数据。
原文摘要 · Abstract (English)
Developing interpretable models for neurodevelopmental disorders (NDDs) diagnosis presents significant challenges in effectively encoding, decoding, and integrating multimodal neuroimaging data. While many existing machine learning approaches have shown promise in brain network analysis, they typically suffer from limited interpretability, particularly in extracting meaningful biomarkers from functional magnetic resonance imaging (fMRI) data and establishing clear relationships between imaging features and demographic characteristics. Besides, current graph neural network methodologies face limitations in capturing both local and global functional connectivity patterns while simultaneously achieving theoretically principled multimodal data fusion. To address these challenges, we propose the Interpretable Information Bottleneck Heterogeneous Graph Neural Network (I2B-HGNN), a unified framework that applies information bottleneck principles to guide both brain connectivity modeling and cross-modal feature integration. This framework comprises two complementary components. The first is the Information Bottleneck Graph Transformer (IBGraphFormer), which combines transformer-based global attention mechanisms with graph neural networks through information bottleneck-guided pooling to identify sufficient biomarkers. The second is the Information Bottleneck Heterogeneous Graph Attention Network (IB-HGAN), which employs meta-path-based heterogeneous graph learning with structural consistency constraints to achieve interpretable fusion of neuroimaging and demographic data. The experimental results demonstrate that I2B-HGNN achieves superior performance in diagnosing NDDs, exhibiting both high classification accuracy and the ability to provide interpretable biomarker identification while effectively analyzing non-imaging data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。