FedEEG: Federated EEG Decoding Via inter-Subject Structure Matching

• Published in: Proceedings of the ... IEEE International Conference on Acoustics, Speech and Signal Processing (1998) pp. 1 - 5
• Main Authors: Hang, Wenlong, Li, Jiaxing, Liang, Shuang, Wu, Yuan, Lei, Baiying, Qin, Jing, Zhang, Yu, Choi, Kup-Sze
• Format: Conference Proceeding
• Language: English
• Published: IEEE 04.06.2023
• Subjects: Benchmark testing; Class center; Deep learning; Discriminative feature; Electroencephalogram (EEG); Electroencephalography; Federated learning; Representation learning; Signal processing; Training; Training data
• ISSN: 2379-190X
• DOI: 10.1109/ICASSP49357.2023.10095564

With sufficient centralized training data coming from multiple subjects, deep learning methods have achieved powerful EEG decoding performance. However, sending each individuals' EEG data directly to a centralized server might cause privacy leakage. To overcome this issue, we present an inter-subject structure matching-based federated EEG decoding (FedEEG) framework. First, we introduce a center loss to each client (subject), which can learn multiple virtual class centers by averaging the corresponding class-specific EEG features. To mitigate the client drift issue, we then explicitly connect the learning across multiple clients by aligning their corresponding virtual class centers, thus helping to correct the local training for individual subject. The proposed FedEEG can promote the discriminative feature learning while preventing the privacy leakage issue. The experimental results on benchmark EEG datasets show that FedEEG outperforms state-of-the-art federated learning methods.