Residential College | false |
Status | 已發表Published |
ADFCNN: Attention-Based Dual-Scale Fusion Convolutional Neural Network for Motor Imagery Brain-Computer Interface | |
Tao, Wei1; Wang, Ze2; Wong, Chi Man1; Jia, Ziyu3; Li, Chang4; Chen, Xun5; Chen, C. L.Philip6; Wan, Feng1 | |
2023-12-13 | |
Source Publication | IEEE Transactions on Neural Systems and Rehabilitation Engineering |
ISSN | 1534-4320 |
Volume | 32Pages:154-165 |
Abstract | Convolutional neural networks (CNNs) have been successfully applied to motor imagery (MI)-based brain-computer interface (BCI). Nevertheless, single-scale CNN fail to extract abundant information over a wide spectrum from EEG signals, while typical multi-scale CNNs cannot effectively fuse information from different scales with concatenation-based methods. To overcome these challenges, we propose a new scheme equipped with attention-based dual-scale fusion convolutional neural network (ADFCNN), which jointly extracts and fuses EEG spectral and spatial information at different scales. This scheme also provides novel insight through self-attention for effective information fusion from different scales. Specifically, temporal convolutions with two different kernel sizes identify EEG μ and β rhythms, while spatial convolutions at two different scales generate global and detailed spatial information, respectively, and the self-attention mechanism performs feature fusion based on the internal similarity of the concatenated features extracted by the dual-scale CNN. The proposed scheme achieves the superior performance compared with state-of-the-art methods in subject-specific motor imagery recognition on BCI Competition IV dataset 2a, 2b and OpenBMI dataset, with the cross-session average classification accuracies of 79.39% and significant improvements of 9.14% on BCI-IV2a, 87.81% and 7.66% on BCI-IV2b, 65.26% and 7.2% on OpenBMI dataset, and the within-session average classification accuracies of 86.87% and significant improvements of 10.89% on BCI-IV2a, 87.26% and 8.07% on BCI-IV2b, 84.29% and 5.17% on OpenBMI dataset, respectively. What is more, ablation experiments are conducted to investigate the mechanism and demonstrate the effectiveness of the dual-scale joint temporal-spatial CNN and self-attention modules. Visualization is also used to reveal the learning process and feature distribution of the model. |
Keyword | Brain-computer Interface (Bci) Convolutional Neural Networks (Cnns) Motor Imagery (Mi) Self-attention Mechanism |
DOI | 10.1109/TNSRE.2023.3342331 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Engineering ; Rehabilitation |
WOS Subject | Engineering, Biomedical ; Rehabilitation |
WOS ID | WOS:001144547700020 |
Publisher | IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC, 445 HOES LANE, PISCATAWAY, NJ 08855-4141 |
Scopus ID | 2-s2.0-85180302280 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | INSTITUTE OF COLLABORATIVE INNOVATION DEPARTMENT OF ELECTRICAL AND COMPUTER ENGINEERING |
Corresponding Author | Wan, Feng |
Affiliation | 1.Institute of Collaborative Innovation, University of Macau, Faculty of Science and Technology, The Centre for Cognitive and Brain Sciences, The Centre for Artificial Intelligence and Robotics, Department of Electrical and Computer Engineering, Taipa, Macao 2.Macau University of Science and Technology, Macao Centre for Mathematical Sciences, The Resp. Dis. AI Laboratory on Epidemic Intelligence and Medical Big Data Instrument Applications, Faculty of Innovation Engineering, Taipa, Macao 3.Institute of Automation, Chinese Academy of Sciences, Brainnetome Center, Beijing, 100190, China 4.Hefei University of Technology, Department of Biomedical Engineering, Hefei, 230009, China 5.University of Science and Technology of China, Department of Electronic Engineering and Information Science, Hefei, 230027, China 6.South China University of Technology, School of Computer Science and Engineering, Guangzhou, 510006, China |
First Author Affilication | INSTITUTE OF COLLABORATIVE INNOVATION |
Corresponding Author Affilication | INSTITUTE OF COLLABORATIVE INNOVATION |
Recommended Citation GB/T 7714 | Tao, Wei,Wang, Ze,Wong, Chi Man,et al. ADFCNN: Attention-Based Dual-Scale Fusion Convolutional Neural Network for Motor Imagery Brain-Computer Interface[J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2023, 32, 154-165. |
APA | Tao, Wei., Wang, Ze., Wong, Chi Man., Jia, Ziyu., Li, Chang., Chen, Xun., Chen, C. L.Philip., & Wan, Feng (2023). ADFCNN: Attention-Based Dual-Scale Fusion Convolutional Neural Network for Motor Imagery Brain-Computer Interface. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 32, 154-165. |
MLA | Tao, Wei,et al."ADFCNN: Attention-Based Dual-Scale Fusion Convolutional Neural Network for Motor Imagery Brain-Computer Interface".IEEE Transactions on Neural Systems and Rehabilitation Engineering 32(2023):154-165. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment