English

Federated Transfer Learning for EEG Signal Classification

Machine Learning 2021-01-26 v5 Signal Processing

Abstract

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.

Keywords

Cite

@article{arxiv.2004.12321,
  title  = {Federated Transfer Learning for EEG Signal Classification},
  author = {Ce Ju and Dashan Gao and Ravikiran Mane and Ben Tan and Yang Liu and Cuntai Guan},
  journal= {arXiv preprint arXiv:2004.12321},
  year   = {2021}
}

Comments

6 pages, 2 figures, Accepted for IEEE Engineering in Medicine and Biology Society (EMBC) 2020 GitHub: https://github.com/DashanGao/Federated-Transfer-Leraning-for-EEG

R2 v1 2026-06-23T15:06:07.618Z