[发明专利]基于深度迁移学习的单通道耳脑电自动睡眠分期方法

权利要求书

1.一种基于深度迁移学习的单通道耳脑电自动睡眠分期方法，其特征在于包括以下步骤：

步骤1、信号采集及预处理，采集受试者耳脑电信号，以及对睡眠数据库单通道头皮脑电信号及采集的耳脑电信号进行预处理，具体包括以下子步骤：

(a)、目标域数据的采集及预处理，采用耳脑电电极采集受试者整夜耳脑电信号，设置采样频率为125Hz，将采集后的耳脑电信号依次选取3750个采样点，作为一个睡眠时期，即一个30s的睡眠时期，对采集到的耳脑电信号进行睡眠专家评分，以便后续进行模型评估，并对耳脑电信号进行0.1-50Hz的带通滤波；

(b)、源域数据的选取及预处理，选取美国国家睡眠研究资源提供的睡眠心脏健康研究数据SHHS中接近正常多导睡眠图记录的100个受试者的C4-A1通道数据作为源域数据；将源域数据中非快速眼动期的深度睡眠期标签S3、S4合并为非快速眼动期N3，并排除体动和未知标签；考虑到非快速眼动期N1数据少，通过合成少数类过采样技术SMOTE将源域数据处理为类平衡数据，并对源域数据进行0.1-50Hz的带通滤波；

步骤2、特征提取，构建卷积神经网络CNN，提取睡眠数据库单通道头皮脑电信号及耳脑电信号的时不变特征，构建的卷积神经网络由源域数据输入，特征提取和分类输出层组成，具体包括以下子步骤：

(a)、源域数据输入，输入源域预处理后的C4-A1通道数据，采样频率为125Hz，即一个30s的睡眠时期为3750个采样点；将数据分成10份以便进行10折交叉验证，对于第一次交叉验证，采用源域数据中90个受试者的睡眠脑电数据做训练，剩余10个受试者睡眠脑电数据用作测试；

(b)、特征提取，选用3个结构相同但网络参数不同的一维CNN，即1D CNN并行提取睡眠脑电EEG的时不变特征，分别为时域、时频域及频域特征，网络训练层共包含8层，由4个一维卷积层、2个最大池化层maxpool、2个Dropout层组成；其中，三个初始卷积层中卷积核的大小分别为Fs/2、2Fs、4Fs，均选用64个卷积核，Fs表示使用的实验数据的采样频率125Hz，步长分别为Fs/16、Fs/6、Fs/2；在卷积层后加入maxpool，池化核大小分别为8、6、4，步长分别为8、6、4；在maxpool后加入Dropout层，使某个神经元的激活值以概率p停止工作，概率p选取0.5；在后续卷积层中，卷积核大小和步长选择小的固定值，3个卷积核的大小分别为8、7、6，步长均为1，均选用128个卷积核，用尺寸小的多层卷积核代替大卷积核的单层卷积核；卷积层后maxpool的池化核大小分别为4、3、2，步长分别为4、3、2；

共有n段30s的单通道脑电信号{x₁,…,x_n}，应用三个1D CNN提取第i个脑电信号x_i第j个特征a_j，通过公式(1)-(4)进行描述，

式中，θ_s、θ_m和θ_l分别表示CNN小、中和大卷积核的参数，和分别表示使用CNN小卷积核、中卷积核和大卷积核将一个30s的睡眠脑电信号x_i转换成一个特征向量h_j^s、h_j^m和h_j^l，||表示将两层卷积核输出的特征向量连接起来；在网络训练过程中，采用交叉熵损失函数度量损失，损失函数通过公式(5)进行描述，

表示CNN参数的集合，l表示网络的层数，J表示交叉熵损失函数，θ(x_i)表示第i个睡眠脑电信号x_i通过CNN训练预测为某个睡眠阶段的概率，y_i表示指示变量，如果x_i的预测分期与真实分期相同则为1，否则为0；

(c)、分类输出层，CNN的最后一层通过全连接层输出，激活函数为softmax，对于预测的睡眠阶段i，通过将5个睡眠阶段映射为[0,1]的输出，Z_i为经过CNN的输出，为经过CNN输出预测为5个睡眠阶段的指数和；

(d)、采用基于mini-batch的Adam优化算法更新网络权重；

(e)、以上步骤2子步骤(a)、(b)、(c)和(d)为一次交叉验证，此(a)、(b)、(c)和(d)子步骤重复进行10次，完成10折交叉验证；

步骤3、模型适配，构建深度迁移学习模型，使得源域脑电信号的睡眠分期模型能应用于目标域的耳脑电信号上，进而完成耳脑电信号的自动睡眠分期，具体包括以下子步骤：

(a)、输入数据：输入源域预处理后的C4-A1通道数据和目标域单通道耳脑电信号数据；

(b)、加入适配层：在特征层即步骤2构建的卷积神经网络实现源域数据的参数和目标域数据的参数共享；在网络深层，即分类器前加入三层适配层，即三层全连接层；在源域数据和目标域数据的三层适配层之间加入多核最大均值差异MK-MMD的度量函数，来度量源域数据和目标域数据的距离，最大均值差异即通过多个高斯核函数K将源域数据和目标域数据映射到再生核希尔伯特空间，在再生核希尔伯特空间度量两个分布p和q的距离，多个核定义的核函数K通过公式(6)进行描述，

β为不同高斯核贡献的权重，贡献大的高斯核权重大，贡献小的高斯核权重小，k_u表示第u个高斯核，共m个高斯核，并将其加入网络的损失中继续训练，MK-MMD通过公式(7)进行描述，

式中，d_k(p,q)表示再生核希尔伯特空间H_K的距离，其中，φ(x^s)、φ(x^t)分别为源域数据x^s和目标域数据x^t在再生核希尔伯特空间的映射，E_p表示数学期望；因此，整个深度迁移学习模型的优化目标由分类损失函数和度量函数构成，通过公式(8)进行描述，

式中，l₁、l₂表示网络适配是从第l₁层到第l₂层；和分别表示源域数据和目标域数据样本经过网络第l层的输出，参数λ决定了混淆域数据的强度；

(c)、采用基于mini-batch的Adam优化算法更新网络权重；

步骤4、模型评估，根据源域脑电信号SHHS的标签和目标域耳脑电信号睡眠专家的评分，分别采用每类指标和总体指标对构建的深度迁移学习模型得到的睡眠5期分期结果进行评估，分别包括准确率PR、召回率RE，以及总体准确率ACC和宏平均F1值，即MF1，具体通过公式(9)-(13)进行描述，

式中，TP表示正实例，即将正类预测为正类数，FP表示假正例，即将负类预测为正类数，FN表示假反例，即将正类预测为负类数，N表示所有睡眠阶段的样本总数，I表示睡眠阶段总类数。