本方案介绍了脑部混合专家模型(Brain Mixture-of-Experts),这是一种用于脑部疾病计算机辅助诊断的自适应且可解释的脑电图-功能磁共振成像融合框架。该方法通过模态特异性专家、共享神经状态专家以及自适应路由机制,整合异质的脑电图(EEG)和功能磁共振成像(fMRI)表征,同时在模拟模态缺失条件下保持可解释性与推理能力。
本方案介绍了脑部混合专家模型(Brain Mixture-of-Experts),这是一种用于脑部疾病计算机辅助诊断的自适应且可解释的脑电图-功能磁共振成像融合框架。该方法通过模态特异性专家、共享神经状态专家以及自适应路由机制,整合异质的脑电图(EEG)和功能磁共振成像(fMRI)表征,同时在模拟模态缺失条件下保持可解释性与推理能力。
脑电图(EEG)和功能磁共振成像(fMRI)提供了关于脑功能的互补信息,在检测多种脑部疾病中的功能异常方面已展现出显著潜力。然而,EEG与fMRI在信号特征和表征空间上的显著差异,给有效的多模态融合带来了严峻挑战,从而限制了传统固定模型在脑部疾病计算机辅助诊断中的准确性。本研究提出了一种自适应且可解释的EEG-fMRI融合框架——脑混合专家模型(Brain Mixture-of-Experts, BrainMoE),通过模态特异性专家、共享神经状态专家以及自适应路由机制来整合多模态脑特征。BrainMoE首先将EEG和fMRI信号投影到统一的Desikan-Killiany(DK)图谱感兴趣区域(ROI)空间,然后利用图编码器提取模态特异的脑网络表征。软路由模块生成路由表征,融合模块中的专家门(Expert Gate)则生成样本特异的权重,用于整合EEG、fMRI和共享神经状态专家的表征。为应对数据采集不完整的情况,模型引入了模态状态掩码和缺失模态标记,使得同一训练模型可执行完整的EEG-fMRI、仅EEG或仅fMRI推理。最后,通过节点遮蔽分析为EEG和fMRI衍生的预测结果提供ROI水平的归因图谱。该框架在Healthy Brain Network(HBN)数据集上进行了评估,涵盖五种二分类脑部疾病诊断任务,包括重度抑郁症、焦虑症、阅读障碍、自闭症谱系障碍和注意缺陷多动障碍。BrainMoE优于当前最先进的对比算法,平均AUC达到86.9 ± 3.0%,消融实验验证了路由机制与专家融合组件的有效性。此外,可解释性分析识别出与既往神经影像学研究结果一致的、在组水平上对疾病分类有贡献的ROI区域。该方法通过解决异构EEG-fMRI神经表征的整合难题,在模拟模态缺失条件下仍保持可解释性和推理能力,从而支持脑部疾病的计算机辅助诊断。
脑部疾病涉及神经活动和脑网络组织的复杂变化,单一成像模态难以对其进行充分表征1源重建后2脑电图(EEG)提供区域水平的电生理活动信息,而功能磁共振成像(fMRI)则捕捉区域水平的功能连接,二者为脑功能研究提供了互补的视角。为促进两者的整合,先前的多模态研究已将这两种模态的数据共同映射至包含68个区域的Desikan-Killiany(DK)图谱3尽管这种空间对应关系并不意味着时间分辨率或生理来源上的等同性,但它为节点水平的图融合、固定的图维度以及一致的感兴趣区域(ROI)水平解释提供了统一的解剖学索引,同时保留了模态特异性的信息。整合这些在解剖结构上对齐但具有模态特异性的表征,有助于丰富与疾病相关的脑信号表征,并支持计算机辅助的脑部疾病诊断3,4.
尽管脑电图-功能磁共振成像(EEG-fMRI)融合技术在脑部疾病计算机辅助诊断方面具有潜力,但两种模态之间的异质性为其有效整合带来了技术挑战。传统的机器学习方法(如支持向量机SVM和多层感知机MLP)可提供基础分类模型,但在捕捉非线性跨模态相互作用方面能力有限。通用深度学习模型(包括图神经网络GNN)5ResNet6和 Transformer7 架构能够提供更强的表征学习能力,但并非专门针对脑区水平的脑图或模态-状态建模而设计。近期先进的模型3,8,9 非线性复杂性增加,以更准确地建模脑网络。BrainNetCNN8 被引入以将卷积操作适配到脑连接矩阵;BrainGNN9 进一步利用图神经网络和池化对脑区(ROI)级图拓扑结构进行建模;以及 BNT10 随后通过基于变换器的多层级注意力机制加强了功能脑网络分析。然而,这些进展仍主要针对单模态场景开发。MultiEpilepsyNet11 然后通过联邦混合框架将多模态学习扩展到EEG-MRI癫痫发作检测,其MRI端的EpiSkullNet++模块改进了脑部分割与预处理。SZAtt-Net12 随后开发了一种多模态精神分裂症分类模型,该模型结合了卷积神经网络(CNN)、双向门控循环单元(BiGRU)和多层感知机(MLP)模块,并引入了通道、自注意力、空间和时间注意力机制。然而,这些方法仍局限于特定任务,依赖于相对固定的融合设计,未能显式支持在完整模态与缺失模态状态之间的自适应路由。
混合专家(Mixture-of-Experts, MoE)模型13,14在多模态学习中日益受到关注,因其能够通过专门化的专家模块处理异构信息源,并通过路由机制动态组合。Softmax门控可提供依赖输入的归一化专家权重,并已在理论上对其收敛速率进行了表征15。相关的多门控架构进一步表明,在大规模多任务学习中,独立的门控机制能够学习共享专家的任务依赖性组合16。在神经科学研究与脑科学领域,MoE的多种变体17,18,19已被越来越多地用于促进异构特征融合。dFCExpert17利用模块化和基于状态的专家模型,从fMRI数据中建模动态功能连接模式。EvoMoE18进一步采用门控网络为独立于用户的SSVEP-EEG分类选择合适的专家。NeuroMoE++19探索了针对患者的自适应多模态融合方法,用于神经系统疾病的分类。尽管这些模型取得了成功,但它们通常依赖于粗粒度的混合与离散路由,忽略了跨模态神经状态高度同步的本质,从而阻碍了对脑部疾病诊断中细微但具有信息量的模式的检测。此外,由于缺乏专门机制来从统一的共享神经状态中解耦模态特异性细微差异,这些模型的可解释性有限,并且在关键模态(如fMRI或EEG)缺失时会出现性能下降。
为解决这些局限性,本研究提出了脑部混合专家模型(Brain Mixture-of-Experts, BrainMoE),这是一种自适应且可解释的脑电图-功能磁共振成像(EEG-fMRI)融合框架,用于计算机辅助诊断五类脑部疾病,包括重度抑郁障碍(MDD)、焦虑障碍(ANX)、伴阅读障碍的特定学习障碍(RI)、自闭症谱系障碍(ASD)以及注意缺陷多动障碍(ADHD)。该方案首先将源重建的EEG与fMRI特征对齐至DK图谱20的感兴趣区域(ROI)空间,随后分别为两种模态构建基于图的模态表征。BrainMoE包含一个EEG专家模块、一个fMRI专家模块以及一个共享的神经状态专家模块,并通过软路由模块整合,以融合模态特异性信息与共享信息。通过引入模态状态掩码和缺失模态标记,可在单一训练模型内实现完整的EEG-fMRI、仅EEG以及仅fMRI推理。为支持生物学可解释性,该方案进一步应用节点遮蔽分析方法,即逐一选择性遮蔽DK图谱中的每个ROI,并利用疾病预测概率的变化来估计由EEG和fMRI导出的各脑区贡献度。本文详细描述了数据对齐、模型构建、训练、评估以及基于节点遮蔽的ROI解释的完整工作流程,提供了一种通过异构EEG-fMRI脑信号融合实现脑部疾病计算机辅助诊断的自适应且可解释的策略。为促进研究可重复性及未来拓展,公开的GitHub代码库提供了BrainMoE模型、训练与评估代码,而EEG和fMRI的预处理则使用公开的第三方软件完成。代码库地址为 https://github.com/zhongruizhe123/BrainMoE。
本研究使用了来自健康脑网络(Healthy Brain Network, HBN)数据库的去标识化数据21,重点关注五种不同的临床障碍,用于后续的诊断分类任务。HBN 项目此前已从所有参与单位和受试者获得伦理批准及书面知情同意。脑电图(EEG)和功能磁共振成像(fMRI)记录在不同时间段分别采集,而非同步进行,并通过 HBN 中提供的受试者和会话标识符进行配对。
1. 准备计算环境和输入数据
2. 将 EEG 和 fMRI 特征配准到共享的解剖空间
3. 构建疾病特异性的二分类任务
4. 构建用于计算机辅助脑疾病诊断的 BrainMoE 架构
注意:BrainMoE 架构被设计为一种紧凑的缺失模态 EEG-fMRI 融合框架,该框架结合了图编码器、共享神经状态软路由以及基于专家的特征整合,用于疾病特异性的二分类诊断。整体架构如图1所示,实现代码见补充文件6。













