本发明涉及一种脑电信号分类模型训练方法、分类方法及相关装置,属于脑电信号分类领域。
背景技术:
1、脑机接口的目的是对包含用户意图的脑信号进行测量和定量的特性分析,并把测量结果实时转化为设备意图指令,同时为用户提供反馈。此类脑信号特性称为信号特征(简称为特征),特征提取是利用各种脑电信号处理技术,从中提取能够反映大脑思维意图的特征向量。在脑机接口研究中,脑电信号特征提取通常可以分为四个方面:时域分析、频域分析、时频分析和空域分析。脑电信号通常包含一些不连贯的伪迹,掩盖了波形中有价值的信息,所以消除伪迹将是预处理阶段的关键步骤。
2、因此脑电信号分类第一步为信号调理或预处理,通过消除已知干扰(伪迹)或无关信号,以及增强与应用相关的信号的空间、频谱或施加特征来增强信号;第二步为通过训练好的分类模型进行分类;但是目前的分类模型的分类精度一般,制约脑机接口技术的发展。
技术实现思路
1、本发明提供了一种脑电信号分类模型训练方法、分类方法及相关装置,解决了背景技术中披露的问题。
2、为了解决上述技术问题,本发明所采用的技术方案是:
3、一种脑电信号分类模型训练方法,包括:
4、对作为训练样本的脑电信号进行滤波;
5、采用预设的多种特征提取方法,提取滤波后的脑电信号的特征,并将多种特征提取方法提取的特征融合为一个整体特征;
6、采用鲸优化算法对整体特征进行优化,采用优化获得的特征对脑电信号分类模型进行训练;其中,在训练过程中,采用鲸优化算法对脑电信号分类模型的权重进行优化。
7、多种特征提取方法包括离散余弦变换方法、离散小波变换方法和blackman-窗短时傅立叶变换方法。
8、脑电信号分类模型为lstm模型。
9、一种脑电信号分类模型训练装置,包括:
10、第一滤波模块,对作为训练样本的脑电信号进行滤波;
11、第一特征提取模块,采用预设的多种特征提取方法,提取滤波后的脑电信号的特征,并将多种特征提取方法提取的特征融合为一个整体特征;
12、优化训练模块,采用鲸优化算法对整体特征进行优化,采用优化获得的特征对脑电信号分类模型进行训练;其中,在训练过程中,采用鲸优化算法对脑电信号分类模型的权重进行优化。
13、一种脑电信号分类方法,包括:
14、对待分类的脑电信号进行滤波;
15、采用预设的多种特征提取方法,提取滤波后的脑电信号的特征,并将多种特征提取方法提取的特征融合为一个整体特征;
16、将整体特征输入预先训练的脑电信号分类模型,获得脑电信号分类结果;其中,脑电信号分类模型采用上述脑电信号分类模型训练方法训练。
17、多种特征提取方法包括离散余弦变换方法、离散小波变换方法和blackman-窗短时傅立叶变换方法。
18、脑电信号分类模型为lstm模型。
19、一种脑电信号分类装置,其特征在于,包括:
20、第二滤波模块,对待分类的脑电信号进行滤波;
21、第二特征提取模块,采用预设的多种特征提取方法,提取滤波后的脑电信号的特征,并将多种特征提取方法提取的特征融合为一个整体特征;
22、分类模块,将整体特征输入预先训练的脑电信号分类模型,获得脑电信号分类结果;其中,脑电信号分类模型采用上述脑电信号分类模型训练方法训练。
23、一种计算机可读存储介质,所述计算机可读存储介质存储一个或多个程序,所述一个或多个程序包括指令,所述指令当由计算设备执行时,使得所述计算设备执行脑电信号分类模型训练或脑电信号分类方法。
24、一种计算机设备,包括一个或多个处理器、以及一个或多个存储器,一个或多个程序存储在所述一个或多个存储器中并被配置为由所述一个或多个处理器执行,所述一个或多个程序包括用于执行脑电信号分类模型训练或脑电信号分类方法的指令。
25、本发明所达到的有益效果:本发明采用多种特征提取方法提取作为训练样本的脑电信号的特征,采用鲸优化算法对融合后的特征进行优化,采用优化后的特征进行模型训练,并且在训练中采用鲸优化算法对脑电信号分类模型的权重进行优化,通过双重优化提升了分类模型的分类准确度。
1.一种脑电信号分类模型训练方法,其特征在于,包括:
2.根据权利要求1所述的脑电信号分类模型训练方法,其特征在于,多种特征提取方法包括离散余弦变换方法、离散小波变换方法和blackman-窗短时傅立叶变换方法。
3.根据权利要求1所述的脑电信号分类模型训练方法,其特征在于,脑电信号分类模型为lstm模型。
4.一种脑电信号分类模型训练装置,其特征在于,包括:
5.一种脑电信号分类方法,其特征在于,包括:
6.根据权利要求5所述的脑电信号分类方法,其特征在于,多种特征提取方法包括离散余弦变换方法、离散小波变换方法和blackman-窗短时傅立叶变换方法。
7.根据权利要求5所述的脑电信号分类模型训练方法,其特征在于,脑电信号分类模型为lstm模型。
8.一种脑电信号分类装置,其特征在于,包括:
9.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质存储一个或多个程序,所述一个或多个程序包括指令,所述指令当由计算设备执行时,使得所述计算设备执行权利要求1~3、或5~7所述的任一方法。
10.一种计算机设备,其特征在于,包括: