东北大学学报(自然科学版) ›› 2023, Vol. 44 ›› Issue (8): 1098-1103.DOI: 10.12068/j.issn.1005-3026.2023.08.005

• 信息与控制 • 上一篇    下一篇

基于注意力机制RNN模型的癫痫患者脑电信号识别方法

周嵩, 高天寒   

  1. (东北大学 软件学院, 辽宁 沈阳110169)
  • 发布日期:2023-08-15
  • 通讯作者: 周嵩
  • 作者简介:周嵩(1989-),男,山东济南人,东北大学博士研究生; 高天寒(1978-),男,辽宁沈阳人,东北大学教授,博士生导师.
  • 基金资助:
    国家自然科学基金资助项目(52130403); 中央高校基本科研业务费专项资金资助项目(N2017003).

EEG Recognition Method for Epileptic Patients Based on RNN Model with Attention Mechanism

ZHOU Song, GAO Tian-han   

  1. School of Software, Northeastern University, Shenyang 110169, China.
  • Published:2023-08-15
  • Contact: GAO Tian-han
  • About author:-
  • Supported by:
    -

摘要: 针对癫痫患者脑电信号(electroencephalogram,EEG)数据识别提出了一种基于注意力机制的RNN(recurrent neural networks) 模型.传统EEG特征分析耗时巨大且过度依赖专家经验,极大限制了脑活动识别方法的应用推广.因此,提出一种新的EEG识别方法以解决上述问题.首先对癫痫患者EEG的基本特征进行分析,进而采用基于注意力机制RNN模型消除各种干扰信号,利用XGBoost分类器识别EEG数据的类别,达到自动细化识别原始EEG的目的,最后在公共EEG数据集上进行大量实验,验证所提方法对EEG识别的准确性.实验结果表明,与一些成熟的EEG识别方法相比,本文所提方法在识别精度上有了进一步提升.

关键词: 脑电信号;注意力机制;RNN模型;XGBoost分类器;癫痫患者

Abstract: A RNN (recurrent neural networks) model based on attention mechanism is proposed for EEG (electroencephalogram) data recognition in epilepsy patients. Traditional EEG feature analysis is time-consuming and excessively dependent on expert experience, which greatly limits the application and popularization of brain activity recognition methods. A new EEG recognition method to solve the above problems is proposed. Firstly, the basic characteristics of EEG from epilepsy patients are analyzed. Then, the RNN model based on attention mechanism is designed to eliminate various interference signals and the XGBoost classifier is used to identify the categories of EEG data, so as to achieve the purpose of automatic refinement and recognition of the original EEG. Finally, a large number of experiments are carried out on the public EEG data set to verify the accuracy of the proposed method. The experimental results show that compared with the mature EEG recognition methods, the proposed method has higher recognition accuracy.

Key words: EEG(electroencephalogram); attention mechanism; RNN (recurrent neural networks) model; XGBoost classifier; epileptic patient

中图分类号: