Federated Motor Imagery Classification for Privacy-Preserving Brain-Computer Interfaces
arXiv:2412.01079 · doi:10.1109/TNSRE.2024.3457504
Abstract
Training an accurate classifier for EEG-based brain-computer interface (BCI) requires EEG data from a large number of users, whereas protecting their data privacy is a critical consideration. Federated learning (FL) is a promising solution to this challenge. This paper proposes Federated classification with local Batch-specific batch normalization and Sharpness-aware minimization (FedBS) for privacy protection in EEG-based motor imagery (MI) classification. FedBS utilizes local batch-specific batch normalization to reduce data discrepancies among different clients, and sharpness-aware minimization optimizer in local training to improve model generalization. Experiments on three public MI datasets using three popular deep learning models demonstrated that FedBS outperformed six state-of-the-art FL approaches. Remarkably, it also outperformed centralized training, which does not consider privacy protection at all. In summary, FedBS protects user EEG data privacy, enabling multiple BCI users to participate in large-scale machine learning model training, which in turn improves the BCI decoding accuracy.
References in corpus (9)
- EEGNet: A Compact Convolutional Network for EEG-based Brain-Computer Interfaces
- Deep learning with convolutional neural networks for EEG decoding and visualization
- Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classification
- MOABB: Trustworthy algorithm benchmarking for BCIs
- Transfer Learning for Motor Imagery Based Brain-Computer Interfaces: A Complete Pipeline
- Federated Transfer Learning for EEG Signal Classification
- Privacy-Preserving Brain-Computer Interfaces: A Systematic Review
- Privacy-Preserving Machine Learning: Methods, Challenges and Directions
- How deep is deep enough? -- Quantifying class separability in the hidden layers of deep neural networks