IEEE TBME | DoC-Informer:自适应脑电配置下的意识障碍自动识别

图片

论文首页标题和作者信息

该论文发表于 IEEE Transactions on Biomedical Engineering(中科院二区top,IF:4.4),题目为《DoC-Informer: Automated Discrimination of Disorders of Consciousness under Adaptive EEG Settings》。

武汉大学计算机学院杨丰源、皮龙霖为共同第一作者,陈丹教授为通讯作者。合作单位包括燕山大学电气工程学院、南方医科大学珠江医院康复医学科及首都医科大学宣武医院康复医学科。

论文链接:https://doi.org/10.1109/TBME.2026.3668768

图片

论文概要

脑电电极不齐,模型还能可靠地区分意识状态吗?DoC-Informer 将这一临床采集难题作为研究出发点:先从每个通道提取时间特征,再结合电极三维坐标学习通道间关系,最后整合较长时间范围内的动态信息;训练时随机遮蔽通道,使模型减少对固定电极组合的依赖。

研究使用两个医院队列、共156名意识障碍患者的静息态脑电数据开展受试者独立的五折交叉验证。模型在两个数据集上的准确率分别为77.45%和77.32%,并在单通道缺失、脑区通道缺失及低密度电极配置下保持较高的相对性能。论文的核心贡献,是将电极空间位置先验与深度学习结合,让意识状态识别适应不完整的脑电输入[1]。

研究背景

从行为评估走向脑电辅助识别

意识障碍(disorders of consciousness,DoC)涵盖多种严重意识受损状态。该文关注无反应觉醒综合征(UWS)与最小意识状态(MCS)的区分,这一判断与治疗方案和预后评估密切相关。临床通常使用修订版昏迷恢复量表(CRS-R),但患者状态波动及评估者判断差异会影响评估的一致性。脑电具有无创、便携、时间分辨率高等优势,为补充行为评估提供了客观信号。

现实采集条件与固定通道模型之间的矛盾

传统机器学习方法依赖人工设计的频谱、复杂度或连接性特征;深度学习虽能自动提取特征,却常要求固定数量与排列的电极。临床中,颅骨缺损可能限制电极放置,设备条件可能限制通道数,运动伪迹、电极移位和信号丢失也会造成数据缺口。通道一旦缺失,原有空间关系或输入维度就可能被破坏。

因此,本文要解决的并非仅是完整脑电下的分类精度,而是如何将“数量和组合可变的脑电通道”映射为UWS或MCS。研究借鉴临床人员结合电极位置解读局部记录的思路,用空间先验帮助模型整合剩余通道中的信息。

研究方法

整体框架

DoC-Informer 采用CNN与Transformer结合的结构,依次完成浅层时间特征编码、空间关系建模、长程时间关系建模以及二分类。这里的“通道独立”不是忽略通道之间的联系,而是避免网络被固定的通道数量与排列绑定:前端逐通道提取特征,后端根据实际电极位置学习联系。

图1 DoC-Informer整体框架(原文图1)

图1 DoC-Informer整体框架

浅层时间特征编码

模型使用两条并行的一维时间卷积路径:标准卷积捕捉局部细节,空洞卷积在不改变卷积核大小的情况下扩大感受野,观察更宽的时间范围。两条路径均对各脑电通道独立处理,随后将特征拼接,经池化、Dropout和1×1卷积压缩与融合。这样得到的特征保留通道身份,又为后续注意力建模降低了计算规模。

图2 浅层时间特征编码模块(原文图2)

图2 浅层时间特征编码模块

空间与时间分层建模

空间Transformer首先引入电极的三维坐标。坐标来自标准10–20电极系统,经归一化和位置编码后与脑电特征相加。与简单的通道序号相比,坐标能够表达电极在头皮上的实际空间位置;即使部分通道不在输入中,模型仍可依据剩余电极的位置学习通道间依赖。

在每个时间切片上,空间注意力整合不同通道的特征,再沿通道维度平均池化,生成不再依赖通道数量的表示。时间Transformer接着将各时间位置作为输入单元,学习跨时间的关联,最终由全连接分类器输出UWS或MCS。整个过程可以理解为先比较“同一时刻不同位置的信号”,再观察“这些信息随时间怎样变化”。

