MMCNN: A Multi-branch Multi-scale Convolutional Neural Network for Motor Imagery Classification

被引:41
作者
Jia, Ziyu [1 ,2 ]
Lin, Youfang [1 ,2 ,3 ]
Wang, Jing [1 ,2 ,3 ]
Yang, Kaixin [1 ]
Liu, Tianhang [1 ]
Zhang, Xinwang [1 ]
机构
[1] Beijing Jiaotong Univ, Sch Comp & Informat Technol, Beijing, Peoples R China
[2] Beijing Key Lab Traff Data Anal & Min, Beijing, Peoples R China
[3] CAAC Key Lab Intelligent Passenger Serv Civil Avi, Beijing, Peoples R China
来源
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2020, PT III | 2021年 / 12459卷
基金
中国国家自然科学基金;
关键词
Motor imagery; Convolutional neural network; EEG signal; Brain-computer interface; EEG; FEATURES;
D O I
10.1007/978-3-030-67664-3_44
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Electroencephalography (EEG) based motor imagery (MI) is one of the promising Brain-computer interface (BCI) paradigms enable humans to communicate with the outside world based solely on brain intentions. Although convolutional neural networks have been gradually applied to MI classification task and gained high performance, the following problems still exist to make effective decoding of raw EEG signals challenging: 1) EEG signals are non-linear, non-stationary, and low signal-noise ratio. 2) Most existing end-to-end MI models utilize single scale convolution which limits the result of classification because the best convolution scale varies with different subject (called subject difference). In addition, even for the same subject, the best convolution scale also differs from time to time (called time difference). In this paper, we propose a novel end-to-end model, named Multi-branch Multi-scale Convolutional Neural Network (MMCNN), for motor imagery classification. The MMCNN model effectively decodes raw EEG signals without any pre-processing including filtering. Meanwhile, the multi-branch multi-scale convolution structure successfully addresses the problems of subject difference and time difference based on parallel processing. In addition, multi-scale convolution can realize the characterization of information in different frequency bands, thereby effectively solving the problem that the best convolution scale is difficult to determine. Experiments on two public BCI competition datasets demonstrate that the proposed MMCNN model outperforms the state-of-the-art models. The implementation code is made publicly available https://github.com/jingwang2020/ECML-PKDD MMCNN.
引用
收藏
页码:736 / 751
页数:16
相关论文
共 20 条
[1]   Filter bank common spatial pattern algorithm on BCI competition IV Datasets 2a and 2b [J].
Ang, Kai Keng ;
Chin, Zheng Yang ;
Wang, Chuanchu ;
Guan, Cuntai ;
Zhang, Haihong .
FRONTIERS IN NEUROSCIENCE, 2012, 6
[2]  
Ang KK, 2008, IEEE IJCNN, P2390, DOI 10.1109/IJCNN.2008.4634130
[3]  
[Anonymous], 2010, ICML
[4]  
Clevert D.-A., 2016, 4 INT C LEARN REPR I
[5]  
Cohen MX, 2014, ISS CLIN COGN NEUROP, P1
[6]   An end-to-end deep learning approach to MI-EEG signal classification for BCIs [J].
Dose, Hauke ;
Moller, Jakob S. ;
Iversen, Helle K. ;
Puthusserypady, Sadasivan .
EXPERT SYSTEMS WITH APPLICATIONS, 2018, 114 :532-542
[7]   Brain Computer Interfaces, a Review [J].
Fernando Nicolas-Alonso, Luis ;
Gomez-Gil, Jaime .
SENSORS, 2012, 12 (02) :1211-1279
[8]  
Fukunaga K, 1990, INTRO STAT PATTERN R, DOI DOI 10.5555/92131
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]