Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Federated Class-Incremental Learning

About

Federated learning (FL) has attracted growing attention via data-private collaborative training on decentralized clients. However, most existing methods unrealistically assume object classes of the overall framework are fixed over time. It makes the global model suffer from significant catastrophic forgetting on old classes in real-world scenarios, where local clients often collect new classes continuously and have very limited storage memory to store old classes. Moreover, new clients with unseen new classes may participate in the FL training, further aggravating the catastrophic forgetting of the global model. To address these challenges, we develop a novel Global-Local Forgetting Compensation (GLFC) model, to learn a global class incremental model for alleviating the catastrophic forgetting from both local and global perspectives. Specifically, to address local forgetting caused by class imbalance at the local clients, we design a class-aware gradient compensation loss and a class-semantic relation distillation loss to balance the forgetting of old classes and distill consistent inter-class relations across tasks. To tackle the global forgetting brought by the non-i.i.d class imbalance across clients, we propose a proxy server that selects the best old global model to assist the local relation distillation. Moreover, a prototype gradient-based communication mechanism is developed to protect privacy. Our model outperforms state-of-the-art methods by 4.4%-15.1% in terms of average accuracy on representative benchmark datasets.

Jiahua Dong, Lixu Wang, Zhen Fang, Gan Sun, Shichao Xu, Xiao Wang, Qi Zhu• 2022

Related benchmarks

TaskDatasetResultRank
Federated Class-Incremental LearningTiny-ImageNet 10 tasks (20 classes per task) (test)
FAA69.1
54
Federated Class-Incremental LearningCIFAR-100 Quantity-based label imbalance
FAA58.2
42
Class-incremental learningCIFAR-100 (test)
Average Accuracy66.9
22
Federated Class-Incremental LearningTinyImageNet (test)
Score_12049
17
Federated Continual LearningCIFAR-100
Average Accuracy95.35
13
Federated Continual LearningImageNet-R
Avg Accuracy72.96
13
Federated Continual LearningDomainNet
Average Performance69.75
13
Federated Class-Incremental LearningImageNet Subset 10 incremental tasks
Accuracy42.7
10
Federated Class-Incremental LearningTinyImageNet first 10 tasks 35
Performance @ 10%68.7
10
Image ClassificationTinyImageNet 35 (test)
Accuracy (10 classes)68.7
10
Showing 10 of 11 rows

Other info

Code

Follow for update