5. 训练五种用于脑部疾病诊断的 BrainMoE 模型


6. 评估完整模态与缺失模态的推理状态





7. 在五种脑部疾病中进行基于节点遮蔽的组水平归因分析
BrainMoE 在不同疾病和模态状态下的性能
该方案生成了五种疾病特异性的健康与疾病对比 BrainMoE 分类器,并为完整的 EEG-fMRI、仅 EEG 和仅 fMRI 推断状态生成了预测结果表。在全部五项任务中,完整 EEG-fMRI 状态均实现了持续较高的判别性能,AUC 值从癫痫发作间期(RI)的 84.4 ± 3.2% 到自闭症谱系障碍(ASD)的 88.4 ± 3.8% 不等(表 1)。五项任务在完整状态下的宏平均性能达到 AUC 86.9 ± 3.0%、准确率 81.4 ± 3.1%、平衡准确率 81.4 ± 2.5% 和 F1 分数 81.2 ± 3.0%。
在模拟缺失模态条件下,当某一输入模态被屏蔽时,BrainMoE 仍能保持可用的性能。在仅使用 EEG 的状态下,模型的宏平均 AUC 达到 83.1 ± 3.7%,其中 ADHD 的 EEG-only AUC 最高,为 87.9 ± 2.8%。在仅使用 fMRI 的状态下,宏平均 AUC 为 80.0 ± 3.9%。这些性能范围以及预期的完整模态状态下的性能优势,为成功实施提供了实用的基准,表明训练好的 BrainMoE 框架能够在无需为每种模态单独构建模型的情况下,执行完整模态、仅 EEG 和仅 fMRI 的推理。
一个典型的次优结果是无法重现预期的全模态性能优势,例如,当全EEG-fMRI的AUC低于仅使用EEG或仅使用fMRI的AUC时。相比之下,成功的实施应能重现全模态优势以及表1中报告的基准性能范围。当观察到次优模式时,在解释模型输出之前,应验证H5输入的维度、DK脑区的顺序、模态可用性掩码的分配以及保存的交叉验证划分。
基准比较
所提出的 BrainMoE 模型与经典机器学习方法(支持向量机(SVM)和多层感知机(MLP))、通用深度学习方法(Transformer7、3D-CNN22 和 ResNet6)、先进深度学习方法(BrainNetCNN8、BNT10、BrainGNN9、MultiEpilepsyNet11、SZAtt-Net12)以及基于 MoE 的深度学习方法(dFCExpert17、EvoMoE18 和 NeuroMoE++19)进行了比较(表 2)。在所有比较方法中,BrainMoE 取得了最高的平均 AUC,为 86.9 ± 3.0%。
经典机器学习方法表现出较低的平均性能,其中支持向量机(SVM)的平均AUC为66.7 ± 4.5%,多层感知机(MLP)为64.6 ± 6.3%。通用深度学习方法表现不一,ResNet的平均AUC为71.0 ± 3.3%,Transformer为63.8 ± 4.7%。在先进的深度学习基线模型中,BNT、BrainGNN、MultiEpilepsyNet和SZAtt-Net优于大多数经典方法和通用深度学习方法,但其平均AUC仍低于BrainMoE。
为了为基准比较提供统计学支持,BrainMoE 在各项性能指标上均与最强的基线模型进行了比较(表3)。BrainMoE 在AUC和平衡准确率(BA)上优于NeuroMoE++,在F1分数和准确率上优于SZAtt-Net。所有比较在经过Holm校正后仍具有统计学意义。
按性别和采集地点进行的亚组分析
任务特异性队列特征包括保留的多模态记录数量、病例与对照的比例、唯一参与者数量、年龄概况、性别分布以及采集地点分布,详见队列特征表(表 4)。病例与对照的比例基于多模态记录数量计算,而人口统计学特征和采集地点特征则在唯一参与者层面进行汇总。
为了评估性别和采集地点的潜在影响,按这些因素对 BrainMoE 的性能进行了分层分析(表5)。男性亚组的平均 AUC、BA、F1 分数和准确率均高于女性亚组,相应的未校正 Welch p 值范围为 0.089 至 0.321。同样,RUBIC 亚组的平均性能高于 Staten Island 亚组,p 值范围为 0.055 至 0.309。在这些分析中,未检测到具有统计学意义的亚组差异。
BrainMoE 各组件的消融分析
为评估模态可用性掩码、卷积路由模块、共享专家模块以及 MoE 专家融合设计的贡献,进行了消融实验(表6)。移除掩码嵌入后,平均 AUC 下降至 79.2 ± 3.1%;将卷积路由模块替换为 MLP 路由模块后,AUC 下降至 79.3 ± 3.7%。移除共享专家模块后,仅使用 fMRI 的 AUC 降至 73.5 ± 3.3%,为所有测试变体中最低。移除所有 MoE 专家模块也使整体平衡准确率降至 74.2 ± 3.0%。这些消融结果表明,完整的 BrainMoE 设计在模态完整和模态缺失状态下均实现了最优的整体性能,而掩码嵌入、卷积路由模块、共享专家模块以及 MoE 专家融合设计各自均对最终模型性能有所贡献。
五种脑部疾病的感兴趣区域(ROI)水平可解释性结果
为探究 BrainMoE 预测背后的区域贡献,在完整的 EEG-fMRI 状态下,对正确分类的疾病阳性样本进行了节点遮蔽归因分析。通过测量遮蔽每个 DK 图谱区域后目标疾病概率的下降程度,分别对 EEG 和 fMRI 衍生的 ROI 贡献进行排序。该分析揭示了在五种疾病任务中模态特异性的贡献模式(图 2)。在 fMRI 衍生的归因中,排名最高的区域分别为:抑郁症(MDD)中的左侧后扣带回、焦虑症(ANX)中的右侧围距状皮层、癫痫发作间期(RI)中的左侧海马旁回皮层、自闭症谱系障碍(ASD)中的右侧三角部额下回,以及注意力缺陷多动障碍(ADHD)中的左侧内嗅皮层。在 EEG 衍生的归因中,排名最高的区域分别为:MDD 中的左侧颞上沟上缘、ANX 中的左侧中央前回皮层、RI 中的左侧楔叶、ASD 中的左侧舌回皮层,以及 ADHD 中的右侧岛叶。每种疾病中排名最高的 EEG 和 fMRI 衍生 ROI 在皮层表面的可视化结果见 图 3。这些结果表明,BrainMoE 在保留 EEG 衍生和 fMRI 衍生表征各自归因特征的同时,实现了 ROI 水平的可解释性。对于正确分类的疾病阳性样本,通过 50 个独立保留折中的 Top-1 ROI 出现频率评估了跨折稳定性(表 7)。观察到的频率范围为 36% 至 68%,显著高于理论上的随机选择基准值 1/68(1.47%),支持基于相对排名而非绝对贡献幅度进行解释。与先前神经影像学研究结果的比较为事后分析,仅用于为归因结果提供背景信息,不作为独立验证。

