P3DL: A Privacy Preserving Personalized Distributed Learning Framework for EEG-based Cognitive State Identification.

Abstract

Electroencephalography (EEG)-based brain cognitive state identification for the elderly allows timely detection and early intervention of cognitive deterioration. Notably, EEG signals carry a great deal of vital personal information. However, a majority of the existing cognitive evaluations focus on improving the accuracy of EEG decoding and enhancing the performance of identification models, while neglecting the privacy protection of EEG data. To address the risky challenge, we propose a privacy-preserving personalized distributed learning framework (P3DL) for cognitive state identification. Specifically, it consists of the clients and a central server. Each client contains a cognitive model and a score model for identifying cognitive states and quantifying cognitive levels, respectively. The central server can aggregate local models' parameters from distributed clients, then, update and downstream the global model's parameters for iterative optimization. A federated dynamic update strategy (FedDBS) is designed to jointly update all global and local models with a supervisory metric. In order to further improve the identification performance and judge the misdiagnosis level, a novel loss function, extreme error Loss (E2Loss), is proposed. Compared with the baseline, experimental results on our self-collected clinical dataset and a public dataset show an average increase in F2Score of 5.58% and 3.31%, and in accuracy of 1.78% and 2.46%, respectively. Furthermore, the scalability of the framework has been proved in the emotion recognition task. Our proposed framework P3DL can not only improve the identification performance, but also protect the privacy of EEG, opening a new window for secure healthcare.