基于多域特征和GATv2网络的癫痫发作预测方法
Seizure prediction method based on multi-domain feature and GATv2 network
通讯作者:
收稿日期: 2025-07-18
| 基金资助: |
|
Received: 2025-07-18
| Fund supported: | 山东省自然科学基金资助项目(ZR2024MF124);济南大学学科交叉融合建设项目2023(XKJC-202308);济南市高校创新团队项目(2019GXRC015). |
作者简介 About authors
韩哲(2000—),男,硕士生,从事时间序列研究.orcid.org/0009-0009-2859-4828.E-mail:
为了降低癫痫患者的发作伤害风险并优化治疗策略,提出基于多域特征融合和图注意力网络的癫痫发作预测方法. 提取时域、频域和复杂网络域的特征,采用新颖的三元注意力机制对特征进行深度融合,构建节点特征表示. 建模电极间的空间拓扑关系生成邻接矩阵,节点特征矩阵与邻接矩阵联合输入GATv2模型进行深度分析,实现癫痫发作的精准预测. 基于CHB-MIT数据集的实验结果表明,所提方法在癫痫发作预测任务中的分类准确率达到97.58%.
关键词:
To reduce seizure-related injury risks for seizure patients and optimize treatment strategies, an seizure prediction method based on multi-domain feature fusion and a graph attention network was proposed. Multi-dimensional features were extracted from the time domain, the frequency domain, and the complex network domain. A novel triple attention mechanism was employed for deep feature fusion, constructing node feature representations. Additionally, by modeling the spatial topological relationships between electrodes, an adjacency matrix was generated. The node feature matrix, together with the generated adjacency matrix, was then fed into a GATv2 model for deep analysis, achieving accurate seizure prediction. Experimental results on the CHB-MIT dataset demonstrate that the proposed method achieves a classification accuracy of 97.58% in seizure prediction tasks.
Keywords:
本文引用格式
韩哲, 孟庆芳, 张强, 张相龙, 赵亚欧.
HAN Zhe, MENG Qingfang, ZHANG Qiang, ZHANG Xianglong, ZHAO Yaou.
深度学习方法在脑电信号处理任务中取得巨大成功. Xiang等[5]提出基于同步性的图时空注意力网络(synchronization-based graph spatio-temporal attention network,SGSTAN),通过相位锁定值构建脑电通道功能连接图,并利用图注意力网络与Transformer分别提取空间拓扑特征与时间依赖关系. SGSTAN在CHB-MIT数据集上实现了高精度、低误报的癫痫发作预测,显著改善了难分病例的分类性能. Li等[6]提出基于模型不确定性学习的癫痫预测方法RepNet-MMCD,通过结构重参数化的轻量CNN(reparameterized convolutional neural network,RepNet)提取多尺度EEG特征,并引入改进的蒙特卡洛丢弃策略(modified EEG-based MC dropout strategy,MMCD),利用连续EEG片段间的时间信息进行概率校准,在降低模型不确定性的同时显著提升预测可靠性. Gao等[7]提出基于多尺度扩张卷积的儿科癫痫发作预测网络,设计时序与空间双阶段多尺度卷积模块,分别提取脑电信号的局部与全局特征,并利用扩张卷积块与注意力加权融合机制增强模型感受野与特征表达能力. 张培玲等[8]提出基于多尺度自适应残差网络(multi-scale adaptive residual network,MSAR)的癫痫检测方法,使用经验模态分解(empirical mode decomposition,EMD)分解EEG,并结合MSAR进行特征提取和分类. 张喜珍等[9]提出基于2D-CNN和Cox-Stuart早停机制的癫痫预测模型,使用Cox-Stuart检验法判断验证集损失函数是否存在上升趋势,从而及时停止训练,降低计算复杂度和时间复杂度. 现有的癫痫预测方法侧重于通过基于时域[10]、频域[11]和时频域[12]方法提取特征,对从大脑连接模式中提取的复杂网络统计特征[13]的探索相对有限;现有模型难以兼顾低误报率和高灵敏度,且对脑电时空复杂性挖掘不足. 采用特征提取方法表示不同通道的脑电信号,往往聚焦于单个通道的局部特征,忽略了各个脑区间的协同作用. 为了解决这个问题,图神经网络被应用于脑电分析. Liu等[14]利用图卷积网络和双向长短期记忆网络提取时空特征,并结合广义学习系统(broad learning system,BLS)进行分类. Chen等[15]提出癫痫脑电图卷积网络,充分利用通道间相关性. 这些基于图卷积网络的方法在处理通道间关系时,对不同通道往往赋予固定的权重,难以根据脑电信号的具体特征动态地突出更具关键信息的通道. 图注意力网络通过计算通道之间的注意力权重,让模型能够自主地聚焦于对当前任务更重要的通道.
本研究提出融合多域特征与图注意力网络(GATv2)的深度学习框架. 1)从原始脑电信号中并行提取时域、频域与复杂网络域的特征. 复杂网络特征通过本研究采用的加权邻域图算法构建功能连接图并计算其统计特性获得,以刻画大脑活动的非线性动力学模式. 2)设计三元注意力机制,对异构的多域特征进行自适应加权与深度融合,构建增强的节点特征表示. 基于电极的物理空间位置建立邻接矩阵,以编码大脑通道间的结构拓扑关系. 3)将融合后的节点特征与邻接矩阵共同输入GATv2中,由动态注意力机制进一步挖掘通道间高阶的空间功能依赖关系,完成发作前期的精准分类预测.
1. 模型介绍
如图1所示为本研究提出的癫痫发作预测方法整体工作流程:从脑电信号中提取时域、频域和复杂网络域的多维特征,基于通道间的空间距离构建邻接矩阵;所有提取的特征通过三元注意力机制进行深层融合,再与邻接矩阵进行融合后输入GATv2中进行有效解码.
图 1
图 1 所提癫痫发作预测方法的整体工作流程
Fig.1 Overall workflow of proposed seizure prediction method
1.1. 特征提取
原始脑电信号直接输入模型会导致计算量过大,且无法有效表征脑电信号的本质特性,为此从原始脑电信号中提取精简且具有判别力的、代表性的特征用于癫痫预测任务.
1.1.1. 时域特征提取
在时间域上,为了降低信号的维数并保留重要信息,共提取16个统计特征来表示序列数据,实现数据的降维;这16个特征也将作为部分节点特征,并用这些特征构成的向量来表示脑电信号段,实现脑电信号的降维. 如表1所示,设脑电片段由序列
表 1 时域的统计特征
Tab.1
| 特征 | 描述 |
| 最大值 | |
| 最小值 | |
| 中值 | 奇数时为中间元素,偶数时为中间2个元素和取均值 |
| 众数 | 频率最高的值 |
| 方差 | |
| 均值 | |
| 标准差 | |
| 取值范围 | |
| 四分位距 | |
| 偏度 | |
| 峰度 | |
| 波动系数 | |
| 分形维数 | |
| 样本熵 | |
| 排列熵 | |
| 模糊熵 |
1.1.2. 频域特征提取
在频域分析中,对脑电信号进行快速傅里叶变换(FFT),可获取包含不同频率成分的正弦波的幅度和相位信息,进而提取各频率分量的功率谱特征. 研究表明,癫痫发作期的脑电信号的功率或能量相比发作前期显著增大,因此功率谱信息可作为区分不同脑电信号状态的有效特征,对于频率
1.1.3. 复杂网络域特征提取
复杂系统通过复杂网络形式表示,复杂系统内部单元、个体之间的相互作用可以用网络形式刻画. 对脑电信号做复杂网络分析,能捕捉不同时间点的脑网络的结构变化,揭示发作前的网络动态,有助于更准确地预测癫痫发作. 本研究在提取脑电信号时域和频域特征的基础上,提取复杂网络的统计特征,以此全面分析癫痫发作前的非线性动力学统计特性. 在把脑电信号构造为复杂网络的基础上,本研究提出基于复杂网络统计特性的脑电信号特征提取方法.
使用加权邻域图[16](weighted neighborhood field graph,WNFG)法把脑电信号构造为复杂网络. 脑电信号经过FFT得到它的频域表示:
式中:
设置
节点之间建立连接并计算所有
图 2
图 2 基于加权邻域图的脑电信号复杂网络构建流程图
Fig.2 Flowchart of weighted neighborhood field graph construction for EEG signal complex network
在构造的加权邻域图的基础上,提取复杂网络的度中心性、介数中心性、特征向量中心性节点和Katz中心性这4种统计特性,作为脑电信号的非线性动力学特征.1)度中心性是在网络分析中刻画节点中心性的最直接度量指标. 一个节点的节点度越大,意味着该节点的度中心性越高,该节点在网络中就越重要. 某个节点度中心性计算式为
式中:
式中:
式中:
式中:
对于4种复杂网络特征,每种特征提取到的特征维度为(C,129),4种特征进行拼接得到(C,516)的数据. 本研究采用多域特征提取策略,从时域提取到(C,16)的特征,从频域提取到(C,129)的特征和从复杂网络域提取到(C,516)的特征作为脑电信号的特征,频域和复杂网络域特征分别进行线性变换,得到2个(C,16)的数据,此时时域、频域和复杂网络域三者的特征均为(C,16),将它们沿一个维度堆叠,得到(3,C,16)的三维特征,该三维特征综合了多域信息,能够更准确地刻画脑电信号.
1.2. 构建邻接矩阵
本研究提出基于空间距离的嵌入方法,用于表征EEG电极间的空间相关性. 定义包含任意2个EEG电极间距离的集合
式中:M(
1.3. 三元注意力机制
提取丰富的特征有助于EEG分类,但不适当的融合方法可能会引入更多冗余信息,导致模型性能不佳. 本研究采用三元注意力机制,为提取的特征赋予不同权重,突出对分类具有判别性的关键特征[17],如图3所示. 三元注意力层(triple attention layer,TAL)的目标是在保持当前维度缩减的同时,以高效的方式建模通道注意力和空间注意力. 对于输入张量
图 3
式中:
式中:
1.4. 图注意力机制
为了深入探索多通道脑电信号的空间关系,不同通道的脑电信号通过电极间的空间距离转换为图结构,送入GATv2[18]的模型中. 对象的脑电信号会被表示成图数据,每个通道作为图中的1个节点. GATv2与GAT非常相似,但修复了静态注意力问题. GAT的计算式为
式中:
为了使每个节点在节点之间具有独立的权重,应用softmax函数:
每个节点特征向量
为了提高模型的拟合能力,采用多头注意机制. 为了保证GATv2的稳定性,对多个注意机制的结果进行平均,并且每个注意机制的学习特征是不同的. 学习到的节点特征向量计算式为
式中:
2. 实验设计
2.1. 数据集介绍及预处理
CHB-MIT数据集包含24个EEG记录,来自23名受试者,其中chb21为chb01受试者经过1.5 a后的追踪记录. 考虑到癫痫异常放电主要集中在0.5~50.0 Hz频段,原始脑电信号须进行预处理. 采用5阶巴特沃斯带通滤波器对该频段进行滤波处理,以去除噪声干扰,提高信噪比,进而优化分类性能. 本研究中所有使用的脑电数据均指经过0.5~50.0 Hz滤波处理后的信号. 由于不同患者在多次实验中使用的电极配置存在差异,为了确保数据的一致性和可比性,选择所有受试者共同拥有的18个导联通道进行分析,具体包括:FP1-F7、F7-T7、T7-P7、P7-O1、FP1-F3、F3-C3、C3-P3、P3-O1、FP2-F4、F4-C4、C4-P4、P4-O2、FP2-F8、F8-T8、T8-P8、P8-O2、FZ-CZ和CZ-PZ. 在癫痫发作预测任务中,准确界定发作前期和发作间期至关重要. 癫痫检测任务是为了区分发作期和发作间期,癫痫预测与癫痫检测有着本质的区别. 鉴于现有发作前期定义标准存在分歧,本研究采用Maiwald等[19]提出的定义方法:发作前30 min设定为发作前期(SPH),该时间窗口既能有效预警患者,又避免引起过度焦虑;发作间期(SIH)选取距离癫痫发作前后4 h以外的时间段,取30 min的数据,来达到平衡数据集的目的. 当SPH时长不足30 min时,采用前一连续样本的后半段数据进行补充,以确保样本时长尽可能接近30 min. 考虑到数据质量要求,排除chb08、chb12、chb15病例(发作间期数据不满足4 h间隔要求),chb16病例(发作前数据包含前次发作信息)以及chb13的chb13_40.edf文件下第二次发作记录(发作前期的时间只有357 s,不满足发作前期30 min的要求),最终选择19名患者的数据用于实验分析. 考虑到大脑活动的复杂性和动态性,EEG信号具有显著的非平稳特征,其统计特性、频率分布和振幅随时间、任务或生理状态发生变化. 为了处理这一问题,将连续的EEG信号分割为5 s时长的片段,假设每个片段内信号具有伪平稳性,保持相对稳定的统计和频率特征.
TUSZ数据集是美国天普大学(Temple University)医院发布的癫痫脑电公共数据集. 该数据集收录来自632名患者的多通道脑电记录,原始数据以欧洲数据格式存储,并附有文本形式的标注信息及临床报告. 相较于CHB-MIT数据集,TUSZ在数据采集时间跨度、来源多样性以及受试者数量等方面具有显著优势. 由于数据采集设备型号多样,该数据集中脑电信号的采样频率并不统一,分布在250~
电极位置差异以及不同患者间头部形状和电极配置的变化可能引起信号幅度差异,须对信号进行归一化处理,提高不同电极和患者间数据的可比性. 2个数据集里的样本均采用Z-score标准化方法,各通道信号转换为均值为0、标准差为1的标准正态分布,有效消除通道间的幅度差异.
2.2. 实验环境
使用Python语言在PyCharm环境中开发,采用PyTorch深度学习框架构建模型,实验在配备NVIDIA GeForce RTX 4060Ti GPU的计算平台上执行. 训练配置方面,采用Adam优化器,学习率设为0.001,在CHB-MIT数据集上进行100轮迭代训练,批处理大小设定为64. 为了确保结果可靠性,采用5折交叉验证策略,最终准确率取5次训练结果的均值. 模型优化采用交叉熵损失函数,通过最小化损失值来提升模型性能.
2.3. 评估指标
选择准确率Acc、敏感性Sen、特异性Spe、F1分数和AUC(ROC曲线下的面积)作为评估指标,用于评价模型在CHB-MIT数据集上的分类效果. 准确率反映模型在测试集上的整体分类正确性.
式中:TP、TN、FP和FN分别为真阳性、真阴性、假阳性和假阴性样本数,P为精确率,R为召回率.
3. 实验结果与分析
3.1. 实验结果
如表2所示为基于多域特征和GATv2的癫痫发作预测方法在CHB-MIT数据集上的评估结果. 可以看到,所提方法在19个病例上取得了优异的整体性能,平均准确率、敏感性、特异性、F1分数和AUC分别达到97.58%、98.11%、97.88%、97.61%和98.63%,且各指标间保持良好平衡,表明模型在不同患者间具有稳定的泛化能力. 方法在chb10和chb21病例上的预测性能表现最佳,chb10的准确率达99.63%,chb21的AUC为99.88%;在chb5、chb14和chb6等病例上的预测性能较差,如chb5的准确率为92.95%,推测与患者个体差异或发作模式不典型有关. 总体而言,所提方法在准确率、敏感性和特异性等关键评估指标上均表现优异,验证了其在癫痫发作预测中的有效性与鲁棒性.
表 2 所提癫痫发作预测方法在CHB-MIT数据集上的评估结果
Tab.2
| 病例 | Acc | Sen | Spe | F1 | AUC |
| chb1 | 98.00 | 99.56 | 96.75 | 98.92 | 98.48 |
| chb2 | 97.74 | 98.05 | 98.18 | 97.80 | 99.14 |
| chb3 | 98.03 | 97.25 | 98.96 | 97.23 | 98.39 |
| chb4 | 97.45 | 96.94 | 96.92 | 96.97 | 98.70 |
| chb5 | 92.95 | 96.39 | 95.74 | 93.34 | 96.82 |
| chb6 | 95.50 | 96.28 | 95.72 | 95.46 | 96.80 |
| chb7 | 98.67 | 98.86 | 97.24 | 99.32 | 98.75 |
| chb9 | 98.59 | 98.55 | 97.77 | 96.58 | 98.21 |
| chb10 | 99.63 | 99.62 | 99.93 | 99.43 | 99.86 |
| chb11 | 97.00 | 98.34 | 97.01 | 97.52 | 98.81 |
| chb13 | 98.59 | 97.28 | 98.83 | 97.83 | 98.84 |
| chb14 | 95.10 | 94.39 | 96.50 | 95.21 | 97.98 |
| chb17 | 98.00 | 99.03 | 97.56 | 98.86 | 99.85 |
| chb18 | 98.52 | 98.48 | 95.43 | 97.42 | 99.21 |
| chb19 | 97.50 | 98.07 | 96.98 | 97.38 | 98.86 |
| chb20 | 98.51 | 99.51 | 96.97 | 99.26 | 98.95 |
| chb21 | 98.06 | 98.67 | 97.45 | 98.42 | 99.88 |
| chb22 | 99.53 | 99.62 | 99.38 | 99.54 | 96.85 |
| chb23 | 98.57 | 99.11 | 97.26 | 98.07 | 99.58 |
| 平均值 | 97.58 | 98.11 | 97.88 | 97.61 | 98.63 |
如表3所示为所提方法在TUSZ数据集上的评估结果. 可以看到,该方法在6名患者上的平均准确率为89.58%,平均敏感性为88.21%,平均特异性为88.27%,平均F1分数为90.35%,平均AUC为89.45%. 各样本间性能差异较大,其中编号00000002表现最佳(准确率为96.88%,AUC为98.18%),编号00007252表现最差(准确率为78.53%,敏感性为76.98%),编号00002521的敏感性仅为77.21%,表明方法在不同样本上的泛化能力存在明显波动. 整体而言,该方法在该数据集上取得中等偏上的检测性能,但在部分样本上仍需进一步优化以提升鲁棒性.
表 3 所提癫痫发作预测方法在TUSZ数据集上的评估结果
Tab.3
| 编号 | Acc | Sen | Spe | F1 | AUC |
| 00000002 | 96.88 | 98.52 | 95.21 | 96.96 | 98.18 |
| 00001587 | 93.02 | 96.06 | 92.73 | 95.06 | 94.41 |
| 00002521 | 86.14 | 77.21 | 84.36 | 83.56 | 81.25 |
| 00003208 | 94.55 | 91.37 | 92.73 | 94.58 | 93.63 |
| 00000302 | 88.36 | 89.14 | 87.54 | 90.39 | 89.64 |
| 00007252 | 78.53 | 76.98 | 77.07 | 81.55 | 79.63 |
| 平均值 | 89.58 | 88.21 | 88.27 | 90.35 | 89.45 |
3.2. 跨受试者研究
采用留一法交叉验证评估所提方法在跨被试任务中的性能. 实验选用CHB-MIT数据集,每次留下一名患者的全部EEG数据作为测试集,其余所有患者的数据作为训练集,最终训练结果如表4所示. 该方法在19个病例上的平均准确率为75.13%,平均敏感性为74.10%,平均特异性为74.61%,平均F1分数为73.99%,平均AUC为73.53%. 各病例间表现差异显著:chb22病例表现最佳,准确率为96.53%、AUC为96.24%,chb9和chb19病例的准确率也都超过90%;chb6病例表现最差,准确率为48.17%、敏感性为42.68%,chb4、chb13、chb14、chb20、chb21等多个病例的准确率低于60%,且敏感性和特异性均较低. 上述结果表明,所提方法在该数据集上的预测性能不稳定,跨受试者泛化能力存在不足问题.
表 4 跨被试者实验结果
Tab.4
| 病例 | Acc | Sen | Spe | F1 | AUC |
| chb1 | 80.94 | 86.00 | 74.00 | 79.21 | 77.54 |
| chb2 | 80.00 | 78.45 | 80.00 | 82.58 | 82.64 |
| chb3 | 78.96 | 85.79 | 71.00 | 74.32 | 77.25 |
| chb4 | 56.23 | 49.56 | 57.00 | 52.75 | 49 |
| chb5 | 84.21 | 81.54 | 82.99 | 81.21 | 82.54 |
| chb6 | 48.17 | 42.68 | 44.10 | 48.91 | 44.73 |
| chb7 | 80.09 | 81.00 | 72.25 | 73.24 | 76.12 |
| chb9 | 90.97 | 96.78 | 80.56 | 89.52 | 90.32 |
| chb10 | 72.31 | 67.04 | 81.33 | 68.25 | 69.21 |
| chb11 | 88.56 | 88.21 | 83.99 | 90.23 | 90.89 |
| chb13 | 57.21 | 53.12 | 62.34 | 55.23 | 49.56 |
| chb14 | 54.39 | 55.82 | 63.75 | 58.22 | 54.88 |
| chb17 | 89.56 | 89.79 | 87.96 | 87.44 | 89.51 |
| chb18 | 86.14 | 82.98 | 91.56 | 85.25 | 86.52 |
| chb19 | 91.46 | 96.50 | 93.97 | 90.45 | 91.23 |
| chb20 | 52.17 | 43.56 | 51.23 | 48.78 | 51.21 |
| chb21 | 53.98 | 53.46 | 52.14 | 57.32 | 53.63 |
| chb22 | 96.53 | 97.21 | 97.45 | 98.51 | 96.24 |
| chb23 | 85.64 | 78.56 | 89.99 | 84.56 | 84.12 |
| 平均值 | 75.13 | 74.10 | 74.61 | 73.99 | 73.53 |
3.3. 分类方法对比
为了验证多域特征提取和图神经网络的方法在 CHB-MIT癫痫预测分类任务中的优势,采用本研究提出的预处理方法,并选取其他相关研究中的模型开展实验,进行对比分析. 1)结合共空间模式(common spatial pattern,CSP)和CNN的癫痫发作预测方法[20]通过时域和频域特征提取以及浅层CNN分类,在EEG数据上取得高灵敏度和低误报率的预测结果. 2)RDANet[21]深度学习模型用于预测癫痫发作,结合了残差网络和双自注意力机制,有效整合局部和全局特征来提高预测性能. 3)TA-STS-ConvNet[22]癫痫发作预测方法通过金字塔卷积网络从脑电图信号中提取多尺度时频特征,并结合三元注意力机制和动态图卷积网络来提高预测准确性. 4)采用滤波器组公共空间模式(filter bank common spatial pattern,FBCSP)算法思想的神经网络架构[23]使用逐层相关性传播(layer-wise relevance propagation,LRP)技术来解释网络模型的决策过程,提高了模型的可解释性. 5)基于多维Transformer和循环神经网络融合的癫痫发作预测模型[24](STFT+TLG)能够有效地从EEG信号中提取时频特征,并准确预测癫痫发作,实现端到端的信号处理. 6)基于多维EEG空间矩阵和残差网络的癫痫预测方法[25](F+ResNET)通过提取熵、分形维数、波动指数等非线性特征并结合脑电电极空间分布信息,构建多维特征矩阵输入ResNet,实现高精度的癫痫发作预测. 7种模型在CHB-MIT数据集18名受试者的分类准确率、敏感性和特异性如表5所示. 可以看出,所提方法的3项评价指标均显著高于传统方法CSP+CNN的对应指标. 在先进模型TA-STS-ConvNet、FBCSP+CNN、STFT+TLG和F+ResNET的分类敏感性中,STFT+TLG的敏感性高于所提方法,这归功于该方法结合STFT提取的时频域特征和Transformer,长短期记忆网络和门控循环单元捕捉的依赖关系展现出较好的性能,所提方法在特异性上更优,适合临床减少误报. 所提方法通过在时域、频域提取特征,使用加权邻域图构造复杂网络提取中心性特征,更加准确地表达了脑电信号,通过三元注意力机制的深度优化及图注意力网络空间信息的提取,与STFT+RDANet、CSP+CNN、FBCSP+CNN、TA-STS-ConvNet等方法相比,在准确率、敏感性和特异性上均表现更为优异.
表 5 不同预测方法的评估结果对比
Tab.5
对比实验主要集中在深度学习方法之间,这在一定程度上限制了对模型优势的全面评估. 传统的统计分析方法在医学信号处理领域依然具有较强的解释性和应用价值. 在提取完时域、频域、复杂网络域特征后,对比所提方法与传统的机器学习分类器的性能,评估结果如表6所示. 可以看出,深度学习方法优于传统的机器学习方法,也验证了本研究提取的时域、频域、复杂网络域特征的有效性.
表 6 不同学习方法的评估结果比较
Tab.6
| 方法 | Acc | Sen | Spe |
| 决策树 | 87.44 | 88.53 | 92.36 |
| KNN | 91.44 | 95.34 | 91.56 |
| SVM | 92.63 | 91.59 | 93.68 |
| XGboost | 95.41 | 96.25 | 96.54 |
| 本研究 | 97.58 | 98.11 | 97.88 |
3.4. 信号特征对模型的影响
为了全面评估复杂网络域特征的有效性,分别把时域、频域、复杂网络域特征单独及组合输入模型进行对比分析,评估结果如表7所示. 实验结果表明,采用复杂网络域特征的模型准确率相比单独使用时域特征提升4.59个百分点,相比单独使用频域特征提升6个百分点;时域、频域和复杂网络域特征的组合相比时域与频域特征组合提升5.14个百分点. 其他性能指标同样呈现显著改善,验证了复杂网络域特征在表征复杂脑电信号方面的有效性.
表 7 多维特征对模型性能的影响
Tab.7
| 特征 | Acc | Sen | Spe |
| 时域特征 | 87.97 | 88.62 | 88.26 |
| 频域特征 | 86.56 | 89.25 | 88.23 |
| 复杂网络域 | 92.56 | 93.15 | 92.65 |
| 时域+频域 | 92.44 | 92.54 | 93.65 |
| 时域+频域+复杂网络域 | 97.58 | 98.11 | 97.88 |
在复杂网络域提取到的特征当中,对于Katz中心性,0<
表 8 衰减参数对模型性能的影响
Tab.8
| Acc | Sen | Spe | F1 | AUC | |
| 0.01 | 97.34 | 97.89 | 97.57 | 96.81 | 97.89 |
| 0.10 | 97.58 | 98.11 | 97.88 | 97.61 | 98.63 |
| 0.20 | 97.28 | 97.82 | 98.01 | 97.24 | 97.99 |
| 0.50 | 96.82 | 97.51 | 94.12 | 96.81 | 98.13 |
3.5. 消融实验
通过逐步移除或替换核心组件,定量评估各模块对模型整体性能的贡献度. 1)移除三元注意力网络:直接把提取的多域特征作为节点输入,验证注意力机制的必要性. 2)替换注意力机制:三元注意力替换为传统注意力机制,对比验证三元注意力的优势. 3)替换图注意力网络:使用GAT替换GATv2,验证动态注意力机制相比静态注意力机制的性能提升. 实验结果如表9所示. 可以看出,三元注意力机制和图注意力网络是模型性能的核心组件. 三元注意力机制在多域特征聚合提取中发挥关键作用,移除后准确率下降7.17个百分点;当替换为普通注意力机制时,性能下降3.25个百分点,表明三元注意力网络相比传统注意力机制的优势在于能够同时关注3个维度的特征,更全面地捕捉序列数据中的复杂模式和依赖关系. 当GATv2网络被GAT网络替换时,模型准确率下降4.32个百分点,说明GATv2通过把GAT的静态注意力机制改进为动态注意力机制,显著提升了网络的表达能力和模型性能. 综合分析可见,完整模型通过各模块的高效协作,实现特征的全面建模和关键信息的充分表达,达到最优性能. 消融实验结果验证了模块设计的合理性及其协同优化的必要性,为模型整体性能提升提供了有力支撑.
表 9 模块消融实验结果
Tab.9
| 模型 | Acc | Sen | Spe |
| 移除三元注意力网络 | 90.41 | 91.33 | 89.65 |
| 使用自注意力模块 | 94.33 | 93.66 | 94.98 |
| 替换图注意力网络 | 93.26 | 92.98 | 93.54 |
| 本研究 | 97.58 | 98.11 | 97.88 |
4. 结 语
本研究提出融合多域特征与GATv2的癫痫发作预测方法. 该方法提取脑电信号的时域、频域及复杂网络域特征,借助三元注意力机制构建特征映射间的多维依赖关系,实现多域信息的高效融合;利用GATv2捕获脑电通道间的空间拓扑关系,完成癫痫脑电信号的分类识别. 实验结果表明,所提模型在准确率、敏感性、特异性等评估指标上均表现优异,且在单条样本推理时具有较快的响应速度,能够满足实际预测系统对实时性与准确性的双重需求,为癫痫患者的日常监测与预警提供了可行的技术方案. 尽管本研究取得了较好的预测效果,但仍存在以下局限:实验数据来自特定采集环境下的公开数据集,样本规模有限且患者个体差异较大,模型的跨个体泛化能力有待进一步验证;当前方法仅依赖脑电模态信号,未融合其他生理或行为数据. 后续研究将引入迁移学习与域自适应技术,提升模型在不同患者、不同采集设备间的泛化性能;还将融合多模态生理信号(如心电、肌电)并利用多视图学习,增强癫痫发作前特征的表征能力.
参考文献
An overview of EEG-based machine learning methods in seizure prediction and opportunities for neurologists in this field
[J].DOI:10.1016/j.neuroscience.2021.11.017 [本文引用: 1]
EEG seizure detection: concepts, techniques, challenges, and future trends
[J].DOI:10.1007/s11042-023-15052-2 [本文引用: 1]
Wireless EEG: a survey of systems and studies
[J].DOI:10.1016/j.neuroimage.2022.119774 [本文引用: 1]
The applied principles of EEG analysis methods in neuroscience and clinical neurology
[J].DOI:10.1186/s40779-023-00502-7 [本文引用: 1]
Synchronization-based graph spatio-temporal attention network for seizure prediction
[J].DOI:10.1038/s41598-025-88492-5 [本文引用: 1]
EEG-based seizure prediction via model uncertainty learning
[J].DOI:10.1109/TNSRE.2022.3217929 [本文引用: 1]
Pediatric seizure prediction in scalp EEG using a multi-scale neural network with dilated convolutions
[J].DOI:10.1109/jtehm.2022.3144037 [本文引用: 1]
基于多尺度自适应残差网络的癫痫检测方法
[J].DOI:10.3969/j.issn.1005-202X.2025.03.015 [本文引用: 1]
Epilepsy detection method based on multi-scale adaptive residual network
[J].DOI:10.3969/j.issn.1005-202X.2025.03.015 [本文引用: 1]
基于2D-CNN和Cox-Stuart早停机制的癫痫预测模型
[J].
Epilepsy prediction model based on 2D-CNN and Cox-Stuart early stopping mechanism
[J].
Evaluation of time domain features on detection of epileptic seizure from EEG signals
[J].DOI:10.1007/s12553-019-00363-y [本文引用: 1]
Is high-frequency activity at seizure onset inhibitory? A stereoelectroencephalographic study of motor cortex seizures
[J].DOI:10.1002/ana.26883 [本文引用: 1]
Deep learning based automatic seizure prediction with EEG time-frequency representation
[J].DOI:10.1016/j.bspc.2024.106447 [本文引用: 1]
A review of graph and complex network theory in water distribution networks: mathematical foundation, application and prospects
[J].DOI:10.1016/j.watres.2024.121238 [本文引用: 1]
Scalp EEG-based automatic detection of epileptiform events via graph convolutional network and bi-directional LSTM co-embedded broad learning system
[J].DOI:10.1109/LSP.2023.3263433 [本文引用: 1]
SSGCNet: a sparse spectra graph convolutional network for epileptic EEG signal classification
[J].DOI:10.1109/TNNLS.2023.3252569 [本文引用: 1]
Rock image classification based on EfficientNet and triplet attention mechanism
[J].DOI:10.3390/app13053180 [本文引用: 1]
SAT-GATv2: a dynamic attention-based graph neural network for solving Boolean satisfiability problem
[J].DOI:10.3390/electronics14030423 [本文引用: 1]
Comparison of three nonlinear seizure prediction methods by means of the seizure prediction characteristic
[J].DOI:10.1016/j.physd.2004.02.013 [本文引用: 1]
Epilepsy seizure prediction on EEG using common spatial pattern and convolutional neural network
[J].DOI:10.1109/JBHI.2019.2933046 [本文引用: 2]
An effective dual self-attention residual network for seizure prediction
[J].DOI:10.1109/TNSRE.2021.3103210 [本文引用: 2]
CLEP: contrastive learning for epileptic seizure prediction using a spatio-temporal-spectral network
[J].DOI:10.1109/TNSRE.2023.3322275 [本文引用: 2]
An interpretable deep learning classifier for epileptic seizure prediction using EEG data
[J].DOI:10.1109/ACCESS.2022.3176367 [本文引用: 2]
Epileptic seizure prediction via multidimensional Transformer and recurrent neural network fusion
[J].DOI:10.1186/s12967-024-05678-7 [本文引用: 2]
/
| 〈 |
|
〉 |