图1:用于计算机辅助脑疾病诊断的 BrainMoE 框架概览,采用自适应 EEG-fMRI 融合方法。 来源对齐的 EEG 和 fMRI ROI 特征分别由独立的图编码器处理。所得表征被传递至 EEG 专家模块和 fMRI 专家模块,并与模态可用性掩码一同输入共享的软路由模块。模态表征与路由表征在共享的神经状态专家模块中进行融合与处理。三个专家模块的输出随后在融合模块中合并,并传递至诊断分类头,以生成健康对照(HC)和目标疾病的概率。请点击此处查看该图的放大版本。

图 2:基于节点遮蔽的脑区贡献度分析。 在完整的 EEG-fMRI 推断状态下,分别可视化了每种疾病任务中 EEG 和 fMRI 源脑区(ROI)贡献度排名前 10 的结果。图中面板 (A–E) 分别显示了抑郁症(MDD)、焦虑症(ANX)、物质使用障碍(RI)、自闭症谱系障碍(ASD)和注意缺陷多动障碍(ADHD)的 fMRI 源分析结果,而面板 (F–J) 则按相同顺序显示了相应的 EEG 源分析结果。每张图中仅标注了贡献度最高的脑区。ROI 贡献度定义为遮蔽相应 DK 图谱区域后,目标疾病预测概率的下降程度。请点击此处查看该图的放大版本。

