CN114903440A - 一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 - Google Patents
一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 Download PDFInfo
- Publication number
- CN114903440A CN114903440A CN202210524128.2A CN202210524128A CN114903440A CN 114903440 A CN114903440 A CN 114903440A CN 202210524128 A CN202210524128 A CN 202210524128A CN 114903440 A CN114903440 A CN 114903440A
- Authority
- CN
- China
- Prior art keywords
- electroencephalogram
- sequence
- fragment
- data
- model
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000012549 training Methods 0.000 title claims abstract description 74
- 238000000034 method Methods 0.000 title claims abstract description 39
- 239000012634 fragment Substances 0.000 claims abstract description 104
- 238000009499 grossing Methods 0.000 claims abstract description 24
- 230000008667 sleep stage Effects 0.000 claims abstract description 23
- 238000012545 processing Methods 0.000 claims abstract description 19
- 238000013527 convolutional neural network Methods 0.000 claims abstract 10
- 238000012544 monitoring process Methods 0.000 claims description 36
- 238000012795 verification Methods 0.000 claims description 28
- 210000004556 brain Anatomy 0.000 claims description 20
- 238000012360 testing method Methods 0.000 claims description 18
- 102100033620 Calponin-1 Human genes 0.000 claims description 11
- 102100033591 Calponin-2 Human genes 0.000 claims description 11
- 102100033592 Calponin-3 Human genes 0.000 claims description 11
- 101000945318 Homo sapiens Calponin-1 Proteins 0.000 claims description 11
- 101000945403 Homo sapiens Calponin-2 Proteins 0.000 claims description 11
- 101000945410 Homo sapiens Calponin-3 Proteins 0.000 claims description 11
- 238000007781 pre-processing Methods 0.000 claims description 8
- 230000002457 bidirectional effect Effects 0.000 claims description 6
- 230000006870 function Effects 0.000 claims description 6
- 230000007246 mechanism Effects 0.000 claims description 6
- 238000011176 pooling Methods 0.000 claims description 6
- 230000002618 waking effect Effects 0.000 claims description 6
- 230000004913 activation Effects 0.000 claims description 5
- 238000010276 construction Methods 0.000 claims description 5
- 230000003213 activating effect Effects 0.000 claims description 3
- 238000013528 artificial neural network Methods 0.000 claims description 3
- 125000004122 cyclic group Chemical group 0.000 claims description 3
- 238000013467 fragmentation Methods 0.000 claims description 3
- 238000006062 fragmentation reaction Methods 0.000 claims description 3
- 230000007774 longterm Effects 0.000 claims description 3
- 230000007704 transition Effects 0.000 claims description 3
- 101100384370 Drosophila melanogaster cnn gene Proteins 0.000 claims 7
- 230000000694 effects Effects 0.000 abstract description 11
- 230000003860 sleep quality Effects 0.000 abstract description 6
- 238000013135 deep learning Methods 0.000 abstract description 2
- 230000002708 enhancing effect Effects 0.000 abstract description 2
- 238000010586 diagram Methods 0.000 description 7
- 230000005611 electricity Effects 0.000 description 4
- 230000008569 process Effects 0.000 description 4
- 208000019116 sleep disease Diseases 0.000 description 4
- 206010062519 Poor quality sleep Diseases 0.000 description 3
- 239000011159 matrix material Substances 0.000 description 2
- 230000004461 rapid eye movement Effects 0.000 description 2
- 208000024172 Cardiovascular disease Diseases 0.000 description 1
- 208000008589 Obesity Diseases 0.000 description 1
- 238000002790 cross-validation Methods 0.000 description 1
- 206010012601 diabetes mellitus Diseases 0.000 description 1
- 201000010099 disease Diseases 0.000 description 1
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 1
- 230000005802 health problem Effects 0.000 description 1
- 210000000653 nervous system Anatomy 0.000 description 1
- 235000020824 obesity Nutrition 0.000 description 1
- 210000004789 organ system Anatomy 0.000 description 1
- 230000009323 psychological health Effects 0.000 description 1
- 238000013441 quality evaluation Methods 0.000 description 1
- 230000000284 resting effect Effects 0.000 description 1
- 230000033764 rhythmic process Effects 0.000 description 1
- 210000001519 tissue Anatomy 0.000 description 1
Images
Classifications
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/48—Other medical applications
- A61B5/4806—Sleep evaluation
- A61B5/4812—Detecting sleep stages or cycles
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/24—Detecting, measuring or recording bioelectric or biomagnetic signals of the body or parts thereof
- A61B5/316—Modalities, i.e. specific diagnostic methods
- A61B5/369—Electroencephalography [EEG]
- A61B5/372—Analysis of electroencephalograms
-
- A—HUMAN NECESSITIES
- A61—MEDICAL OR VETERINARY SCIENCE; HYGIENE
- A61B—DIAGNOSIS; SURGERY; IDENTIFICATION
- A61B5/00—Measuring for diagnostic purposes; Identification of persons
- A61B5/72—Signal processing specially adapted for physiological signals or for diagnostic purposes
- A61B5/7235—Details of waveform analysis
- A61B5/7264—Classification of physiological signals or data, e.g. using neural networks, statistical classifiers, expert systems or fuzzy systems
- A61B5/7267—Classification of physiological signals or data, e.g. using neural networks, statistical classifiers, expert systems or fuzzy systems involving training the classification device
Landscapes
- Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Surgery (AREA)
- Public Health (AREA)
- Pathology (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Heart & Thoracic Surgery (AREA)
- Medical Informatics (AREA)
- Molecular Biology (AREA)
- Veterinary Medicine (AREA)
- Animal Behavior & Ethology (AREA)
- General Health & Medical Sciences (AREA)
- Biophysics (AREA)
- Psychiatry (AREA)
- Psychology (AREA)
- Evolutionary Computation (AREA)
- Fuzzy Systems (AREA)
- Mathematical Physics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Physiology (AREA)
- Signal Processing (AREA)
- Measurement And Recording Of Electrical Phenomena And Electrical Characteristics Of The Living Body (AREA)
Abstract
本发明面向单信道脑电数据,提出一种基于“片段—序列”两阶段训练框架的睡眠分期方法及系统,属于深度学习算法智能应用领域。将单信道脑电数据进行平滑处理、片段化以及多通道生成等步骤,利用基于脑电片段的CNN网络模型对脑电数据进行训练,得到脑电特征表示方法,并通过基于脑电“片段—序列”的“CNN‑RNN”网络模型,增强睡眠序列之间学习,通过两阶段训练框架来提升模型的睡眠分期效果,进而提升利用单信道脑电数据进行睡眠质量评估的效果。本发明基于“片段—序列”两阶段训练框架的睡眠分期方法与基于脑电片段的CNN网络模型分期相比,效果有较大提升,通过在数据集Sleep_EDFx 39的单信道Fpz‑Cz脑电数据上验证,本发明将睡眠分期的准确率从80%提升到了87%。
Description
技术领域
本发明属于深度学习算法智能应用领域,尤其涉及一种基于“片段-序列”两阶段训练框架的睡眠分期方法及系统。
背景技术
睡眠在人的一生中占据大约1/3的时间,是人体各组织器官和神经系统休息的过程,对人的身心健康起着重要的作用。然而,全球有27%的人有睡眠障碍,我国的睡眠障碍患者的比例也越来越高。睡眠障碍易引发健康问题,增加肥胖、糖尿病、心血管疾病的患病风险,且容易诱发心理疾病。用以判断睡眠障碍和睡眠质量的基础,通常是睡眠分期。
睡眠分期,根据2007美国睡眠协会的分期标准(2017 AASM),可以分为清醒期(W)、浅睡1期(N1)、浅睡2期(N2)、深睡期(N3)、快速眼动期(REM)五个时期。睡眠存在一定的节律特点,正常周期一般为90~100分钟。入睡20分钟左右进入潜睡期,经过中度睡眠,大约在30~45分钟后进入深度睡眠阶段,然后逐渐退回到中度睡眠和浅睡,进入快速眼动阶段。
在医院中,通常使用睡眠检测仪(PSG)来监测患者的睡眠生物电信号,监测的信号包括脑电(EEG)、眼电(EOG)、心电(ECG)、肌电(EMG)等,并通过电信号分析患者的睡眠分期结果,基于睡眠分期的结果辅助诊断患者的睡眠质量。然而,在其他的场景,如居家、社区医院等,很难通过专业的PSG设备来监测睡眠质量,手环等基于心电的睡眠监测效果不如脑电数据。因此,通过单信道脑电数据进行睡眠分期监测的需求越来越强。
发明内容
发明目的:提出一种基于“片段-序列”两阶段训练框架的睡眠分期方法及系统,基于单信道脑电数据进行睡眠分期,并通过两阶段训练来提升模型的分期效果,并进一步提出一种用于实现上述方法的系统,以解决现有技术存在的上述问题。
技术方案:第一方面,提出一种基于“片段-序列”两阶段训练框架的睡眠分期方法,包括如下步骤:
步骤1、对单信道脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件;
步骤2、将单信道脑电数据EDF文件按照比例分为两个部分,第一个部分P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试;
步骤3、读取P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p';
步骤4、对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;
步骤5、将脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1;
步骤6、将脑电片段训练集M1随机打乱片段顺序,用于训练基于脑电片段的CNN 网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;
步骤7、将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存后缀为“.pth”的脑电片段模型文件F1;
步骤8、将步骤3中未打乱片段顺序的脑电片段数据集T1,按照时间步t进行划分,形成脑电序列数据集T2;
步骤9、将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2;
步骤10、读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;
步骤11、将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为后缀为“.pth”的脑电“片段-序列”模型文件F2;
步骤12、读取脑电“片段-序列”模型F2文件,使用步骤2中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
进一步的,在步骤3中,读取P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p';其平滑方法如下:
对于P中的每个文件,读取当中的单信道脑电监测数据列表{x1,x2,...,xn},对脑电监测数据列表当中的每一个数据都进行移动平滑,得到平滑后的数据{x′1,x′2,...,x′n},将平滑后的数据组合起来形成p',根滑动窗口大小的不同(记作K),可以获取不同的平滑效果,假设平滑窗口大小K=2s+1,则移动平滑后的x′t(t=1,2,...,n)的计算公式为:
进一步的,在步骤4中,对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;其具体操作步骤如下:
步骤4.1、将监测数据按照标签所表示的片段长度(假设长度为s)进行划分,并按照片段长度进行标准化处理,形成通道A1;
步骤4.2、将监测数据整体进行标准化,再按照片段长度s进行划分,形成通道B1;
步骤4.3、将通道A1与通道B1进行合并后,形成基于片段的脑电片段数据集T1。
进一步的,在步骤6中,将脑电片段训练集M1随机打乱片段顺序,用于训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;具体流程如下:
步骤6.1、对打乱顺序的脑电片段训练集M1中的片段通道A1和片段通道B1分别进行两支路的卷积,片段通道A1的两条支路的卷积分别为CNN1和CNN2,片段通道 B1的两条支路的卷积分别为CNN3和CNN4;其中,CNN1、CNN2、CNN3、CNN4的卷积的参数不同,但结构相同,具体结构如附图3b所示;
步骤6.2、将CNN1、CNN2、CNN3、CNN4四个分支的结果进行合并全连接;
步骤6.3、对步骤6.2结果进行dropout处理后,连接5个输出单元输出;
步骤6.4、对步骤6.3的输出结果进行softmax处理后按类别输出。
更进一步的,在步骤6.1中,CNN1、CNN2、CNN3、CNN4的卷积的参数不同,但结构相同,具体的流程如下:
步骤6.1.1、将通道进行卷积层卷积后,使用激活函数Swish进行激活。
步骤6.1.2、连接最大池化maxpooling层P1。
步骤6.1.3、连接dropout层D1。
步骤6.1.4、连接注意力机制层S1。
步骤6.1.5、连续连接3个卷积层和激活函数Swish的组合。
步骤6.1.6、连接最大池化maxpooling层P2。
步骤6.1.7、连接注意力机制层S2。
进一步的,在步骤10中,读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;具体流程如下:
步骤10.1、对打乱顺序的脑电序列训练集M2中的序列按照时间步输入到基于脑电片段的CNN网络模型提取特征;
步骤10.2、将提取的特征输入到双向LSTM单元,进行双向循环神经网络的学习,学习脑电序列的长短期依赖关系;
步骤10.3、将双向LSTM单元的结果合并后,连接条件随机场模块CRF,学习序列标签之间的状态转移关系。
第二方面,提出一种睡眠分期系统,该系统至少包括数据预处理模块、单信道脑电数据比例划分模块、脑电片段数据集构建模块、脑电片段数据集比例划分模块、脑电片段训练集打乱模块、脑电片段模型文件生成模块、脑电序列数据集生成模块、序列模型连接模块、脑电片段-序列模型文件存储模块以及睡眠分期序列分类模块。
数据预处理模块用于对单信道脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件;
单信道脑电数据比例划分模块用于将数据预处理模块中单信道脑电数据EDF文件按照比例分为两个部分,第一个部分P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试;
脑电片段数据集构建模块用于读取第一个部分P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p',并对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;
脑电片段数据集比例划分模块用于将所述脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1;
脑电片段训练集打乱模块用于将脑电片段训练集M1随机打乱片段顺序,训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;
脑电片段模型文件生成模块用于将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存后缀为“.pth”的脑电片段模型文件F1;
脑电序列数据集生成模块用于将所述脑电片段数据集构建模块中未打乱片段顺序的脑电片段数据集T1,按照时间步t进行划分,形成脑电序列数据集T2;将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2;
序列模型连接模块用于读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;
脑电片段-序列模型文件存储模块用于将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为后缀为“.pth”的脑电“片段-序列”模型文件F2;
睡眠分期序列分类模块用于读取脑电“片段-序列”模型F2文件,使用所述单信道脑电数据比例划分模块中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
第三方面,提出一种睡眠分期设备,包括至少一个处理器和存储器;存储器存储计算机执行指令;至少一个处理器执行存储器存储的计算机执行指令,使得至少一个处理器执行第一方面所述的睡眠分期方法。
第四方面,提出一种可读存储介质,该可读存储介质中存储有计算机执行指令,当处理器执行所述计算机执行指令时,实现第一方面所述的睡眠分期方法。
有益效果:本发明面向单信道脑电数据,提出一种基于“片段-序列”两阶段训练框架的睡眠分期方法及系统。将单信道脑电数据进行平滑处理、片段化、多通道生成等步骤,利用基于脑电片段的CNN网络模型对脑电数据进行训练,得到脑电特征表示方法,并通过基于脑电“片段-序列”的“CNN-RNN”网络模型,增强睡眠序列之间学习,通过两阶段训练框架来提升模型的睡眠分期效果,进而提升利用单信道脑电数据进行睡眠质量评估的效果。
附图说明
图1是本专利的基于“片段-序列”两阶段训练框架的睡眠分期方法流程图。
图2是对单信道脑电数据使用不同的移动平滑窗口进行数据平滑后的效果图。
图3a是基于脑电片段的网络模型结构图。
图3b是基于脑电片段的网络模型CNN分支结构图。
图4是基于脑电“片段-序列”的网络模型结构图。
图5是基于脑电“片段-序列”的网络模型对Sleep_EDFx39数据集交叉验证的混淆矩阵图。
图6是基于“脑电片段的网络模型”和“脑电‘片段-序列’的网络模型”对睡眠实例进行睡眠分期的结果对比图。
具体实施方式
在下文的描述中,给出了大量具体的细节以便提供对本发明更为彻底的理解。然而,对于本领域技术人员而言显而易见的是,本发明可以无需一个或多个这些细节而得以实施。在其他的例子中,为了避免与本发明发生混淆,对于本领域公知的一些技术特征未进行描述。
申请人认为,现有睡眠分期方法在其他的场景,如居家、社区医院等,很难通过专业的PSG设备来监测睡眠质量,手环等基于心电的睡眠监测效果不如脑电数据,因此,通过单信道脑电数据进行睡眠分期监测的需求越来越强。
为此,申请人提出一种基于“片段-序列”两阶段训练框架的睡眠分期方法及系统如图1-图6所示,包括第一方面:一种基于“片段-序列”两阶段训练框架的睡眠分期方法,包括以下步骤:
选择数据集Sleep_EDFx 39的单信道Fpz-Cz脑电数据,先将单信道Fpz-Cz脑电数据按照AASM 2007睡眠标准进行标签处理,形成数据文件,再利用本文的基于“片段 -序列”两阶段训练框架的睡眠分期方法进行睡眠分期。
步骤1、对单信道Fpz-Cz脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件。
步骤2、将单信道Fpz-Cz脑电数据EDF文件按照比例分为两个部分,第一个部分 P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试。
步骤3、读取P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p'。
步骤4、对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1。
步骤5、将脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1。
步骤6、将脑电片段训练集M1随机打乱片段顺序,用于训练基于脑电片段的CNN 网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数。
其中,脑电片段的CNN网络模型如附图3a和附图3b所示。
步骤7、将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存为脑电片段模型文件“cnn.pth”。
步骤8、将步骤3中未打乱片段顺序的脑电片段数据集T1,按照时间步64进行划分,形成脑电序列数据集T2。
步骤9、将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2。
步骤10、读取脑电片段模型文件“cnn.pth”获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数。
其中,基于脑电“片段-序列”的“CNN-RNN”网络模型结构如附图4所示。
步骤11、将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为模型文件“cnn_rnn.pth”。
步骤12、读取脑电“片段-序列”模型文件“cnn_rnn.pth”,使用步骤2中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
其中,附图5展示了基于脑电“片段-序列”的网络模型对Sleep_EDFx39数据集进行20折交叉验证的混淆矩阵图,附图6展示了基于“脑电片段的网络模型”和“脑电‘片段-序列’的网络模型”对Sleep_EDFx数据集当中某个未经训练的睡眠实例进行睡眠分期的结果对比图。
在进一步实施例中,在步骤3中,读取P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p';其平滑方法如下:
对于P中的每个文件,读取当中的单信道脑电监测数据列表{x1,x2,...,xn},对脑电监测数据列表当中的每一个数据都进行移动平滑,得到平滑后的数据{x′1,x′2,...,x′n},将平滑后的数据组合起来形成p'。
根据滑动窗口大小的不同(记作K),可以获取不同的平滑效果,如附图2所示。
假设平滑窗口大小K=2s+1,则移动平滑后的x′t(t=1,2,...,n)的计算公式为:
在进一步实施例中,在步骤4中,对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;其具体操作步骤如下:
步骤4.1、将监测数据按照标签所表示的片段长度(假设长度为s)进行划分,并按照片段长度进行标准化处理,形成通道A1。
步骤4.2、将监测数据整体进行标准化,再按照片段长度s进行划分,形成通道B1。
步骤4.3、将通道A1与通道B1进行合并后,形成基于片段的脑电片段数据集T1。
在进一步实施例中,在步骤6中,将脑电片段训练集M1随机打乱片段顺序,用于训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;其基于脑电片段的CNN网络模型结构如附图3a所示,具体流程如下:
步骤6.1、对打乱顺序的脑电片段训练集M1中的片段通道A1和片段通道B1分别进行两支路的卷积,片段通道A1的两条支路的卷积分别为CNN1和CNN2,片段通道 B1的两条支路的卷积分别为CNN3和CNN4。
其中,CNN1、CNN2、CNN3、CNN4的卷积的参数不同,但结构相同,具体结构如附图3b所示。
步骤6.2、将CNN1、CNN2、CNN3、CNN4四个分支的结果进行合并全连接。
步骤6.3、对上一步结果进行dropout处理后,连接5个输出单元输出。
步骤6.4、对上一步输出进行softmax处理后进行类别输出。
在更进一步实施例中,在步骤6.1中,CNN1、CNN2、CNN3、CNN4的卷积的参数不同,但结构相同,具体结构如附图3b所示。具体流程如下:
步骤6.2.1、将通道进行卷积层卷积后,使用激活函数Swish进行激活。
步骤6.2.2、连接最大池化maxpooling层P1。
步骤6.2.3、连接dropout层D1。
步骤6.2.4、连接注意力机制层S1。
步骤6.2.5、连续连接3个卷积层和激活函数Swish的组合。
步骤6.2.6、连接最大池化maxpooling层P2。
步骤6.2.7、连接注意力机制层S2。
在进一步实施例中,在步骤10中,读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;其基于脑电“片段-序列”的“CNN-RNN”网络模型结构如附图4所示,具体流程如下:
步骤10.1、对打乱顺序的脑电序列训练集M2中的序列按照时间步输入到基于脑电片段的CNN网络模型提取特征;
步骤10.2、将提取的特征输入到双向LSTM单元,进行双向循环神经网络的学习,学习脑电序列的长短期依赖关系;
步骤10.3、将双向LSTM单元的结果合并后,连接条件随机场模块CRF,学习序列标签之间的状态转移关系。
第二方面,提出一种睡眠分期系统,该系统至少包括数据预处理模块、单信道脑电数据比例划分模块、脑电片段数据集构建模块、脑电片段数据集比例划分模块、脑电片段训练集打乱模块、脑电片段模型文件生成模块、脑电序列数据集生成模块、序列模型连接模块、脑电片段-序列模型文件存储模块以及睡眠分期序列分类模块。
数据预处理模块用于对单信道脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件;
单信道脑电数据比例划分模块用于将数据预处理模块中单信道脑电数据EDF文件按照比例分为两个部分,第一个部分P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试;
脑电片段数据集构建模块用于读取第一个部分P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p',并对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;
脑电片段数据集比例划分模块用于将所述脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1;
脑电片段训练集打乱模块用于将脑电片段训练集M1随机打乱片段顺序,训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;
脑电片段模型文件生成模块用于将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存后缀为“.pth”的脑电片段模型文件F1;
脑电序列数据集生成模块用于将所述脑电片段数据集构建模块中未打乱片段顺序的脑电片段数据集T1,按照时间步t进行划分,形成脑电序列数据集T2;将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2;
序列模型连接模块用于读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段-序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;
脑电片段-序列模型文件存储模块用于将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为后缀为“.pth”的脑电“片段-序列”模型文件F2;
睡眠分期序列分类模块用于读取脑电“片段-序列”模型F2文件,使用所述单信道脑电数据比例划分模块中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
第三方面,提出一种睡眠分期设备,包括至少一个处理器和存储器;存储器存储计算机执行指令;至少一个处理器执行该存储器存储的计算机执行指令,使得至少一个处理器执行第一方面的睡眠分期方法。
第四方面,提出一种可读存储介质,该可读存储介质中存储有计算机执行指令,当处理器执行所述计算机执行指令时,实现第一方面的睡眠分期方法。
以上详细描述了本发明的实施方式,但是,本发明并不限于上述实施方式中的具体细节,在本发明的技术构思范围内,可以对本发明的技术方案进行多种等同变换,这些等同变换均属于本发明的保护范围。
Claims (9)
1.基于“片段—序列”两阶段训练框架的睡眠分期方法,其特征在于,包括如下步骤:
步骤1、对单信道脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件;
步骤2、将步骤1中单信道脑电数据EDF文件按照比例分为两个部分,第一个部分P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试;
步骤3、读取第一个部分P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p';
步骤4、对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;
步骤5、将所述脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1;
步骤6、将脑电片段训练集M1随机打乱片段顺序,用于训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;
步骤7、将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存后缀为“.pth”的脑电片段模型文件F1;
步骤8、将步骤3中未打乱片段顺序的脑电片段数据集T1,按照时间步t进行划分,形成脑电序列数据集T2;
步骤9、将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2;
步骤10、读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段—序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;
步骤11、将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为后缀为“.pth”的脑电“片段—序列”模型文件F2;
步骤12、读取脑电“片段—序列”模型F2文件,使用步骤2中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
3.根据权利要求1所述的睡眠分期方法,其特征在于,所述步骤4中,形成脑电片段数据集T1的具体操作步骤如下:
步骤4.1、将监测数据按照标签所表示的片段长度s进行划分,并按照片段长度s进行标准化处理,形成通道A1;
步骤4.2、将监测数据整体进行标准化,再按照片段长度s进行划分,形成通道B1;
步骤4.3、将通道A1与通道B1进行合并后,形成基于片段的脑电片段数据集T1。
4.根据权利要求3所述的睡眠分期方法,其特征在于,所述步骤6的具体流程如下:
步骤6.1、对打乱顺序的脑电片段训练集M1中的片段通道A1和片段通道B1分别进行两支路的卷积,片段通道A1的两条支路的卷积分别为CNN1和CNN2,片段通道B1的两条支路的卷积分别为CNN3和CNN4;
步骤6.2、将CNN1、CNN2、CNN3、CNN4四个分支的结果进行合并全连接;
步骤6.3、对步骤6.2的结果进行dropout处理后,连接5个输出单元输出;
步骤6.4、对步骤6.3的输出结果进行softmax处理后按类别输出。
5.根据权利要求4所述的睡眠分期方法,其特征在于,所述步骤6.1中,CNN1、CNN2、CNN3、CNN4的卷积的参数不同,但结构相同,具体流程如下:
步骤6.1.1、将通道进行卷积层卷积后,使用激活函数Swish进行激活;
步骤6.1.2、连接最大池化maxpooling层P1;
步骤6.1.3、连接dropout层D1;
步骤6.1.4、连接注意力机制层S1;
步骤6.1.5、连续连接3个卷积层和激活函数Swish的组合;
步骤6.1.6、连接最大池化maxpooling层P2;
步骤6.1.7、连接注意力机制层S2。
6.根据权利要求1所述的睡眠分期方法,其特征在于,所述步骤10的具体流程如下:
步骤10.1、对打乱顺序的脑电序列训练集M2中的序列按照时间步输入到基于脑电片段的CNN网络模型提取特征;
步骤10.2、将提取的特征输入到双向LSTM单元,进行双向循环神经网络的学习,学习脑电序列的长短期依赖关系;
步骤10.3、将双向LSTM单元的结果合并后,连接条件随机场模块CRF,学习序列标签之间的状态转移关系。
7.一种睡眠分期系统,其特征在于,包括:
数据预处理模块,用于对单信道脑电数据文件进行处理,去除首尾预定长度的清醒期数据,只保留睡眠阶段的监测数据,并统一存储为EDF文件;
单信道脑电数据比例划分模块,用于将数据预处理模块中单信道脑电数据EDF文件按照比例分为两个部分,第一个部分P用作脑电片段模型和脑电序列模型的训练和验证,第二个部分R用作脑电片段模型和脑电序列模型测试;
脑电片段数据集构建模块,用于读取第一个部分P中带有标签的单信道脑电监测数据,进行数据移动平滑处理,得到p',并对p'数据进行片段化和多通道扩展,形成脑电片段数据集T1;
脑电片段数据集比例划分模块,用于将所述脑电片段数据集T1按比例划分为脑电片段训练集M1、脑电片段验证集V1;
脑电片段训练集打乱模块,用于将脑电片段训练集M1随机打乱片段顺序,训练基于脑电片段的CNN网络模型,并通过脑电片段验证集V1选择最佳的脑电片段模型结构和模型参数;
脑电片段模型文件生成模块,用于将脑电片段最佳CNN网络模型结构和模型参数存储下来,保存后缀为“.pth”的脑电片段模型文件F1;
脑电序列数据集生成模块,用于将所述脑电片段数据集构建模块中未打乱片段顺序的脑电片段数据集T1,按照时间步t进行划分,形成脑电序列数据集T2;
将脑电序列数据集T2按照比例划分为脑电序列训练集M2、脑电序列验证集V2;
序列模型连接模块,用于读取脑电片段模型文件F1获取片段模型及其参数,连接序列模型,将脑电序列训练集M2随机打乱序列顺序,用来训练基于脑电“片段—序列”的“CNN-RNN”网络模型,并通过脑电序列验证集V2选择最佳的模型结构和参数;
脑电片段—序列模型文件存储模块,用于将脑电最佳“CNN-RNN”网络模型结构和模型参数存储下来,保存为后缀为“.pth”的脑电“片段—序列”模型文件F2;
睡眠分期序列分类模块,用于读取脑电“片段—序列”模型F2文件,使用所述单信道脑电数据比例划分模块中的脑电测试集R,对测试集R中的单信道脑电数据文件,进行睡眠分期序列分类。
8.一种睡眠分期设备,其特征在于,包括:至少一个处理器和存储器;
所述存储器存储计算机执行指令;
至少一个处理器执行所述存储器存储的计算机执行指令,使得至少一个处理器执行如权利要求1至6任一项所述的睡眠分期方法。
9.一种可读存储介质,其特征在于,所述可读存储介质中存储有计算机执行指令,当处理器执行所述计算机执行指令时,实现如权利要求1至6任一项所述的睡眠分期方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210524128.2A CN114903440A (zh) | 2022-05-13 | 2022-05-13 | 一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210524128.2A CN114903440A (zh) | 2022-05-13 | 2022-05-13 | 一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN114903440A true CN114903440A (zh) | 2022-08-16 |
Family
ID=82766998
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210524128.2A Pending CN114903440A (zh) | 2022-05-13 | 2022-05-13 | 一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114903440A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116369941A (zh) * | 2023-04-20 | 2023-07-04 | 南方医科大学南方医院 | 基于脑电eeg生理信息的睡眠质量判断方法 |
CN116636815A (zh) * | 2023-06-06 | 2023-08-25 | 中国人民解放军海军特色医学中心 | 基于脑电信号的水下作业人员睡眠质量评估方法及系统 |
-
2022
- 2022-05-13 CN CN202210524128.2A patent/CN114903440A/zh active Pending
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116369941A (zh) * | 2023-04-20 | 2023-07-04 | 南方医科大学南方医院 | 基于脑电eeg生理信息的睡眠质量判断方法 |
CN116636815A (zh) * | 2023-06-06 | 2023-08-25 | 中国人民解放军海军特色医学中心 | 基于脑电信号的水下作业人员睡眠质量评估方法及系统 |
CN116636815B (zh) * | 2023-06-06 | 2024-03-01 | 中国人民解放军海军特色医学中心 | 基于脑电信号的水下作业人员睡眠质量评估方法及系统 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN107495962B (zh) | 一种单导联脑电的睡眠自动分期方法 | |
Sun et al. | A hierarchical neural network for sleep stage classification based on comprehensive feature learning and multi-flow sequence learning | |
CN114903440A (zh) | 一种基于片段-序列两阶段训练框架的睡眠分期方法及系统 | |
Chen et al. | Emotion recognition based on fusion of long short-term memory networks and SVMs | |
Zhao et al. | SleepContextNet: A temporal context network for automatic sleep staging based single-channel EEG | |
Amiriparian et al. | Deep unsupervised representation learning for abnormal heart sound classification | |
CN112426147B (zh) | 睡眠呼吸事件检测模型处理方法、系统和存储介质 | |
CN113925459B (zh) | 一种基于脑电特征融合的睡眠分期方法 | |
Cesari et al. | Validation of a new data-driven automated algorithm for muscular activity detection in REM sleep behavior disorder | |
CN112263218A (zh) | 睡眠分期方法及装置 | |
Zhang et al. | Dual-CNN based multi-modal sleep scoring with temporal correlation driven fine-tuning | |
Travieso et al. | Building a Cepstrum-HMM kernel for Apnea identification | |
CN114841219A (zh) | 基于半监督学习的单通道脑电睡眠分期方法 | |
Längkvist et al. | A deep learning approach with an attention mechanism for automatic sleep stage classification | |
Lee et al. | SeriesSleepNet: an EEG time series model with partial data augmentation for automatic sleep stage scoring | |
CN118000664A (zh) | 基于多尺度时间残差收缩网络的睡眠分期模型构建方法 | |
Chen et al. | ivae: An improved deep learning structure for eeg signal characterization and reconstruction | |
Ma et al. | Parallel Dual-Branch Fusion Network for Epileptic Seizure Prediction | |
Hangaragi et al. | An evolutionary model for sleep quality analytics using fuzzy system | |
CN112037906A (zh) | 一种长时生理信号时间序列的样本数据的扩充方法及系统 | |
CN115770043A (zh) | 梦境情绪识别方法、装置、电子设备和存储介质 | |
CN115399735A (zh) | 基于时频双流增强的多头注意机制睡眠分期方法 | |
CN115349821A (zh) | 一种基于多模态生理信号融合的睡眠分期方法及系统 | |
Kumar et al. | Sleep Track: Automated Detection and Classification of Sleep Stages | |
Delimayanti et al. | The Effect of Feature Selection on Automatic Sleep Stage Classification Based on Multichannel EEG Signals |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |