一种双分支融合的深度学习脑电降噪方法、设备及介质技术

本发明专利技术公开了一种双分支融合的深度学习脑电降噪方法、设备及介质，该方法的步骤包括：1、将采集到的单通道干净脑电信号以及伪迹信号线性混叠，从而构建训练数据集；2、建立双分支融合网络模型，利用两个分支分别建模干净脑电信号和伪迹信号，并在建模过程中引入交互机制实现两者特征增强，进一步设计融合模块自适应地融合两个分支的结果来进一步恢复干净脑电信号；3、构建损失函数并训练得到最优降噪模型；4、利用训练好的模型实现脑电降噪任务。本发明专利技术同时考虑恢复干净脑电和伪迹信号，从而能提高重构脑电信号的质量，对人机交互和医疗健康等领域具有重要意义。康等领域具有重要意义。康等领域具有重要意义。

【技术实现步骤摘要】
一种双分支融合的深度学习脑电降噪方法、设备及介质


[0001]本专利技术涉及脑电信号处理及人工智能领域，具体涉及一种基于深度学习的脑电降噪方法、设备及介质。

技术介绍

[0002]脑电信号（Electroencephalogram, EEG）是记录大脑活动的有力工具，能够更客观、可靠地捕捉人类的脑部状态。由于时间分辨率高、安全性高和获取成本低等优势，EEG在认知科学研究、脑机接口、睡眠监测、情感计算、疾病诊断等领域具有广泛的应用。然而，EEG十分微弱，容易受到眼电、肌电、心电、工频干扰、运动伪迹等多种伪迹污染。这些伪迹的存在严重阻碍后续的脑电分析，限制EEG的实际应用。因此，脑电降噪至关重要，在科研和应用中均受到广泛关注。
[0003]传统上，独立成分分析、典型相关分析等盲原分离（Blind Source Separation, BSS）算法是脑电降噪最常用的方法。然而，BSS算法需要对伪迹组件进行手动检查和剔除，或针对特定的伪迹设置合适的阈值，不利于实际应用。相比之下，基于深度学习的降噪方法可以自动从原始信号中提取EEG特征，而不需要先验知识，获得了研究人员的青睐。近年来，卷积神经网络、长短时记忆网络等深度神经网络被用于脑电降噪领域，并被证明比传统方法更有前景。
[0004]然而，存在的深度学习方法仅考虑估计干净的EEG，而忽略了对污染信号中存在伪迹的建模，这会导致重建EEG不可避免地存在伪迹残留或信息丢失，从而影响最终的降噪性能。

技术实现思路

