计算机与现代化 ›› 2023, Vol. 0 ›› Issue (12): 19-23.doi: 10.3969/j.issn.1006-2475.2023.12.004

• 人工智能 • 上一篇    下一篇

基于高效通道注意力模块的运动想象脑电识别

  

  1. (1.江苏科技大学计算机学院,江苏 镇江 212003; 2.江苏科技大学自动化学院,江苏 镇江 212003;
    3.北京邮电大学集成电路学院,北京 100876)
  • 出版日期:2023-12-24 发布日期:2024-01-24
  • 作者简介:周成诚(1994—),女,陕西安康人,硕士研究生,研究方向:脑电信号处理和识别,E-mail: z1063050825@163.com; 通信作者:曾庆军(1969—),男,江苏镇江人,教授,硕士生导师,博士,研究方向:机器人传感与人机交互,导航与控制,E-mail: zheng28501@163.com; 杨康(1995—),男,广东茂名人,博士研究生,研究方向:芯片EDA与人工智能,E-mail: yangkang@bupt.edu.cn; 胡家铭(1996—),男,江苏徐州人,硕士研究生,研究方向:手部康复机器人和肌电信号识别,E-mail: 1540415386@qq.com; 韩春伟(1998—),男,江西九江人,硕士研究生,研究方向:康复机器人和信号处理,E-mail: 1622334047@qq.com。
  • 基金资助:
    国家自然科学基金资助项目(11574120); 江苏省产业前瞻与共性关键技术项目(BE201803)

EEG Recognition of Motor Imagination Based on Efficiency Channel Attention Module

  1. (1. College of Computer Science, Jiangsu University of Science and Technology, Zhenjiang 212003, China;
    2. College of Automation, Jiangsu University of Science and Technology, Zhenjiang 212003, China;
    3. College of Integrated Circuits, Beijing University of Posts and Telecommunications, Beijing 100876, China)
  • Online:2023-12-24 Published:2024-01-24

摘要: 摘要:基于运动想象的脑机接口技术有助于手部运动障碍的患者康复,因而广泛被用于康复医疗领域。针对目前运动想象脑电信号信噪比低,导致运动想象左右手脑电信号(Motor Imagination-Electroencephalogram, MI-EEG)分类效果不佳的问题,本文鉴于注意力模块能够关注与运动想象分类任务相关的重要特征和忽视不重要特征的特性,提出一种基于高效通道注意力(Efficient Channel Attention, ECA)模块的卷积神经网络对左右手MI-EEG进行特征提取和分类。为便于卷积神经网络(Convolutional Neural Network, CNN)对脑电信号进行识别,本文使用小波变换将脑电时序信号转换为二维时频图;然后调整基于ECA模块的CNN结构和参数;最后,对本文方法在脑电信号数据集上进行实验。实验结果表明,与一些基于深度学习的运动想象分类识别方法相比,基于ECA模块的CNN能够有效提升MI-EEG的识别准确率,说明本文方法在运动想象脑电识别方面具有有效性。

关键词: 关键词:运动想象, 脑电信号识别, 小波变换, 高效通道注意力模块, 卷积神经网络, 脑机接口

Abstract: Abstract: The brain-computer interface technology based on motor imagination is helpful to the rehabilitation of patients with hand movement disorders, so it is widely used in the field of rehabilitation medicine. Aiming at the problem of poor classification of motor imagination-electroencephalogram (MI-EEG) due to its low signal-to-noise ratio in current motor imagination-electroencephalogram, in view of the ability of the attention module to focus on important features related to motor imagination classification tasks and ignore unimportant features, we propose a convolutional neural network based on the efficient channel attention (ECA) module for feature extraction and classification of left and right-handed MI-EEG. In order to facilitate the recognition of EEG signals by convolutional neural network (CNN), this paper uses wavelet transform to convert the timing signals of C3 and C4 channels into two-dimensional time-frequency graphs, then designs a CNN structure and parameters based on ECA. Finally, the proposed method is tested on EEG data set. The experimental results show that compared with CNN and the CNN method based on fusion convolution attention, the CNN method based on ECA can effectively improve the recognition accuracy of MI-EEG, indicating that the proposed method is effective in motor imagination EEG recognition.

Key words: Key words: motor imagination, EEG recognition, wavelet transform, efficiency channel attention module, convolutional neural network, brain-computer interface

中图分类号: