Adaptive Federated Learning for EEG Emotion Recognition

Calvin Chan, Qingqing Zheng, Chengjian Xu, Qiong Wang, Pheng-Ann Heng

Abstract: Emotion classification based on electroencephalogram (EEG) signals has drawn huge attention in affective brain computer interface (BCI). Recently, plenty of deep learning approaches have been proposed to improve the performance of EEG emotion recognition, especially the application of domain adaptation methods to tackle the challenge of large individual differences of EEG signals from subject to subject. However, these conventional transfer learning methods would result in information leakage during the sharing of domain data to enhance the accuracy of the target tasks. Therefore, in this paper, we proposed a distributed deep learning method, named adaptive federated learning (AdaFL) for EEG emotion recognition. In AdaFL, a server collaboratively learns a global model by adaptively aggregating the local models according to their importance in several communication rounds. In particular, an importance function is developed to evaluate each client, which would determine to select a subset of optimal clients for subsequent global model aggregation. The function is a simple transformation of the training loss and the sample size of the local models. Then, the resulting importance scores of selected clients are further converted into aggregation coefficients to measure the weights of the local models for global model aggregation. The distinct advantage of AdaFL is that the cross-subject information could be well utilized and the information leakage risk could be significantly reduced. To validate the efficacy of the proposed AdaFL, we conduct extensive experiments on two real EEG emotion datasets, i.e., SEED and DEAP. The experimental results show that the proposed AdaFL has achieved 94.95 ± 0.40% and 91.06 ± 0.29% average classification accuracy on the SEED and DEAP datasets, respectively, which reflects the superiority of our method over the state-of-the-art approaches.