Dynamic heterogeneous federated learning with multi-level prototypes

被引:8
作者
Guo, Shunxin
Wang, Hongsong
Geng, Xin [1 ]
机构
[1] Southeast Univ, Sch Comp Sci & Engn, Nanjing 210096, Peoples R China
基金
美国国家科学基金会;
关键词
Heterogeneous federated learning; Multi-level prototypes;
D O I
10.1016/j.patcog.2024.110542
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Federated learning shows promise as a privacy -preserving collaborative learning technique. Existing research mainly focuses on skewing the class distribution across clients. However, most approaches suffer from catastrophic forgetting and classifier shift, mainly when the global distribution of all classes is extremely unbalanced and the data distribution of the client dynamically evolves over time. In this paper, we study the Dynamic Heterogeneous Federated Learning, which addresses the practical scenario where heterogeneous data distributions exist among different clients and dynamic tasks within the client. Accordingly, we propose a novel federated learning framework named Federated Multi -Level Prototypes and design federated multi -level regularizations. To mitigate classifier shift, we construct semantic prototypes to provide fruitful generalization knowledge. To maintain the model stability and consistency convergence, three regularizations are introduced as training losses, i.e., prototype -based regularization, semantic prototype -based regularization, and federated inter -task regularization. Extensive experiments show that the proposed method achieves advanced performance in various settings.
引用
收藏
页数:12
相关论文
共 45 条
[1]   Robust fe derate d learning under statistical heterogeneity via Hessian spectral decomposition [J].
Ahmad, Adnan ;
Luo, Wei ;
Robles-Kelly, Antonio .
PATTERN RECOGNITION, 2023, 141
[2]  
Cao KD, 2019, ADV NEUR IN, V32
[3]  
Chou YH, 2022, Data Min, P451
[4]  
Cohen G, 2017, IEEE IJCNN, P2921, DOI 10.1109/IJCNN.2017.7966217
[5]  
Collins L, 2021, PR MACH LEARN RES, V139
[6]   A tutorial on the cross-entropy method [J].
De Boer, PT ;
Kroese, DP ;
Mannor, S ;
Rubinstein, RY .
ANNALS OF OPERATIONS RESEARCH, 2005, 134 (01) :19-67
[7]  
Deng Li., 2012, Signal Processing Magazine, IEEE, V29, P141, DOI [DOI 10.1109/MSP.2012.2211477, 10.1109/msp.2012.2211477]
[8]  
Dinh CT, 2020, ADV NEUR IN, V33
[9]  
Dongetal J., 2022, P IEEE CVF C COMP VI, P10164
[10]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448