The limited availability of sleep data collected by individual institutions poses a significant challenge in developing automatic sleep staging models, especially given that most existing models rely heavily on data-driven approaches. The advent of federated learning has introduced an innovative and reliable paradigm for inter-institutional collaboration, facilitating task knowledge sharing and implicit data augmentation while preserving privacy. However, in traditional federated learning methodologies, local models are typically initialized with the global model at the onset of each iteration. This global-knowledge-centric approach often results in catastrophic forgetting during local training and inadequately adapts to the diversity of local data, compromising the overall generalization performance of the model. In this paper, we introduce a novel re-aggregation strategy for local model initialization that synergizes the global model with historical local models, thereby mitigating the undue influence of global knowledge and preserving local task-specific information. The re-aggregation weights are adaptively determined based on the confidence of the global and historical models on local datasets. Furthermore, to address the inherent ambiguity among sleep stages, a dual prototype-contrastive learning module is proposed, comprising Prototype-Consistency Contrastive (PCC) and Prototype-Distinguishability Contrastive (PDC) components. Specifically, the PCC component is designed to ensure consistency between local prototypes and unbiased prototypes derived from re-aggregation, enhancing intra-class knowledge coherence. The goal of the PDC component is to strengthen the discriminability among various sleep stages, improving inter-class differentiation. Comprehensive experiments conducted on two public datasets demonstrate the effectiveness, superiority, and flexibility of the proposed method.
基金:
STI2030-Major Projects [2021ZD0204300]; National Natural Science Foundation of China [62176009]; Beijing Municipal Education Commission [23JA002]; Municipal Natural Science Foundation of China [23JA002]
第一作者机构:[1]Beijing Univ Technol, Coll Comp Sci, 100 Pingleyuan, Beijing 100124, Peoples R China[2]Beijing Key Lab Trusted Comp, Beijing 100124, Peoples R China[3]China Natl Engn Lab Crit Technol Informat Secur Cl, Beijing 100124, Peoples R China
通讯作者:
通讯机构:[1]Beijing Univ Technol, Coll Comp Sci, 100 Pingleyuan, Beijing 100124, Peoples R China[2]Beijing Key Lab Trusted Comp, Beijing 100124, Peoples R China[3]China Natl Engn Lab Crit Technol Informat Secur Cl, Beijing 100124, Peoples R China[4]Capital Med Univ, Xuanwu Hosp, Dept Neurol, 45 Changchun St, Beijing 10053, Peoples R China[5]Beijing Key Lab Neuromodulat, Beijing 100053, Peoples R China
推荐引用方式(GB/T 7714):
Ma Bian,Duan Lijuan,Huang Zhaoyang,et al.A federated sleep staging method based on adaptive re-aggregation and double prototype-contrastive using single-channel electroencephalogram[J].ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE.2025,156:doi:10.1016/j.engappai.2025.111060.
APA:
Ma, Bian,Duan, Lijuan,Huang, Zhaoyang&Qiao, Yuanhua.(2025).A federated sleep staging method based on adaptive re-aggregation and double prototype-contrastive using single-channel electroencephalogram.ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE,156,
MLA:
Ma, Bian,et al."A federated sleep staging method based on adaptive re-aggregation and double prototype-contrastive using single-channel electroencephalogram".ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE 156.(2025)