[0005]本专利技术是为了解决上述现有技术存在的不足之处，提出一种双分支融合的深度学习脑电降噪方法、设备及介质，以期能利用两个分支分别重建EEG和伪迹，并融合两个分支的结果来进一步恢复干净EEG，从而能提升重建EEG信号的质量，并获得更好的降噪效果。
[0006]本专利技术为达到上述专利技术目的，采用如下技术方案：本专利技术一种双分支融合的深度学习脑电降噪方法的特点在于，包括以下步骤：步骤1、获取一批单通道的干净脑电信号和伪迹信号，并通过线性混叠的方式对干净脑电信号与伪迹信号进行处理，获得半仿真的含噪脑电信号，用于构建样本对并得到训练数据集，所述训练数据集中任意一个样本对记为{(x,a,y)|x,a,y∈R1×
T
}，其中，y表示含噪脑电信号，x和a表示对应的干净脑电信号和伪迹信号，T表示信号长度；步骤2、建立双分支融合网络模型，包括：两路并行的脑电重建分支和伪迹重建分支，以及特征增强模块和双分支融合模块，并将所述含噪脑电信号y输入到双分支融合网络模型，得到最终预测的干净脑电信号x
p
；步骤2.1、所述含噪脑电信号y分别输入所述脑电重建分支、伪迹重建分支中进行处理，并通过所述特征增强模块将两个分支的中间结果进行交互后，再由所述脑电重建分
支、伪迹重建分支分别输出重建脑电信号x
’
和重建伪迹信号a
’
；步骤2.2、将所述含噪脑电信号y与重建伪迹信号a
’
作差后，得到另一个重建脑电信号x
’’
，将两个重建脑电信号x
’
和x
’’
合并输入到所述双分支融合模块中进行处理，得到最终预测的干净脑电信号x
p
；步骤3、构建损失函数并训练得到最优降噪模型；步骤3.1、利用式(1)构建损失函数L：L(θ
E
,θ
A
,θ
I
,θ
F
)=MSE(x
’
,x;θ
E
,θ
A
,θ
I
)+MSE(a
’
,a;θ
E
,θ
A
,θ
I
)+MSE(x
p
,x;θ
F
)(1)式(1)中，θ
E
,θ
A
,θ
I
,θ
F
分别表示脑电重建分支、伪迹重建分支、特征增强模块和双分支融合模块的待优化参数，MSE(
•
)表示均方误差损失函数；步骤3.2、基于训练数据集，利用Adam优化器对所述双分支融合网络模型进行训练，并计算所述损失函数L并反向传播至网络模型中进行优化，当损失函数L收敛时停止训练，从而获得最优降噪模型；步骤4、以所述最优降噪模型对一批含噪脑电信号进行降噪处理，得到干净脑电信号。
[0007]本专利技术所述的双分支融合的深度学习脑电降噪方法的特点也在于，所述步骤2中的脑电重建分支和伪迹重建分支均由若干个一维卷积层和全连接层构成，其中，一维卷积层用于提取深度特征，全连接层用于重建干净脑电信号和伪迹信号。
[0008]所述步骤2.1包括：步骤2.1.1、所述含噪脑电信号y分别输入到脑电重建分支、伪迹重建分支中，并相应经过若干个一维卷积层中的处理后，对应输出脑电特征F
E
和伪迹特征F
A
；步骤2.1.2、所述脑电特征F
E
和伪迹特征F
A
合并输入到所述特征增强模块中进行处理，相应输出增强脑电特征F
’
E
和增强伪迹特征F
’
A
；步骤2.1.3、所述增强脑电特征F
’
E
和增强伪迹特征F
’
A
分别输入到脑电重建分支和伪迹重建分支的全连接层中进行处理，相应输出重建脑电信号x
’
和重建伪迹信号a
’
。
[0009]所述特征增强模块是由一维卷积层和Sigmoid激活函数构成，用于从对脑电重建分支、伪迹重建分支中提取增强特征。
[0010]所述步骤2.2中的双分支融合模块包括：一维卷积层和Sigmoid激活函数，并用于对两个重建脑电信号x
’
和x
’’
进行处理后，输出掩码向量M；从而利用式(2)计算出最终预测的干净脑电信号x
p
：(2)式(2)中，表示点乘。
[0011]本专利技术一种电子设备，包括存储器以及处理器的特点在于，所述存储器用于存储支持处理器执行任一所述深度学习脑电降噪方法的程序，所述处理器被配置为用于执行所述存储器中存储的程序。
[0012]本专利技术一种计算机可读存储介质，计算机可读存储介质上存储有计算机程序的特点在于，所述计算机程序被处理器运行时执行一所述深度学习脑电降噪方法的步骤。
[0013]与现有技术相比，本专利技术的有益效果在于：1、本专利技术设计两个分支能同时重建污染信号中存在的干净EEG和伪迹信号，并利
用融合模块自适应地融合两个分支的结果来进一步恢复干净EEG，从而提升了重建EEG信号的质量，并大幅提升了深度神经网络的降噪能力。在公开数据集中的实验结果显示，本专利技术提出的方法降噪性能高于多个现有的降噪网络。
[0014]本专利技术设计特征增强模块来实现两个分支之间的特征交互。由于两个分支本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种双分支融合的深度学习脑电降噪方法，其特征在于，包括以下步骤：步骤1、获取一批单通道的干净脑电信号和伪迹信号，并通过线性混叠的方式对干净脑电信号与伪迹信号进行处理，获得半仿真的含噪脑电信号，用于构建样本对并得到训练数据集，所述训练数据集中任意一个样本对记为{(x,a,y)|x,a,y∈R1×
T
}，其中，y表示含噪脑电信号，x和a表示对应的干净脑电信号和伪迹信号，T表示信号长度；步骤2、建立双分支融合网络模型，包括：两路并行的脑电重建分支和伪迹重建分支，以及特征增强模块和双分支融合模块，并将所述含噪脑电信号y输入到双分支融合网络模型，得到最终预测的干净脑电信号x
p
；步骤2.1、所述含噪脑电信号y分别输入所述脑电重建分支、伪迹重建分支中进行处理，并通过所述特征增强模块将两个分支的中间结果进行交互后，再由所述脑电重建分支、伪迹重建分支分别输出重建脑电信号x
’
和重建伪迹信号a
’
；步骤2.2、将所述含噪脑电信号y与重建伪迹信号a
’
作差后，得到另一个重建脑电信号x
’’
，将两个重建脑电信号x
’
和x
’’
合并输入到所述双分支融合模块中进行处理，得到最终预测的干净脑电信号x
p
；步骤3、构建损失函数并训练得到最优降噪模型；步骤3.1、利用式(1)构建损失函数L：L(θ
E
,θ
A
,θ
I
,θ
F
)=MSE(x
’
,x;θ
E
,θ
A
,θ
I
)+MSE(a
’
,a;θ
E
,θ
A
,θ
I
)+MSE(x
p
,x;θ
F
)(1)式(1)中，θ
E
,θ
A
,θ
I
,θ
F
分别表示脑电重建分支、伪迹重建分支、特征增强模块和双分支融合模块的待优化参数，MSE(
•
)表示均方误差损失函数；步骤3.2、基于训练数据集，利用Adam优化器对所述双分支融合网络模型进行训练，并计算所述损失函数...

【专利技术属性】
技术研发人员：陈勋，崔恒，刘爱萍，
申请(专利权)人：中国科学技术大学，
类型：发明
国别省市：