图3 融合电极位置编码的时空表示建模(原文图3)

图3 融合电极位置编码的时空表示建模

通道遮蔽训练

训练时,模型在时间切片上随机遮蔽通道特征,模拟不完整脑电,促使网络从多个通道分布的信息中学习,而不是依赖某几个固定电极;同时设置补偿机制,避免某个时间步全部通道均不可用。该策略与电极位置编码配合,提高缺失输入下的适应能力。主实验采用0.2的遮蔽概率,其选择依据见后文参数实验。

实验结果

数据与评估设置

数据集A来自南方医科大学珠江医院,包含105名患者,其中UWS 74名、MCS 31名,采用30通道、1000 Hz采样;数据集B来自首都医科大学宣武医院,包含51名患者,其中UWS 14名、MCS 37名,采用26通道、2048 Hz采样。预处理包括平均重参考、1–45 Hz带通滤波、降采样至250 Hz、Autoreject伪迹处理,以及划分不重叠的10秒脑电片段。

实验按受试者独立设置进行五折交叉验证,测试受试者的数据不参与训练。模型使用交叉熵损失和Adam优化器,初始学习率0.0004、批量大小16,并结合学习率衰减与早停。比较对象为GA-SVM及EEGNet、EEG-Conformer、EEG-Deformer、STPA-Net、STAnet、SST-EmotionNet。评估同时报告准确率、F1和AUC,以从不同角度观察分类表现。两个队列分别开展评估,不应将其理解为“在一家医院训练、直接迁移到另一家医院测试”。

完整通道下的识别性能

DoC-Informer在两个数据集上的三项指标均高于所比较的基线。数据集A的准确率、F1和AUC分别为0.7745、0.8389和0.7562;数据集B分别为0.7732、0.8825和0.7354。相较准确率最高的基线EEG-Deformer,准确率分别增加4.19和2.40个百分点。

表1 不同模型的意识状态识别性能表1 不同模型的意识状态识别性能(原文表II)

配对t检验显示,多数比较达到统计显著性,但并非全部:例如数据集A中,相比EEG-Deformer的准确率和F1,其p值分别为0.06和0.07。另一方面,模型的计算量为293.96 M FLOPs,高于原文效率表中所列的其他模型,因此不能把较好的识别性能直接等同于最低计算成本。

单个通道缺失时的稳定性

缺失实验先用完整通道训练,再在测试阶段移除指定通道。对于不能直接处理可变通道的深度学习基线,研究以零填充缺失位置。论文用相对性能保留率(RPR)衡量稳定性:将缺失配置下的性能除以完整配置下的性能,再对相应缺失情况取平均。因此,RPR接近100%代表性能接近完整通道水平,并不表示分类准确率接近100%。

图4 逐个移除通道后的识别性能(原文图4)

图4 逐个移除通道后的识别性能

单通道移除时,DoC-Informer在两个数据集、三项指标上的RPR为99.95%–99.99%,而深度学习基线约为95%–97%。这说明在所测试的单电极缺失情形中,模型的平均相对性能损失很小。

表2 单通道移除后的相对性能保留率

表2 单通道移除后的相对性能保留率(原文表IV)

整组脑区通道缺失时的表现

研究进一步按电极名称将通道划分为Fp、F、FC、C、P、O、T七组,逐组移除。DoC-Informer的RPR为97.15%–99.29%,总体高于基线约89%–91%的水平。这一结果反映的是跨移除情形的平均保留率,不能推断每一个脑区移除后的损失都小于3%。

图5 移除不同脑区通道后的识别性能(原文图5)

图5 移除不同脑区通道后的识别性能

表3 脑区通道移除后的相对性能保留率表3 脑区通道移除后的相对性能保留率(原文表V)

图5还显示,移除前额Fp组电极时下降最明显,提示该组信号对当前模型的判别较重要。这为简化电极配置提供了线索,但属于模型在当前数据上的观察,不能单凭电极移除实验确定某一脑区对意识的因果作用。