图3:五种脑部疾病中基于皮层感兴趣区域(ROI)水平的可解释性图谱。 图中(A–E)分别展示重度抑郁障碍(MDD)、焦虑症(ANX)、物质使用障碍(RI)、自闭症谱系障碍(ASD)和注意缺陷多动障碍(ADHD)。每个子图在DK皮层表面上以红色显示由脑电图(EEG)推导出的排名最高的感兴趣区域(ROI),以橙色显示由功能性磁共振成像(fMRI)推导出的排名最高的感兴趣区域(ROI)。颜色仅表示模态类型,而非贡献程度大小,因此未应用定量的颜色标尺。前缀“lh”和“rh”分别表示左半球和右半球。请点击此处查看该图的放大版本。
| 疾病任务 | 状态 | AUC | 准确率 | 平衡准确率 | F1分数 | 敏感性 | 特异性 |
| MDD | 完整 | 88.0 ± 3.2% | 84.8 ± 3.7% | 82.0 ± 2.3% | 76.2 ± 4.1% | 72.7 ± 3.0% | 91.3 ± 4.7% |
| MDD | 仅EEG | 83.4 ± 3.4% | 81.8 ± 4.5% | 77.9 ± 3.0% | 73.7 ± 3.9% | 68.2 ± 2.8% | 87.6 ± 4.2% |
| MDD | 仅fMRI | 82.1 ± 3.3% | 78.5 ± 4.7% | 74.0 ± 4.1% | 71.8 ± 4.9% | 66.6 ± 3.5% | 81.3 ± 5.6% |
| ANX | 完整 | 85.5 ± 2.5% | 76.1 ± 2.4% | 78.2 ± 2.0% | 79.4 ± 3.3% | 78.1 ± 3.6% | 78.2 ± 3.1% |
| ANX | 仅EEG | 81.7 ± 3.9% | 73.2 ± 3.2% | 75.6 ± 3.1% | 78.1 ± 3.9% | 76.6 ± 3.8% | 74.6 ± 4.0% |
| ANX | 仅fMRI | 74.8 ± 4.1% | 74.4 ± 3.7% | 72.8 ± 3.6% | 76.4 ± 4.2% | 72.6 ± 3.9% | 73.0 ± 4.4% |
| RI | 完整 | 84.4 ± 3.2% | 76.5 ± 3.0% | 77.3 ± 2.9% | 76.7 ± 2.8% | 77.9 ± 3.1% | 76.7 ± 3.3% |
| RI | 仅EEG | 80.2 ± 4.0% | 71.5 ± 4.4% | 73.4 ± 3.5% | 73.0 ± 3.4% | 73.8 ± 3.9% | 72.9 ± 3.3% |
| RI | 仅fMRI | 81.3 ± 4.5% | 71.1 ± 3.9% | 67.9 ± 3.7% | 68.5 ± 3.2% | 69.4 ± 4.5% | 66.3 ± 4.2% |
| ASD | 完整 | 88.4 ± 3.8% | 79.5 ± 3.5% | 81.4 ± 3.0% | 81.0 ± 2.8% | 83.4 ± 2.9% | 79.4 ± 3.0% |
| ASD | 仅EEG | 82.5 ± 4.3% | 77.3 ± 4.1% | 78.2 ± 3.7% | 78.6 ± 3.5% | 79.3 ± 3.4% | 77.1 ± 4.1% |
| ASD | 仅fMRI | 81.3 ± 4.4% | 73.8 ± 3.7% | 74.4 ± 3.4% | 75.0 ± 3.2% | 76.1 ± 4.2% | 72.6 ± 4.5% |
| ADHD | 完整 | 88.2 ± 2.2% | 90.3 ± 2.7% | 88.4 ± 2.5% | 92.8 ± 1.9% | 87.0 ± 2.4% | 89.7 ± 2.5% |
| ADHD | 仅EEG | 87.9 ± 2.8% | 88.1 ± 2.6% | 86.4 ± 3.1% | 90.2 ± 3.4% | 85.1 ± 3.1% | 87.7 ± 3.0% |
| ADHD | 仅fMRI | 80.5 ± 3.2% | 85.7 ± 3.0% | 84.7 ± 3.5% | 87.9 ± 4.1% | 83.2 ± 3.6% | 86.2 ± 3.4% |
表1:BrainMoE在不同疾病任务和模态可用性状态下的分类性能。 BrainMoE在五种健康与疾病分类任务中,于完整的EEG-fMRI、仅EEG和仅fMRI推断状态下的性能表现。指标以均值±标准差形式报告,包括受试者工作特征曲线下面积(AUC)、准确率、平衡准确率(BA)、F1分数、敏感性和特异性。
| 方法 | 方法组 | 平均 AUC | 平均 BA | 平均 F1 | 平均准确率 |
| SVM | 经典机器学习 | 66.7 ± 4.5% | 70.6 ± 3.4% | 61.7 ± 5.8% | 70.7 ± 4.6% |
| MLP | 经典机器学习 | 64.6 ± 6.3% | 63.5 ± 8.2% | 64.5 ± 6.1% | 64.2 ± 6.2% |
| Transformer | 通用深度学习 | 63.8 ± 4.7% | 64.4 ± 4.8% | 65.2 ± 5.1% | 66.1 ± 4.9% |
| 3D-CNN | 通用深度学习 | 65.3 ± 4.3% | 65.2 ± 4.0% | 64.8 ± 4.5% | 60.0 ± 4.2% |
| ResNet | 通用深度学习 | 71.0 ± 3.3% | 70.2 ± 3.6% | 67.5 ± 3.8% | 69.3 ± 3.0% |
| BrainNetCNN | 高级深度学习 | 70.2 ± 3.6% | 70.9 ± 3.1% | 69.2 ± 3.6% | 71.2 ± 3.1% |
| BNT | 高级深度学习 | 73.6 ± 3.1% | 76.4 ± 2.7% | 74.7 ± 2.3% | 76.6 ± 2.6% |
| BrainGNN | 高级深度学习 | 72.4 ± 2.8% | 72.7 ± 2.3% | 71.0 ± 3.4% | 72.3 ± 2.5% |
| MultiEpilepsyNet | 高级深度学习 | 78.3 ± 3.7% | 75.2 ± 3.5% | 76.5 ± 3.6% | 77.2 ± 3.3% |
| SZAtt-Net | 高级深度学习 | 78.7 ± 3.2% | 76.1 ± 3.8% | 77.4 ± 3.9% | 78.1 ± 3.4% |
| dFCExpert | 基于MoE的深度学习 | 80.1 ± 3.2% | 77.3 ± 2.9% | 76.7 ± 3.1% | 77.6 ± 2.9% |
| EvoMoE | 基于MoE的深度学习 | 79.6 ± 3.6% | 76.2 ± 3.1% | 76.2 ± 3.3% | 76.5 ± 3.2% |
| NeuroMoE++ | 基于MoE的深度学习 | 81.5 ± 2.8% | 77.9 ± 2.7% | 77.1 ± 3.4% | 78.0 ± 2.7% |
| BrainMoE (ours) | 基于MoE的深度学习 | 86.9 ± 3.0% | 81.4 ± 2.5% | 81.2 ± 3.0% | 81.4 ± 3.1% |
表2:BrainMoE与经典机器学习方法、通用深度学习模型以及先进的神经影像深度学习架构在分类性能上的比较(以平均值表示)。结果汇总了各项评估的疾病分类任务,以AUC、BA、F1分数和准确率的均值±标准差形式报告。
| 比较 | 平均 AUC | 平均 BA | 平均 F1 | 平均准确率 |
| 最强基线模型 | NeuroMoE++ | NeuroMoE++ | SZAtt-Net | SZAtt-Net |
| 最强基线模型性能 | 81.5 ± 2.8% | 77.9 ± 2.7% | 77.4 ± 3.9% | 78.1 ± 3.4% |
| BrainMoE | 86.9 ± 3.0% | 81.4 ± 2.5% | 81.2 ± 3.0% | 81.4 ± 3.1% |
| 差异 | +5.4 | +3.5 | +3.8 | +3.3 |
| t 检验 p 值 | 0.0006 | 0.0065 | 0.0187 | 0.0276 |
| 经 Holm 校正后的 p 值* | p < 0.01 | p < 0.05 | p < 0.05 | p < 0.05 |
| Cohen's dz | 1.63 | 1.11 | 0.91 | 0.83 |
表3:BrainMoE与各性能指标最强基线方法的比较。 P值通过双侧配对t检验计算,并使用Holm法进行校正。Cohen的dz表示标准化的配对差异。
| 队列 | 多模态记录数 (n) | 疾病与对照比例 | 唯一参与者数 (n) | 年龄范围(岁) | 性别(男性/女性) | 数据采集地点(史泰登岛/RUBIC) |
| HC | 115 | -- | 75 | 5.02–21.90 | 35/40 | 26/49 |
| MDD | 52 | 0.45:1 | 33 | 8.36–19.73 | 14/19 | 15/18 |
| ANX | 156 | 1.36:1 | 98 | 5.53–21.00 | 46/52 | 45/53 |
| RI | 141 | 1.23:1 | 85 | 5.75–19.66 | 47/38 | 39/46 |
| ASD | 82 | 0.71:1 | 51 | 5.66–19.79 | 45/6 | 27/24 |
| ADHD | 555 | 4.83:1 | 338 | 5.04–21.72 | 241/97 | 136/202 |
表4:五项疾病特异性分类任务中所用研究队列的特征。 该表格报告了保留的多模态记录数量、疾病组与对照组的比例、唯一参与者数量、人口统计学特征以及数据采集站点的分布情况。
| 亚组 | 数量 | 平均 AUC | 平均 BA | 平均 F1 | 平均准确率 |
| 性别 | |||||
| 男性 | 705 | 87.2 ± 3.7 | 83.4 ± 3.2 | 82.1 ± 3.5 | 82.5 ± 3.8 |
| 女性 | 396 | 85.6 ± 3.3 | 81.2 ± 4.0 | 79.7 ± 3.8 | 79.6 ± 3.4 |
| 差异 | +1.6 | +2.2 | +2.4 | +2.9 | |
| Welch p | -- | 0.321 | 0.192 | 0.159 | 0.089 |
| 采集地点 | |||||
| RUBIC | 674 | 87.6 ± 4.0 | 83.3 ± 3.1 | 82.3 ± 3.7 | 83.4 ± 3.5 |
| Staten Island | 427 | 85.2 ± 3.6 | 81.7 ± 3.7 | 79.4 ± 3.4 | 80.1 ± 3.7 |
| 差异 | +2.4 | +1.6 | +2.9 | +3.3 | |
| Welch p | -- | 0.176 | 0.309 | 0.085 | 0.055 |
表5:按性别和采集地点分层的 BrainMoE 性能。 结果以10次5折交叉验证的均值±标准差形式报告。差异表示第一亚组减去第二亚组的值,P值通过双侧Welch t检验获得。
| 变体 | 完整 AUC | 仅 EEG AUC | 仅 fMRI AUC | 平均 AUC | 平均 BA |
| 无掩码嵌入 | 83.2 ± 2.9% | 79.5 ± 3.9% | 76.4 ± 3.7% | 79.2 ± 3.1% | 76.1 ± 3.4% |
| 无卷积路由模块 | 84.6 ± 3.6% | 80.7 ± 3.5% | 78.2 ± 4.2% | 81.6 ± 3.3% | 75.7 ± 3.1% |
| MLP 路由模块 | 82.8 ± 2.7% | 80.1 ± 4.1% | 76.6 ± 4.4% | 79.3 ± 3.7% | 74.1 ± 3.8% |
| 无共享专家模块 | 83.3 ± 3.6% | 80.8 ± 4.0% | 73.5 ± 3.3% | 80.1 ± 3.6% | 75.5 ± 3.2% |
| 无 MoE 专家模块 | 81.9 ± 3.3% | 78.7 ± 3.6% | 78.0 ± 3.8% | 80.8 ± 4.1% | 74.2 ± 3.0% |
| BrainMoE(本研究) | 86.9 ± 3.0% | 83.1 ± 3.7% | 80.0 ± 3.9% | 86.9 ± 3.0% | 81.4 ± 2.5% |
表6:关键BrainMoE组件的消融分析。 消融实验结果展示了模态可用性掩码、卷积路由模块、共享专家模块以及MoE专家融合设计的贡献。每种变体均在完整的EEG-fMRI、仅EEG和仅fMRI推理条件下进行评估,以平均AUC和平均BA总结整体性能。
| 疾病 | 脑电图:排名最高的感兴趣区域 | 脑电图:第一频率,n/N (%) | 功能磁共振成像:排名最高的感兴趣区域 | 功能磁共振成像:第一频率,n/N (%) | 随机参考值 (%) |
| MDD | 左侧颞上沟 banks | 22/50 (44%) | 左侧后扣带皮层 | 25/50 (50%) | 1.47 |
| ANX | 左侧中央前回皮层 | 18/50 (36%) | 右侧围距状皮层 | 21/50 (42%) | 1.47 |
| RI | 左侧楔叶 | 26/50 (52%) | 左侧海马旁回皮层 | 29/50 (58%) | 1.47 |
| ASD | 左侧舌回皮层 | 27/50 (54%) | 右侧三角部 | 28/50 (56%) | 1.47 |
| ADHD | 右侧岛叶 | 31/50 (62%) | 左侧内嗅皮层 | 34/50 (68%) | 1.47 |
表7:基于脑电图和功能磁共振成像所提取的高排名感兴趣区域的交叉验证稳定性。 首位频率表示在50次交叉验证分析中,所报告的感兴趣区域排名首位的次数及其占比。理论上随机选择的参考值为1/68(1.47%)。
补充文件 1:H5 输入检查脚本。 在进行 BrainMoE 训练之前,用于验证所需 H5 输入键、数据类型、诊断标签以及与 DK-atlas 兼容的输入维度的 Python 脚本。 请点击此处下载该文件。
补充文件 2:fMRI 预处理脚本。 用于 fMRI 预处理的 C-PAC 配置和执行文件,包括初始体积去除、运动和失真校正、配准与标准化、干扰协变量回归、时间滤波以及空间平滑。 请点击此处下载该文件。
补充文件3:FreeSurfer处理脚本。用于处理结构MRI数据、将Desikan-Killiany皮层分区图共配准至个体fMRI空间,以及提取ROI层面的fMRI信号的脚本。 请点击此处下载该文件。
补充文件4:脑电图预处理脚本。 用于脑电图预处理的MATLAB/EEGLAB脚本,包括滤波、伪迹成分识别与去除以及重参考。 请点击此处下载该文件。
补充文件 5:MNE-Python 源定位与特征提取脚本。用于 EEG 源定位、DK 图谱 ROI 提取以及生成作为 BrainMoE 输入的 ROI 水平 EEG 特征的 Python 脚本。 请点击此处下载该文件。
补充文件6:BrainMoE 实现代码。 包含 BrainMoE 架构、图编码器、模态-状态处理、专家路由与融合、模型训练、评估及消融实验变体的 Python 代码和配置文件。 请点击此处下载该文件。
补充文件 7:节点遮蔽归因代码。 用于模态特异性节点遮蔽分析、ROI 贡献分数计算、基于 EEG 和 fMRI 衍生的 ROI 排序以及生成归因输出的 Python 代码。 请点击此处下载该文件。
多模态脑信号分析已成为计算机辅助脑部疾病诊断的重要方向,因为脑电图(EEG)和功能磁共振成像(fMRI)提供了关于神经活动的互补信息。在基准比较中,支持向量机(SVM)和多层感知机(MLP)等经典机器学习方法提供了基本的诊断性能,但在建模分层结构、图结构以及跨模态特征交互方面能力有限。通用深度学习模型(包括3D-CNN、ResNet和Transformer)具有更强的非线性建模能力,但这些架构并非专门为EEG-fMRI融合或脑网络表征而设计。先进的深度学习方法相比大多数经典和通用模型取得了更好的性能,但其中许多方法仍依赖固定的特征融合策略,且未明确区分模态特异性信息与共享的神经状态信息。
本研究提出了BrainMoE,通过结合基于图的模态编码器、模态特异性专家、共享的神经状态专家以及自适应路由机制,以解决这一融合问题。该设计使得从EEG和fMRI导出的表征能够被分别建模,再通过专家层级的融合进行整合。通过引入模态-状态掩码和缺失模态标记,同一训练模型还可在无需为每种缺失模态情况构建独立模型的前提下,执行仅EEG和仅fMRI的推理。实验结果表明,BrainMoE在五项二分类疾病诊断任务中均取得了最优的整体性能,并在仅EEG和仅fMRI状态下均保持了可用的性能表现。消融分析进一步验证了掩码嵌入、卷积路由器、共享专家以及MoE专家融合模块的贡献。这些发现表明,性能的提升并非源于单一组件,而是图编码、模态-状态建模、自适应路由与专家融合协同设计的结果。
关键实验步骤与故障排除
关键实验步骤包括在 EEG 和 fMRI 节点及图矩阵中保持相同的 68 区域 DK 顺序,为每位参与者独立应用预定义的预处理设置,以及实施参与者水平的交叉验证,以确保同一参与者的全部记录保留在同一折叠中。模态可用性掩码还必须与每种推理状态下的所提供输入相对应。
如果推理失败或未能复现预期的全状态性能优势,首先应验证所需的 H5 键、EEG 和 fMRI 矩阵维度、DK 脑区顺序、模态可用性掩码的分配以及保存的交叉验证划分。对于缺少必要键、维度无效或脑区顺序不一致的文件,应在训练或评估前予以排除。该框架可修改以适配其他皮层分割方案或 EEG/fMRI 特征表示形式,前提是两种模态均映射到一致的 ROI 顺序,并相应调整模型输入维度。疾病特异性分类头也可适配于其他二分类任务,同时保留图编码与专家融合框架。此类修改需重新训练与验证,而不能直接应用本文报告的模型。
基于节点遮蔽的可解释性分析
节点遮蔽分析进一步提供了脑区水平上对 BrainMoE 预测结果的可解释性,排名最高的疾病相关脑区见图3。BrainMoE 通过脑电图(EEG)识别出的左侧后部脑区,与先前基于体素的荟萃分析23证据一致,该研究报道了重度抑郁障碍(MDD)患者后部皮层区域的内在脑活动发生改变。通过 EEG/fMRI 识别出的左侧中央前回和右侧距状旁脑区,也与既往焦虑障碍的神经影像学证据相符:一项皮层厚度的荟萃分析24报道,焦虑障碍患者左侧中央前回皮层厚度增加;而一项关于社交焦虑障碍的结构协变网络研究25则报道,右侧距状旁皮层参与了节点中心性的异常。对于阅读障碍,EEG 衍生的左侧楔叶脑区与一项全脑连接性研究26结果一致,该研究报道了阅读障碍患者左侧楔叶连接性的改变;而 fMRI 衍生的左侧海马旁回脑区则与另一项研究27相符,该研究报道了具有特定阅读理解缺陷的青少年存在海马旁回/海马区耦合异常。在自闭症谱系障碍任务中,EEG 衍生归因所突出的左侧舌回脑区,呼应了先前静息态 fMRI 的证据28,即患有自闭症谱系障碍(ASD)的青春期前男孩左侧舌回局部一致性(ReHo)降低。fMRI 衍生的右侧三角部脑区也具有生物学合理性,因为在自闭症儿童中已有研究报道29,额下回右侧三角部的低频振幅(ALFF)发生改变。对于注意缺陷多动障碍(ADHD),EEG 衍生的右侧岛叶脑区与结构 MRI 证据30相符,该证据显示 ADHD 青少年的前岛叶体积减小,尤其涉及右侧岛叶短回。fMRI 衍生的左侧内嗅皮层脑区可能反映了一种更具亚型特异性的发现,因为另一项《心理医学》(Psychological Medicine)研究31在经过 FDR 校正后,报道了 ADHD-C 亚组中左侧内嗅皮层体积降低。然而,这些结果的解释需考虑脑区间相互依赖性,因为相关联的脑区信号可能阻碍单节点遮蔽完全分离出单个脑区的贡献,从而导致对实际贡献的估计偏于保守。
局限性与未来研究方向
尽管本研究采用重复5折交叉验证以获得内部性能估计,但未来研究若采用嵌套交叉验证或独立的外部验证,将进一步增强对模型选择稳定性与泛化能力的评估。由于仅使用EEG和仅使用fMRI的输出是通过对完整的多模态数据掩蔽某一模态生成的,且未在外部验证队列中进行评估,因此未来的研究应纳入外部单模态验证队列,以评估其泛化能力。另一个局限性在于,健康对照与单一疾病对比的设计无法反映共病情况,这限制了临床泛化性,并提示未来应开展多标签分类与鉴别诊断的相关研究。未来的工作还可利用考虑信号泄漏的连接性估计方法,评估源空间EEG关联结果的稳健性。尽管采用的DK脑区划分方案为多模态融合提供了基于解剖结构的接口,但这一方法仍是一种建模假设,可能无法充分捕捉不同模态在时间分辨率和生理来源上的特异性差异。除了本研究涉及的五种疾病外,该框架还可拓展至其他涉及解剖对齐的多模态脑数据的神经或精神疾病分类任务,并可进一步推广至多标签分类或鉴别诊断应用。
结论
综上所述,BrainMoE 为脑部疾病计算机辅助诊断中的脑电图-功能磁共振成像(EEG-fMRI)融合提供了一个实用且可解释的框架。其主要优势在于通过多专家架构和软路由机制实现自适应的多模态特征融合,动态平衡模态特异性信息与共享信息。此外,通过无缝整合模态状态掩码和缺失模态标记,同一训练模型在模态不完整的情况下进行推理时,无需额外配置即可实现稳健的性能。关键的是,该可解释框架提供了跨五种不同脑部疾病的透明化、群体水平的脑区归因通路,将传统的“黑箱”架构转变为具有生理学依据的计算机辅助诊断工具。这对于未来的计算神经影像工作流程具有重要意义,因为在这些流程中,异构数据源、模态信息的不完整性以及模型输出的可解释性均为核心考量因素。
作者声明不存在利益冲突。作者声明在撰写本文稿、编写代码、数据分析或图表制作过程中均未使用生成式人工智能工具。
本研究由国家自然科学基金(项目编号:62433002、62277001 和 U25A20446)、北京市属高校高水平创新团队建设与支持项目(项目编号:BPHR20220104)以及北京学者计划(项目编号:099)资助。
| 姓名 | 公司 | 目录编号 | 评论 |
|---|---|---|---|
| Bash | GNU 项目 | 5.1.16(1)-release | 软件 |
| C-PAC | FCP-INDI | 版本 1.8.7;容器标签 release-v1.8.7.dev1 | 软件 |
| CUDA Toolkit | NVIDIA 公司 | 版本 12.4 | 软件 |
| CUDA 兼容 GPU | NVIDIA 公司 | GeForce RTX 4060 Laptop GPU | 设备 |
| Desikan-Killiany 皮层图谱 | FreeSurfer,Athinoula A. Martinos 生物医学成像中心,麻省总医院 | aparc;68 个皮层区域 (RRID:SCR_001847) | 图谱/资源 |
| EEGLAB | 加州大学圣地亚哥分校 Swartz 计算神经科学中心 | 版本 2022.1 (RRID:SCR_007292) | 软件 |
| FreeSurfer | Athinoula A. Martinos 生物医学成像中心,麻省总医院 | 版本 7.4.1 (RRID:SCR_001847) | 软件 |
| FSL | 牛津大学 FMRIB | 随 C-PAC 1.8.7 一起打包;具体版本未指定 (RRID:SCR_002823) | 软件 |
| 健康脑网络 (HBN) 数据集 | 儿童心智研究所 | RRID:SCR_016989 | 数据集 |
| MATLAB | MathWorks | R2022a (RRID:SCR_001622) | 软件 |
| MNE-Python | MNE-Python 开发团队 | 版本 1.9 (RRID:SCR_005972) | 软件/库 |
| Python | Python 软件基金会 | 版本 3.12.4 (RRID:SCR_008394) | 软件 |
| PyTorch | PyTorch 基金会 | 版本 2.6.0+cu124 (RRID:SCR_018536) | 库 |
本文已发表
视频即将推出