Classification of EEG signals using Transformer based deep learning and ensemble models

被引:9
|
作者
Zeynali, Mahsa [1 ]
Seyedarabi, Hadi [1 ]
Afrouzian, Reza [2 ]
机构
[1] Univ Tabriz, Fac Elect & Comp Engn, Tabriz, Iran
[2] Univ Tabriz, Fac Engn, Miyaneh, Iran
关键词
Electroencephalography (EEG); Brain -Computer Interface (BCI); Transformer; Ensemble learning; NEURAL-NETWORKS; COLOR;
D O I
10.1016/j.bspc.2023.105130
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
A Brain-Computer Interface (BCI) is a communication and control system designed to provide interaction be-tween a user and a computer device. This interaction is based on the brain's electrical signals that are generated when users do specific tasks. Different categories of visual stimuli evoke distinct activation patterns in the human brain. The generated patterns can be recorded with EEG signals for use in BCI applications. Recently, deep learning-based Transformer models have demonstrated significant potential for analyzing diverse data. In this paper, a new Transformer-based model has been presented that extracts temporal and spectral features from EEG signals for classification purposes. The proposed Spectral Transformer model converts the EEG signal to the frequency domain using PSD before applying Transformer models to extract frequency features. Deep ensemble learning models are used to enhance the generalization performance of the final model by combining the benefits of both deep learning models and ensemble learning. The proposed ensemble model combines Temporal and Spectral Transformers to simultaneously utilize the time and frequency features of the signal. The accuracy of 96.1 %, 94.20 %, and 93.60 % are achieved using an ensemble model, Temporal Transformer, and Spectral Transformer, respectively. These results demonstrate the effectiveness of the proposed model for accurately classifying EEG signals in BCI applications.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] Ensemble deep learning for automated visual classification using EEG signals
    Zheng, Xiao
    Chen, Wanzhong
    You, Yang
    Jiang, Yun
    Li, Mingyang
    Zhang, Tao
    [J]. PATTERN RECOGNITION, 2020, 102
  • [2] Ensemble deep learning for automated visual classification using EEG signals
    Zheng, Xiao
    Chen, Wanzhong
    You, Yang
    Jiang, Yun
    Li, Mingyang
    Zhang, Tao
    [J]. Pattern Recognition, 2020, 102
  • [3] Ensemble Learning for Alcoholism Classification Using EEG Signals
    Cohen, Seffi
    Katz, Or
    Presil, Dan
    Arbili, Ofir
    Rokach, Lior
    [J]. IEEE SENSORS JOURNAL, 2023, 23 (15) : 17714 - 17724
  • [4] Classification of Motor Imagery EEG Signals with Deep Learning Models
    Shen, Yurun
    Lu, Hongtao
    Jia, Jie
    [J]. INTELLIGENCE SCIENCE AND BIG DATA ENGINEERING, ISCIDE 2017, 2017, 10559 : 181 - 190
  • [5] Bangla Documents Classification using Transformer Based Deep Learning Models
    Rahman, Md Mahbubur
    Pramanik, Md Aktaruzzaman
    Sadik, Rifat
    Roy, Monikrishna
    Chakraborty, Partha
    [J]. 2020 2ND INTERNATIONAL CONFERENCE ON SUSTAINABLE TECHNOLOGIES FOR INDUSTRY 4.0 (STI), 2020,
  • [6] Classification of Alcoholic EEG Signals Using a Deep Learning Method
    Farsi, Leila
    Siuly, Siuly
    Kabir, Enamul
    Wang, Hua
    [J]. IEEE SENSORS JOURNAL, 2021, 21 (03) : 3552 - 3560
  • [7] Multi-class Classification of Motor Imagery EEG Signals Using Deep Learning Models
    Khemakhem R.
    Belgacem S.
    Echtioui A.
    Ghorbel M.
    Ben Hamida A.
    Kammoun I.
    [J]. SN Computer Science, 5 (5)
  • [8] EEG signals classification based on wavelet packet and ensemble Extreme Learning Machine
    Han, Min
    Sun, Zhuoran
    Wang, Jun
    [J]. 2015 SECOND INTERNATIONAL CONFERENCE ON MATHEMATICS AND COMPUTERS IN SCIENCES AND IN INDUSTRY (MCSI), 2015, : 80 - 85
  • [9] Citrus pests classification using an ensemble of deep learning models
    Khanramaki, Morteza
    Asli-Ardeh, Ezzatollah Askari
    Kozegar, Ehsan
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 186
  • [10] Classification of electrocardiogram signal using an ensemble of deep learning models
    Pandey, Saroj Kumar
    Janghel, Rekh Ram
    [J]. DATA TECHNOLOGIES AND APPLICATIONS, 2021, 55 (03) : 446 - 460