Federated Motor Imagery Classification for Privacy-Preserving Brain-Computer Interfaces

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jia, Tianwang, Meng, Lubin, Li, Siyang, Liu, Jiajing, Wu, Dongrui
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910724087873536
author Jia, Tianwang
Meng, Lubin
Li, Siyang
Liu, Jiajing
Wu, Dongrui
author_facet Jia, Tianwang
Meng, Lubin
Li, Siyang
Liu, Jiajing
Wu, Dongrui
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_01079
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated Motor Imagery Classification for Privacy-Preserving Brain-Computer Interfaces
Jia, Tianwang
Meng, Lubin
Li, Siyang
Liu, Jiajing
Wu, Dongrui
Machine Learning
Human-Computer Interaction
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.
title Federated Motor Imagery Classification for Privacy-Preserving Brain-Computer Interfaces
topic Machine Learning
Human-Computer Interaction
url https://arxiv.org/abs/2412.01079