低密度电极配置的可行性

研究模拟三种配置:(a)8通道,(b)16通道,以及(c)仅使用额区电极。数据集A中,8通道和仅额区配置各项RPR均超过97%,16通道配置约为100%;数据集B中,三种配置各项RPR均超过95%。相比之下,所比较基线的保留率大体处于75%–88%。这些结果支持进一步探索更简便的床旁采集方案。

表4 数据集A在三种低密度配置下的性能保留率

表4 数据集A在三种低密度配置下的性能保留率(原文表VI)

表5 数据集B在三种低密度配置下的性能保留率

表5 数据集B在三种低密度配置下的性能保留率(原文表VII)

读表时应区分相对值与绝对值。例如数据集B在8通道配置下的准确率RPR为95.43%,乘以完整通道准确率77.32%,对应准确率约73.79%。个别配置的RPR略超过100%,仅表示该次减少通道后性能略有改善。这里采用既有记录的通道子集模拟稀疏配置,尚不能直接等同于在相应便携设备上完成了前瞻性临床验证。

消融实验揭示模块贡献

删除空间Transformer后,数据集A和B的准确率分别从0.7745、0.7732降至0.7194、0.7130,是准确率下降最明显的变体。删除时间Transformer也会降低性能,其中数据集A的AUC从0.7562降至0.6788。两项结果说明,通道间空间关系与较长时间范围内的动态信息都对分类有贡献。

表6 各模块与训练策略的消融结果表6 各模块与训练策略的消融结果(原文表VIII)

去掉空洞卷积、位置编码或遮蔽训练也会降低表中各项指标。不过,采用固定顺序位置编码的变体在数据集A上的AUC为0.7658,高于完整模型的0.7562。因此,更准确的结论是完整模型取得了较均衡的总体表现,而非每个指标都必然最优。

遮蔽概率的影响

遮蔽并非越强越好。实验将概率从0增加到0.8,结果呈现非单调变化:适中的遮蔽有助于泛化,0.2附近的准确率与AUC表现较好;过高概率会丢失过多空间信息,0.7时性能明显下降,0.8时训练未能收敛。数据集A的F1峰值并不恰好位于0.2,论文据综合表现将0.2作为默认参数。

图6 不同通道遮蔽概率对模型性能的影响(原文图6)

图6 不同通道遮蔽概率对模型性能的影响

结论

DoC-Informer通过逐通道时间卷积、电极三维位置编码、空间与时间Transformer以及通道遮蔽训练,将“利用哪些电极”与“如何整合脑电信息”联系起来。两个临床队列的实验表明,这种设计既能提高UWS与MCS分类表现,也能在所测试的缺失通道与低密度配置下保留较多判别能力。对脑电算法研究而言,值得学习的是把采集条件的变化纳入模型结构和训练过程,而不只追求完整数据上的指标。

进一步应用仍需扩大样本、开展跨中心迁移及真实稀疏设备验证,并提高模型可解释性。当前研究聚焦UWS与MCS二分类;论文提出未来可结合fMRI/PET等信息,并扩展至昏迷和脱离最小意识状态等类别。现有结果支持其作为脑电辅助评估方法继续研究,尚不足以证明可替代临床综合评估。

代码已公开:https://github.com/pilonglin/docinformer。研究数据未公开;有学术使用需求的研究者可联系通讯作者,在合理申请并签订适当数据使用协议的条件下申请获取。

参考文献:

Yang, F., Pi, L., Chen, D., Liang, Z., Wang, Y., Zhang, Y. (2026). DoC-Informer: Automated Discrimination of Disorders of Consciousness under Adaptive EEG Settings. IEEE Transactions on Biomedical Engineering. DOI: 10.1109/TBME.2026.3668768.

——END——

撰稿人:尹越

审核人:邱丽娜

图片

脑机接口与混合智能研究团队


登录用户可以查看和发表评论, 请前往  登录 或  注册
SCHOLAT.com 学者网
免责声明 | 关于我们 | 用户反馈
联系我们: