Federated Transfer Learning for EEG Signal Classification
By Ce Ju, Dashan Gao, Ravikiran Mane, Ben Tan, Yang Liu, Cuntai Guan. 2020
131 citations recorded by the source corpus.
The success of deep learning (DL) methods in the Brain-Computer Interfaces (BCI) field for classification of electroencephalographic (EEG) recordings has been restricted by the lack of large datasets. Privacy concerns associated with EEG signals limit the possibility of constructing a large EEG-BCI dataset by the conglomeration of multiple small ones for jointly training machine learning models. Hence, in this paper, we propose a novel privacy-preserving DL architecture named federated transfer learning (FTL) for EEG classification that is based on the federated learning framework. Working with the single-trial covariance matrix, the proposed architecture extracts common discriminative information from multi-subject EEG data with the help of domain adaptation techniques. We evaluate the performance of the proposed architecture on the PhysioNet dataset for 2-class motor imagery classification. While avoiding the actual data sharing, our FTL approach achieves 2% higher classification accuracy in a subject-adaptive analysis. Also, in the absence of multi-subject data, our architecture provides 6% better accuracy compared to other state-of-the-art DL architectures.
Subfields: Cognitive Neuroscience, Electrical and Electronic Engineering
How researchers see this paper
The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.
In Cognitive Neuroscience
In Electrical and Electronic Engineering
Citations in the corpus
- FedML: A Research Library and Benchmark for Federated Machine Learning (2020, background): “…computer vision [77, 78], natural language processing [79, 80, 81, 82, 83], finance [74, 35, 84], transportation [85, 86, 87, 88, 89, 90, 91, 92, 93, 94], digital health [95, 96, 97, 98, 99, 100, 101], recommendation [102, 103, 104, 105, 106, 107], robotics [108, 109], and smart cities [110, 111].”
- Privacy-Preserving Technology to Help Millions of People: Federated Prediction Model for Stroke Prevention (2020, background): “Ju and Gao [Gao et al., 2019; Ju et al., 2020] are first to investigate the feasibility of the federated learning framework to enable a distributed training of brain-computer interface models from multiple datasets with heterogeneous configurations.”
- Federated Learning for Healthcare: Systematic Review and Architecture Proposal (2022, background): “[40] explore transfer learning to improve the accuracy of models to analyze EEG (electroencephalogram) data to enable brain-computer interfaces.”
- Vertical Federated Learning: Concepts, Advances, and Challenges (2022, methodology): “For example, EEG data from multiple subjects with heterogeneous distributions collaboratively build BCI models using FTL [5].”