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

Federated Continual Learning with Weighted Inter-client Transfer

About

There has been a surge of interest in continual learning and federated learning, both of which are important in deep neural networks in real-world scenarios. Yet little research has been done regarding the scenario where each client learns on a sequence of tasks from a private local data stream. This problem of federated continual learning poses new challenges to continual learning, such as utilizing knowledge from other clients, while preventing interference from irrelevant knowledge. To resolve these issues, we propose a novel federated continual learning framework, Federated Weighted Inter-client Transfer (FedWeIT), which decomposes the network weights into global federated parameters and sparse task-specific parameters, and each client receives selective knowledge from other clients by taking a weighted combination of their task-specific parameters. FedWeIT minimizes interference between incompatible tasks, and also allows positive knowledge transfer across clients during learning. We validate our FedWeIT against existing federated learning and continual learning methods under varying degrees of task similarity across clients, and our model significantly outperforms them with a large reduction in the communication cost. Code is available at https://github.com/wyjeong/FedWeIT

Jaehong Yoon, Wonyong Jeong, Giwoong Lee, Eunho Yang, Sung Ju Hwang• 2020

Related benchmarks

TaskDatasetResultRank
Federated Continual LearningCIFAR-100
Average Accuracy95.17
13
Federated Continual LearningImageNet-R
Avg Accuracy71.1
13
Federated Continual LearningDomainNet
Average Performance67.84
13
Accuracy PreservationDG5
Steps above reference threshold24
12
Accuracy PreservationDomainNet
Steps Above Threshold23
12
Image ClassificationDomainNet
Efficiency (E)1.42
12
Accuracy PreservationPACS
Steps above reference threshold21
12
Image ClassificationDG5
Efficiency9.29
12
Image ClassificationPACS
Efficiency Score (E)2.19
12
Continual LearningCIFAR100 Coarse
Avg Per-Task Accuracy55.16
9
Showing 10 of 10 rows

Other info

Follow for update