[发明专利]一种基于视觉Transformer的自动睡眠分期方法

说明书

本发明公开一种基于视觉Transformer的自动睡眠分期方法。本发明将原始PSG信号通过滑窗处理得到PSG信号序列；将PSG信号序列通过数据增强得到增强信号样本；通过视觉Transformer帧级编码器、双向GRU序列级编码器、softmax层级联构建睡眠分期网络，将每组PSG信号样本输入睡眠分期网络,预测PSG信号样本的睡眠阶段，通过跨模态迁移学习初始化网络，结合PSG信号样本的真实睡眠阶段构建损失函数，使用ADAM优化器训练得到优化后睡眠分期网络；实时采集PSG信号，将PSG信号样本通过优化后睡眠分期网络，预测睡眠阶段。本发明根据PSG信号的噪声和伪迹设计数据增强，提高网络对PSG信号噪声和伪迹的鲁棒性；引入基于视觉Transformer的编码器，提高网络特征表示能力；通过迁移学习，减轻对大量PSG数据的依赖。

技术领域

本发明属于睡眠质量评估技术领域，具体涉及一种基于视觉Transformer 的自动睡眠分期方法。

背景技术

多导睡眠图(Polysomnography，PSG)是临床上用于睡眠状态监测的标准技术，全面记录被监测对象睡眠过程中的各项生理指标，包括脑电(EEG)、眼电(EOG)、肌电(EMG)等神经信号和口鼻气流、胸腹压力、血氧饱和浓度等呼吸监测数据，可以作为评估睡眠质量和诊断睡眠障碍症状的有效依据。但长期以来睡眠生理信号分析都依赖于睡眠专家人工检查多导睡眠图信号，这种方式存效率低、人力成本高等问题，且专家知识的主观差异性也会导致评估结果的误差。因此，需要开发鲁棒的高性能自动化睡眠分期工具来辅助医生进行睡眠分期，提高睡眠分期效率和准确率。

自动睡眠分期技术是扩大睡眠评估和诊断的基础，能服务数百万睡眠障碍人群，并使得在家庭环境中进行睡眠监测成为可能。尽管现有的自动化睡眠分期模型取得了不错的自动睡眠分期的性能，能够超越单一人类专家的分期准确率，然而目前的模型仍然存在一些亟待解决的问题：

视觉Transformer能提取有效的特征表示，但是在PSG信号上的性能还未被探索过；同时时间轴和频率轴上的位置信息对于经过傅立叶变换的PSG信号至关重要。但最近的睡眠分期方法只考虑时间轴上的位置关系，而视觉 Transformer能够同时捕获在时间轴和频率轴上的位置信息，弥补了现有模型的不足。

基于Transformer的深度学习方法需要大量训练数据才能超越CNN的表现。当前基于Transformer的自动睡眠分期模型在大规模PSG数据集进行预训练时表现良好，但在较小的数据集上睡眠分期的准确率显著下降。然而，获取大规模、标记准确的PSG数据集十分困难，并且在大型PSG数据集上从头开始训练模型需要耗费大量的计算资源。

现有自动睡眠分期模型对噪声和伪迹的鲁棒性不强。由于人为因素和采集环境的影响，PSG信号中存在难以避免的噪声和伪影。然而，针对PSG信号的数据增强模块设计的研究仍然十分有限。大多数工作直接采用在图像和音频任务中的增强手段，没有考虑PSG信号本身的特性。

因此研究一种基于视觉Transformer的鲁棒自动睡眠分期技术具有充分的必要性。

发明内容

针对睡眠分期任务中特征表示的性能有限、可靠PSG数据集稀缺以及模型对PSG信号噪声和伪影的鲁棒性不佳等问题，本发明引入了一种基于视觉 Transformer的编码器，通过迁移学习减轻对大量PSG数据的依赖，同时针对 PSG信号的噪声和伪迹设计一个数据增强模块，从而学习高性能、高鲁棒性的特征表示。

本模型包括三个关键思想：设计一个基于视觉Transformer的帧级编码器，采用滑动窗口来捕获短期上下文信息，并使用GRU来实现长期的序列级建模；通过跨模态的迁移学习，使用在域外数据集上的预训练模型，在睡眠PSG数据集上进行微调，以减轻对大规模PSG数据集的依赖；提出一个针对EEG和EOG 通道的动态数据增强模块，使模型能够学习更加鲁棒的特征表示。

本发明的方法为一种基于视觉Transformer的自动睡眠分期方法，具体步骤如下：