[发明专利]一种基于聚类的多任务情感脑电特征提取与识别方法

说明书

本发明提供一种基于聚类的多任务情感脑电特征提取与识别方法。本发明步骤如下：1、多个被试者分别在诱发情感状态场景下进行脑电数据采集。2、对步骤1所得的脑电数据进行预处理。3、将处理后的脑电数据进行聚类得到子类标签矩阵。4、建立基于聚类多任务特征提取算法求解得到特征权重分布。5、根据特征权重分布对脑电数据进行特征提取以训练脑电情感识别模型。本发明通过聚类算法、多任务学习和特征提取的方式提高了脑电情感识别模型的预测精度。

技术领域

本发明属于脑电信号处理技术领域，具体涉及一种基于聚类的多任务情感脑电特征提取与识别方法。

背景技术

随着科学技术的发展，脑电情感识别及相关脑机制的研究已经逐渐成为神经科学的热门研究领域。利用计算机技术进行脑电情感识别是实现高级人机交互的关键性技术，实时高效的脑电情感识别对于实现人机交互、人机接口以及智能计算机等有重要意义。在具体应用方面，通过脑电情感识别，可以判断用户的情感状态，帮助计算机做出合适的响应，改善人机交互效果。脑电情感识别技术在人工智能领域得到了广泛关注。

由于脑电信号的不稳定性，采集的脑电信号具有低信噪比的特点，导致脑电数据特征存在冗余，影响脑电情感识别模型的准确度。现有的科学理论表明，真实脑电数据分布存在重要的内在结构信息，即通过探索脑电数据中的固有子类信息可以为脑电情感识别模型提供更多的先验知识，从而提高脑电情感识别模型的准确度。本发明提出一种基于聚类的多任务情感脑电特征提取与识别方法。我们在原始情感类别下探索多个情感子类，采用多任务学习对子类进行情感识别，每个子类的情感识别任务作为一个子任务，通过共享任务之间的信息来帮助提升各个子任务的性能，从而更有效地挖掘脑电情感识别任务中数据特征权重分布。通过特征权重分布，提取脑电情感数据中更具有辨别性的特征。采用特征提取后的脑电特征训练脑电情感识别模型，可以有效提升脑电情感识别模型的准确度。

发明内容

本发明的目的在于提供一种基于聚类的多任务情感脑电特征提取与识别方法。通过该方法可以实现脑电情感数据特征提取以及脑电情感识别。

本发明的具体步骤如下：

步骤1、对多个被试者进行脑电数据采集实验；对被试者进行外界刺激，使其产生情绪变化，同时采集被试者的脑电信号。

步骤2、对步骤1所得的所有脑电数据进行预处理。每一个被试者的脑电样本处理后的数据作为一个样本矩阵X；每个样本矩阵X均对应一个标签向量y；标签向量y对应被试者的情感类别。

步骤3、将处理后的脑电数据进行聚类，探索脑电数据原始情感类别中的情感子类，得到情感子类标签矩阵Y。

步骤4、建立基于聚类多任务特征提取模型以求解特征权重矩阵。

步骤4-1、建立基于聚类多任务特征提取算法模型目标函数如式(1)所示：

式中，是经过聚类后的子类情感标签矩阵，其中n表示样本个数，c表示子类情感类别个数；是脑电情感数据，其中n表示样本个数，d表示数据维度；表示脑电情感数据特征权重矩阵，其中d表示数据维度，c表示子类情感类别个数；表示脑电情感数据的拉普拉斯矩阵，具体计算方法为L＝D-S，其中是一个描述子类关系的相似性矩阵，具体定义为x_i表示脑电情感数据X的第i个样本，x_j表示脑电情感数据X的第j个样本；是一个关于S的对角矩阵，具体定义为λ是用来控制权重矩阵W稀疏性的超参数；β是用来控制子类关系对目标函数影响的超参数；表示矩阵的F-范数的平方，具体计算方法为表示矩阵的2,1-范数的平方，具体计算方法为tr(·)表示矩阵的迹，具体计算方法为矩阵对角线元素之和。

步骤4-2、对(1)式求解，得到脑电情感识别任务中脑电情感数据的特征权重矩阵W。

步骤5、根据特征权重矩阵W对脑电情感数据进行特征提取，根据提取后的脑电情感数据特征进行分类模型训练，得到脑电情感识别模型。