节点文献
基于残差模块和自注意力机制GAN的脑电信号增广方法
EEG signal augmentation method based on generative adversarial network with ResBlock and self-attention machenism
【摘要】 针对脑电信号(EEG)数据量过少的问题,提出一种基于残差模块(ResBlock)和自注意力(Self-Attention)机制的生成对抗网络(GAN),记为RBSAGAN。该模型首先对ResBlock进行改进,设计了Up ResBlock和Down ResBlock网络用于提取信号中不同尺度感受野的特征并对数据维度进行扩大和缩小;然后根据Self-Attention机制设计1D Self-Attention网络挖掘EEG中各离散时刻之间的时间相关性;最后通过生成器和判别器的对抗训练生成逼真的信号。该模型在公开的BCI Competition IV dataset 2a数据集进行了大量实验,结果表明,RBSAGAN具有生成接近于真实脑电信号样本的能力,并且将分类器1D卷积网络(CNN)的平均识别率提升至96.04%,可以为EEG数据增广任务提供参考。
【Abstract】 Concerning the problem that the amount of ElectroEncephaloGram(EEG)data is too small,a Generative Adversarial Network(GAN)based on ResBlock and Self-Attention mechanism is proposed,namely RBSAGAN. The model first improved the ResBlock by designing the Up ResBlock and Down ResBlock networks to extract the features of different receptive fields in the signal and to expand and reduce the data dimension;then,the 1D Self-Attention network was designed according to the Self-Attention mechanism to mine the temporal correlation between discrete moments in the EEG;finally,realistic signals were generated by adversarial training of generator and discriminator. The model was extensively experimented on the publicly available BCI Competition IV dataset 2a. Experimental results show that the proposed RBSAGAN has the ability to generate samples close to real EEG signals and improve the average recognition rate of the classifier 1D convolutional network(CNN) to 96. 04%. The proposed method can provide a reference for EEG data augmentation tasks.
【Key words】 ElectroEncephaloGram(EEG) signal; Generative Adversarial Network(GAN); Convolutional Neural Network(GAN); residual network; self-attention mechanism;
- 【文献出处】 计算机应用 ,Journal of Computer Applications , 编辑部邮箱 ,2022年S1期
- 【分类号】R318;TN911.7
- 【下载频次】154