A Systematic Deep Learning Model Selection for P300-Based Brain-Computer Interfaces

被引:32
作者
Abibullaev, Berdakh [1 ]
Zollanvari, Amin [2 ]
机构
[1] Nazarbayev Univ, Sch Engn & Digital Sci, Dept Robot & Mechatron, Nur Sultan 010000, Kazakhstan
[2] Nazarbayev Univ, Sch Engn & Digital Sci, Dept Elect & Comp Engn, Nur Sultan 010000, Kazakhstan
来源
IEEE TRANSACTIONS ON SYSTEMS MAN CYBERNETICS-SYSTEMS | 2022年 / 52卷 / 05期
关键词
Brain modeling; Deep learning; Feature extraction; Electroencephalography; Visualization; Electrodes; Decoding; Brain-computer interfaces; convolutional neural network (CNN); deep learning; event-related potential (ERP); long short-term memory unit; model selection; P300; waves; MENTAL PROSTHESIS; CALIBRATION TIME; NEURAL-NETWORKS; POTENTIALS; EEG;
D O I
10.1109/TSMC.2021.3051136
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Predicting attention-modulated brain responses is a major area of investigation in brain-computer interface (BCI) research that aims to translate neural activities into useful control and communication commands. Such studies involve collecting electroencephalographic (EEG) data from subjects to train classifiers for decoding users' mental states. However, various sources of inter or intrasubject variabilities in brain signals render training classifiers in BCI systems challenging. From a machine learning perspective, this model training generally follows a common methodology: 1) apply some type of feature extraction, which can be time-consuming and may require domain knowledge and 2) train a classifier using extracted features. The advent of deep learning technologies has offered unprecedented opportunities to not only construct remarkably accurate classifiers but also to integrate the feature extraction stage into the classifier construction. Although integrating feature extraction, which is generally domain-dependent, into the classifier construction is a considerable advantage of deep learning models, the process of architecture selection for BCIs generally depends on domain knowledge. In this study, we examine the feasibility of conducting a systematic model selection combined with mainstream deep learning architectures to construct accurate classifiers for decoding P300 event-related potentials. In particular, we present the results of 232 convolutional neural networks (CNNs) (4 datasets x 58 structures), 36 long short-term memory cells (LSTMs) (4 datasets x 9 structures), and 320 hybrid CNN-LSTM models (4 datasets x 80 structures) of varying complexity. Our empirical results show that in the classification of P300 waveforms, the constructed predictive models can outperform the current state-of-the-art deep learning architectures, which are partially or entirely inspired by domain knowledge. The source codes and constructed models are available at https://github.com/berdakh/P3Net.
引用
收藏
页码:2744 / 2756
页数:13
相关论文
共 87 条
[31]  
Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1
[32]   On the Utility of Power Spectral Techniques With Feature Selection Techniques for Effective Mental Task Classification in Noninvasive BCI [J].
Gupta, Akshansh ;
Agrawal, Ramesh Kumar ;
Kirar, Jyoti Singh ;
Andreu-Perez, Javier ;
Ding, Wei-Ping ;
Lin, Chin-Teng ;
Prasad, Mukesh .
IEEE TRANSACTIONS ON SYSTEMS MAN CYBERNETICS-SYSTEMS, 2021, 51 (05) :3080-3092
[33]   A Wireless BCI and BMI System for Wearable Robots [J].
He, Wei ;
Zhao, Yue ;
Tang, Haoyue ;
Sun, Changyin ;
Fu, Wei .
IEEE TRANSACTIONS ON SYSTEMS MAN CYBERNETICS-SYSTEMS, 2016, 46 (07) :936-946
[34]   Cross-Participant EEG-Based Assessment of Cognitive Workload Using Multi-Path Convolutional Recurrent Neural Networks [J].
Hefron, Ryan ;
Borghetti, Brett ;
Kabban, Christine Schubert ;
Christensen, James ;
Estepp, Justin .
SENSORS, 2018, 18 (05)
[35]  
Hochreiter S, 1997, NEURAL COMPUT, V9, P1735, DOI [10.1162/neco.1997.9.8.1735, 10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]
[36]   An efficient P300-based brain-computer interface for disabled subjects [J].
Hoffmann, Ulrich ;
Vesin, Jean-Marc ;
Ebrahimi, Touradj ;
Diserens, Karin .
JOURNAL OF NEUROSCIENCE METHODS, 2008, 167 (01) :115-125
[37]  
Ioffe S., 2015, PMLR, P448, DOI DOI 10.48550/ARXIV.1502.03167
[38]   MOABB: trustworthy algorithm benchmarking for BCIs [J].
Jayaram, Vinay ;
Barachant, Alexandre .
JOURNAL OF NEURAL ENGINEERING, 2018, 15 (06)
[39]   The Study of Generic Model Set for Reducing Calibration Time in P300-Based Brain-Computer Interface [J].
Jin, Jing ;
Li, Shurui ;
Daly, Ian ;
Miao, Yangyang ;
Liu, Chang ;
Wang, Xingyu ;
Cichocki, Andrzej .
IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2020, 28 (01) :3-12
[40]   Two-Factor Authentication System Using P300 Response to a Sequence of Human Photographs [J].
Kaongoen, Netiwit ;
Yu, Moonwon ;
Jo, Sungho .
IEEE TRANSACTIONS ON SYSTEMS MAN CYBERNETICS-SYSTEMS, 2020, 50 (03) :1178-1185