（论文加源码）基于连续卷积神经网络（CNN）（SVM）（MLP）提取脑电微分熵特征的DEAP脑电情绪识别

简介：

---

       主要内容是采用DEAP数据集将脑电信号进行频域分段并提取其微分熵特征，为了充分利用空间特征，结合微分熵特征将其构建为一个三维脑电特征，输入到连续卷积神经网络，并最终取得了90.24%的准确率。

         提出了一种脑电特征的三维输入形式，并将其输入到连续卷积神经网络中进行情感识别。三维输入的优点是在集成多个频带的微分熵特征的同时保留电极之间的空间特征。

         主体内容是采用卷积神经网络（CNN）达到了最优解，与此同时，运用了支持向量机（SVM）和多层感知器（MLP）进行了对比

---

主要内容：

---

为了后续处理方便，将DEAP数据集40*40*8064转换为40*8064*32.

1. 根据以前研究，脑电信号的五个频段对情绪识别有特殊影响，如下图。

图一

在本实验中使用滤波器将脑电信号分为上图的后四个频段。划分好频段之后，数据集从40*8064*32(视频*脑电信号数值*通道)变为40*8064*4*32（视频*脑电信号数值*波段*通道）.由于8064维数过大，将其分割为长度为L的N段脑电信号。此时脑电信号数据变为40*N*L*4*32。(本实验中L设置为128)

2.提取脑电信号的微分熵特征（DE）。对于每一个脑电分段，提取其微分熵特征，因此每一段的特征可以用一个一维数值（32）代替。再对其进性归一化处理，此时脑电数据转换为了40*N*4*32。

提取微分熵的部分代码如下：

 完整提取微分熵的代码链接：

提取微分熵特征

（论文加代码）计算提取脑电（EEG）的微分熵特征（DE）。_微分熵-深度学习文档类资源-CSDN下载

3. 构建三维的特征输入

●第一步先将一维特征转换为二维特征：

         国际10-20系统是一种国际公认的描述和应用电极位置和大脑皮层底层区域的方法。国际10-20系统的平面图，其中填充黄色的EEG电极是DEAP数据集中使用的测试点。在EEG电极图中，每个电极物理上相邻多个电极，这些电极记录大脑特定区域的EEG信号。为了保留多个相邻通道之间的空间信息，根据电极分布图，将一维的DE特征变换为二维平面（如图）（9*9），其中横纵长度均分9，0代表未使用该通道。

部分代码如下：

●第二步再把二维特征变为三维特征

        到目前为止，我们已经为每个EEG片段获得了4个2D平面（大概意思就是四个频域段，每个频域段对应一个二维微分熵特征图）。下一步是将这些平面堆叠成三维的 EEG特征立方体，并将其用作CNN的输入。该过程如图2所示（步骤4）。3维设计的灵感来自计算机视觉领域。在彩色图像分类任务中，图像由三种原色（红色、绿色和蓝色）表示。采用RGB颜色通道组织图像。具体而言，0到255的值用于指示每个颜色通道中颜色的强度。我们借鉴彩色图像的表示方法，采用simila表示来构造DE特征的三维输入。从而数据变为40*N*4*9*9.构造过程如图：

 

部分代码如下：

 

整体流程图如下：

 4. 构建连续卷积神经网络

       CNN具有从图像中提取特征的强大能力。根据我们之前的类比，构建的三维脑电立方体可以被视为彩色图像，这使得我们可以充分利用CNN作为一个强大的工具从输入中提取代表性的特征。在本文中，如图3所示，我们使用了一个具有四个卷积层的连续卷积神经网络来从输入立方体中提取特征，添加了一个具有退出操作的全连接层用于特征融合，softmax层用于最终分类。“连续”意味着在两个相邻的卷积层之间没有池层。虽然在计算机视觉领域中，卷积层后面通常是池层，但是在这个模型中是不必要的。由于池层的主要功能是以一些信息丢失为代价降低数据维数，因此三维脑电立方体的大小远小于计算机视觉场的大小。因此，模型中池层被丢弃。此外，在每个卷积层中，采用零填充来防止立方体边缘的信息丢失。更具体地说，在前3个卷积层中，内核大小设置为4*4，步幅设置为1。在卷积运算后，加入RELU激活函数，使模型具有非线性特征变换能力。使用64个特征映射开始第一个卷积层，并在接下来的2个卷积层中将特征映射加倍；因此，在第二层和第三层中有128个和256个要素地图。为了融合不同的特征映射并降低计算成本，增加了一个包含64个特征映射的1×1卷积层。在这4个连续的卷积层之后，添加了一个完全连接的层，以将64个9*9特征映射映射到最终的特征向量（1024）。然后，下面的softmax层接收f以预测人类的情绪状态。模型图如下:

构建模型代码如下：

 5. 最终结果：

同时编写了支持向量机(SVM)和多层感知器（MLP）的代码

 

如图所示，达到了90.24%和89.45%的正确率。

---

结论：

---

       研究表明所提出的三维脑电特征结合了不同频带的信号特征，同时保持了通道间的空间信息。在连续卷积神经网络模型上的实验结果表明，这种表示方法是非常有效的。我们的结果也证实了多频带组合比使用单个频带可以获得更好的结果。与其他相关方法相比，该方法在DEAP数据集上的唤醒和价分类任务的平均准确率分别为90.24%和89.45%，取得了最好的效果。

完整代码和论文：

完整代码和论文

基于连续卷积神经网络（CNN）提取脑电微分熵特征的DEAP脑电情绪识别-深度学习文档类资源-CSDN下载