1
|
Ma J, Ma W, Zhang J, Li Y, Yang B, Shan C. Partial prior transfer learning based on self-attention CNN for EEG decoding in stroke patients. Sci Rep 2024; 14:28170. [PMID: 39548177 PMCID: PMC11568294 DOI: 10.1038/s41598-024-79202-8] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 07/28/2024] [Accepted: 11/06/2024] [Indexed: 11/17/2024] Open
Abstract
The utilization of motor imagery-based brain-computer interfaces (MI-BCI) has been shown to assist stroke patients activate motor regions in the brain. In particular, the brain regions activated by unilateral upper limb multi-task are more extensive, which is more beneficial for rehabilitation, but it also increases the difficulty of decoding. In this paper, self-attention convolutional neural network based partial prior transfer learning (SACNN-PPTL) is proposed to improve the classification performance of patients' MI multi-task. The backbone network of the algorithm is SACNN, which accords with the inherent features of electroencephalogram (EEG) and contains the temporal feature module, the spatial feature module and the feature generalization module. In addition, PPTL is introduced to transfer part of the target domain while preserving the generalization of the base model while improving the specificity of the target domain. In the experiment, five backbone networks and three training modes are selected as comparison algorithms. The experimental results show that SACNN-PPTL had a classification accuracy of 55.4%±0.17 in four types of MI tasks for 22 patients, which is significantly higher than comparison algorithms (P < 0.05). SACNN-PPTL effectively improves the decoding performance of MI tasks and promotes the development of BCI-based rehabilitation for unilateral upper limb.
Collapse
Affiliation(s)
- Jun Ma
- Department of Rehabilitation Medicine, Tong Ren Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, 200336, China.
| | - Wanlu Ma
- China-Japan Friendship Hospital, Beijing, 100029, China
| | - Jingjing Zhang
- Department of Rehabilitation Medicine, Tong Ren Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, 200336, China
| | - Yongcong Li
- School of Mechatronic Engineering and Automation, Shanghai University, Shanghai, 200444, China
| | - Banghua Yang
- School of Mechatronic Engineering and Automation, Shanghai University, Shanghai, 200444, China.
| | - Chunlei Shan
- Department of Rehabilitation Medicine, Tong Ren Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, 200336, China.
- Institute of Rehabilitation, Shanghai Jiao Tong University School of Medicine, Shanghai, 200025, China.
| |
Collapse
|
2
|
Ma J, Yang B, Rong F, Gao S, Wang W. Motor imagery decoding using source optimized transfer learning based on multi-loss fusion CNN. Cogn Neurodyn 2024; 18:2521-2534. [PMID: 39555257 PMCID: PMC11564427 DOI: 10.1007/s11571-024-10100-5] [Citation(s) in RCA: 1] [Impact Index Per Article: 1.0] [Reference Citation Analysis] [Abstract] [Key Words] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/11/2022] [Revised: 12/24/2023] [Accepted: 03/05/2024] [Indexed: 11/19/2024] Open
Abstract
Transfer learning is increasingly used to decode multi-class motor imagery tasks. Previous transfer learning ignored the optimizability of the source model, weakened the adaptability to the target domain and limited the performance. This paper first proposes the multi-loss fusion convolutional neural network (MF-CNN) to make an optimizable source model. Then we propose a novel source optimized transfer learning (SOTL), which optimizes the source model to make it more in line with the target domain's features to improve the target model's performance. We transfer the model trained from 16 healthy subjects to 16 stroke patients. The average classification accuracy achieves 51.2 ± 0.17% in the four types of unilateral upper limb motor imagery tasks, which is significantly higher than the classification accuracy of deep learning (p < 0.001) and transfer learning (p < 0.05). In this paper, an MI model from the data of healthy subjects can be used for the classification of stroke patients and can demonstrate good classification results, which provides experiential support for the study of transfer learning and the modeling of stroke rehabilitation training.
Collapse
Affiliation(s)
- Jun Ma
- School of Mechatronic Engineering and Automation, School of Medicine, Research Center of Brain-Computer Engineering, Shanghai University, Shanghai, 200444 China
| | - Banghua Yang
- School of Mechatronic Engineering and Automation, School of Medicine, Research Center of Brain-Computer Engineering, Shanghai University, Shanghai, 200444 China
| | - Fenqi Rong
- School of Mechatronic Engineering and Automation, School of Medicine, Research Center of Brain-Computer Engineering, Shanghai University, Shanghai, 200444 China
| | - Shouwei Gao
- School of Mechatronic Engineering and Automation, School of Medicine, Research Center of Brain-Computer Engineering, Shanghai University, Shanghai, 200444 China
| | - Wen Wang
- Department of Radiology & Functional and Molecular Imaging Key Lab of Shaanxi Province, Tangdu Hospital, Fourth Military Medical University, Xi’an, 710038 Shaanxi China
| |
Collapse
|
3
|
Zhu Q, Li S, Meng X, Xu Q, Zhang Z, Shao W, Zhang D. Spatio-Temporal Graph Hubness Propagation Model for Dynamic Brain Network Classification. IEEE TRANSACTIONS ON MEDICAL IMAGING 2024; 43:2381-2394. [PMID: 38319754 DOI: 10.1109/tmi.2024.3363014] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Track Full Text] [Subscribe] [Scholar Register] [Indexed: 02/08/2024]
Abstract
Dynamic brain network has the advantage over static brain network in characterizing the variation pattern of functional brain connectivity, and it has attracted increasing attention in brain disease diagnosis. However, most of the existing dynamic brain networks analysis methods rely on extracting features from independent brain networks divided by sliding windows, making them hard to reveal the high-order dynamic evolution laws of functional brain networks. Additionally, they cannot effectively extract the spatio-temporal topology features in dynamic brain networks. In this paper, we propose to use optimal transport (OT) theory to capture the topology evolution of the dynamic brain networks, and develop a multi-channel spatio-temporal graph convolutional network that collaboratively extracts the temporal and spatial features from the evolution networks. Specifically, we first adaptively evaluate the graph hubness of brain regions in the brain network of each time window, which comprehensively models information transmission among multiple brain regions. Second, the hubness propagation information across adjacent time windows is captured by optimal transport, describing high-order topology evolution of dynamic brain networks. Moreover, we develop a spatio-temporal graph convolutional network with attention mechanism to collaboratively extract the intrinsic temporal and spatial topology information from the above networks. Finally, the multi-layer perceptron is adopted for classifying the dynamic brain network. The extensive experiment on the collected epilepsy dataset and the public ADNI dataset show that our proposed method not only outperforms several state-of-the-art methods in brain disease diagnosis, but also reveals the key dynamic alterations of brain connectivities between patients and healthy controls.
Collapse
|
4
|
Kong X, Wu C, Chen S, Wu T, Han J. Efficient Feature Learning Model of Motor Imagery EEG Signals with L1-Norm and Weighted Fusion. BIOSENSORS 2024; 14:211. [PMID: 38785685 PMCID: PMC11117874 DOI: 10.3390/bios14050211] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Grants] [Track Full Text] [Subscribe] [Scholar Register] [Received: 01/24/2024] [Revised: 03/24/2024] [Accepted: 04/18/2024] [Indexed: 05/25/2024]
Abstract
Brain-computer interface (BCI) for motor imagery is an advanced technology used in the field of medical rehabilitation. However, due to the poor accuracy of electroencephalogram feature classification, BCI systems often misrecognize user commands. Although many state-of-the-art feature selection methods aim to enhance classification accuracy, they usually overlook the interrelationships between individual features, indirectly impacting the accuracy of feature classification. To overcome this issue, we propose an adaptive feature learning model that employs a Riemannian geometric approach to generate a feature matrix from electroencephalogram signals, serving as the model's input. By integrating the enhanced adaptive L1 penalty and weighted fusion penalty into the sparse learning model, we select the most informative features from the matrix. Specifically, we measure the importance of features using mutual information and introduce an adaptive weight construction strategy to penalize regression coefficients corresponding to each variable adaptively. Moreover, the weighted fusion penalty balances weight differences among correlated variables, reducing the model's overreliance on specific variables and enhancing accuracy. The performance of the proposed method was validated on BCI Competition IV datasets IIa and IIb using the support vector machine. Experimental results demonstrate the effectiveness and superiority of the proposed model compared to the existing models.
Collapse
Affiliation(s)
- Xiangzeng Kong
- College of Mechanical and Electrical Engineering, Fujian Agriculture and Forestry University, Fuzhou 350108, China;
| | - Cailin Wu
- School of Future Technology, Fujian Agriculture and Forestry University, Fuzhou 350002, China; (C.W.); (S.C.)
| | - Shimiao Chen
- School of Future Technology, Fujian Agriculture and Forestry University, Fuzhou 350002, China; (C.W.); (S.C.)
| | - Tao Wu
- College of Mechanical and Electrical Engineering, Fujian Agriculture and Forestry University, Fuzhou 350108, China;
| | - Junfeng Han
- College of Mechanical and Electrical Engineering, Fujian Agriculture and Forestry University, Fuzhou 350108, China;
| |
Collapse
|
5
|
Polyakov D, Robinson PA, Muller EJ, Shriki O. Recruiting neural field theory for data augmentation in a motor imagery brain-computer interface. Front Robot AI 2024; 11:1362735. [PMID: 38694882 PMCID: PMC11061403 DOI: 10.3389/frobt.2024.1362735] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 12/28/2023] [Accepted: 03/20/2024] [Indexed: 05/04/2024] Open
Abstract
We introduce a novel approach to training data augmentation in brain-computer interfaces (BCIs) using neural field theory (NFT) applied to EEG data from motor imagery tasks. BCIs often suffer from limited accuracy due to a limited amount of training data. To address this, we leveraged a corticothalamic NFT model to generate artificial EEG time series as supplemental training data. We employed the BCI competition IV '2a' dataset to evaluate this augmentation technique. For each individual, we fitted the model to common spatial patterns of each motor imagery class, jittered the fitted parameters, and generated time series for data augmentation. Our method led to significant accuracy improvements of over 2% in classifying the "total power" feature, but not in the case of the "Higuchi fractal dimension" feature. This suggests that the fit NFT model may more favorably represent one feature than the other. These findings pave the way for further exploration of NFT-based data augmentation, highlighting the benefits of biophysically accurate artificial data.
Collapse
Affiliation(s)
- Daniel Polyakov
- Department of Cognitive and Brain Sciences, Ben-Gurion University of the Negev, Be’er Sheva, Israel
- Agricultural, Biological, Cognitive Robotics Initiative, Ben-Gurion University of the Negev, Be’er Sheva, Israel
| | | | - Eli J. Muller
- Brain and Mind Centre, The University of Sydney, Sydney, NSW, Australia
| | - Oren Shriki
- Department of Cognitive and Brain Sciences, Ben-Gurion University of the Negev, Be’er Sheva, Israel
- Agricultural, Biological, Cognitive Robotics Initiative, Ben-Gurion University of the Negev, Be’er Sheva, Israel
| |
Collapse
|
6
|
Rakhmatulin I, Dao MS, Nassibi A, Mandic D. Exploring Convolutional Neural Network Architectures for EEG Feature Extraction. SENSORS (BASEL, SWITZERLAND) 2024; 24:877. [PMID: 38339594 PMCID: PMC10856895 DOI: 10.3390/s24030877] [Citation(s) in RCA: 13] [Impact Index Per Article: 13.0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Track Full Text] [Subscribe] [Scholar Register] [Received: 12/04/2023] [Revised: 01/12/2024] [Accepted: 01/20/2024] [Indexed: 02/12/2024]
Abstract
The main purpose of this paper is to provide information on how to create a convolutional neural network (CNN) for extracting features from EEG signals. Our task was to understand the primary aspects of creating and fine-tuning CNNs for various application scenarios. We considered the characteristics of EEG signals, coupled with an exploration of various signal processing and data preparation techniques. These techniques include noise reduction, filtering, encoding, decoding, and dimension reduction, among others. In addition, we conduct an in-depth analysis of well-known CNN architectures, categorizing them into four distinct groups: standard implementation, recurrent convolutional, decoder architecture, and combined architecture. This paper further offers a comprehensive evaluation of these architectures, covering accuracy metrics, hyperparameters, and an appendix that contains a table outlining the parameters of commonly used CNN architectures for feature extraction from EEG signals.
Collapse
Affiliation(s)
- Ildar Rakhmatulin
- Department of Electrical and Electronic Engineering, Imperial College London, London SW7 2AZ, UK; (A.N.)
| | - Minh-Son Dao
- National Institute of Information and Communications Technology (NICT), Tokyo 184-0015, Japan
| | - Amir Nassibi
- Department of Electrical and Electronic Engineering, Imperial College London, London SW7 2AZ, UK; (A.N.)
| | - Danilo Mandic
- Department of Electrical and Electronic Engineering, Imperial College London, London SW7 2AZ, UK; (A.N.)
| |
Collapse
|
7
|
Li H, Ji H, Yu J, Li J, Jin L, Liu L, Bai Z, Ye C. A sequential learning model with GNN for EEG-EMG-based stroke rehabilitation BCI. Front Neurosci 2023; 17:1125230. [PMID: 37139522 PMCID: PMC10150013 DOI: 10.3389/fnins.2023.1125230] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 12/16/2022] [Accepted: 03/27/2023] [Indexed: 05/05/2023] Open
Abstract
Introduction Brain-computer interfaces (BCIs) have the potential in providing neurofeedback for stroke patients to improve motor rehabilitation. However, current BCIs often only detect general motor intentions and lack the precise information needed for complex movement execution, mainly due to insufficient movement execution features in EEG signals. Methods This paper presents a sequential learning model incorporating a Graph Isomorphic Network (GIN) that processes a sequence of graph-structured data derived from EEG and EMG signals. Movement data are divided into sub-actions and predicted separately by the model, generating a sequential motor encoding that reflects the sequential features of the movements. Through time-based ensemble learning, the proposed method achieves more accurate prediction results and execution quality scores for each movement. Results A classification accuracy of 88.89% is achieved on an EEG-EMG synchronized dataset for push and pull movements, significantly outperforming the benchmark method's performance of 73.23%. Discussion This approach can be used to develop a hybrid EEG-EMG brain-computer interface to provide patients with more accurate neural feedback to aid their recovery.
Collapse
Affiliation(s)
- Haoyang Li
- Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Electronic and Information Engineering, Tongji University, Shanghai, China
| | - Hongfei Ji
- Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Electronic and Information Engineering, Tongji University, Shanghai, China
- Hongfei Ji
| | - Jian Yu
- Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Electronic and Information Engineering, Tongji University, Shanghai, China
- Jian Yu
| | - Jie Li
- Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Electronic and Information Engineering, Tongji University, Shanghai, China
- *Correspondence: Jie Li
| | - Lingjing Jin
- Department of Neurology and Neurological Rehabilitation, Shanghai Disabled Person's Federation Key Laboratory of Intelligent Rehabilitation Assistive Devices and Technologies, Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Medicine, Tongji University, Shanghai, China
- Neurotoxin Research Center of Key Laboratory of Spine and Spinal Cord Injury Repair and Regeneration of Ministry of Education, Neurological Department of Tongji Hospital, School of Medicine, Tongji University, Shanghai, China
| | - Lingyu Liu
- Department of Neurology and Neurological Rehabilitation, Shanghai Disabled Person's Federation Key Laboratory of Intelligent Rehabilitation Assistive Devices and Technologies, Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Medicine, Tongji University, Shanghai, China
| | - Zhongfei Bai
- Department of Neurology and Neurological Rehabilitation, Shanghai Disabled Person's Federation Key Laboratory of Intelligent Rehabilitation Assistive Devices and Technologies, Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Medicine, Tongji University, Shanghai, China
| | - Chen Ye
- Translational Research Center, Shanghai Yangzhi Rehabilitation Hospital (Shanghai Sunshine Rehabilitation Center), School of Electronic and Information Engineering, Tongji University, Shanghai, China
| |
Collapse
|
8
|
Yang B, Ma J, Qiu W, Zhang J, Wang X. The unilateral upper limb classification from fMRI-weighted EEG signals using convolutional neural network. Biomed Signal Process Control 2022. [DOI: 10.1016/j.bspc.2022.103855] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Indexed: 11/30/2022]
|