Better Generative Replay for Continual Federated Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qi, Daiqing, Zhao, Handong, Li, Sheng
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911532973031424
author Qi, Daiqing
Zhao, Handong
Li, Sheng
author_facet Qi, Daiqing
Zhao, Handong
Li, Sheng
contents Federated learning is a technique that enables a centralized server to learn from distributed clients via communications without accessing the client local data. However, existing federated learning works mainly focus on a single task scenario with static data. In this paper, we introduce the problem of continual federated learning, where clients incrementally learn new tasks and history data cannot be stored due to certain reasons, such as limited storage and data retention policy. Generative replay based methods are effective for continual learning without storing history data, but adapting them for this setting is challenging. By analyzing the behaviors of clients during training, we find that the unstable training process caused by distributed training on non-IID data leads to a notable performance degradation. To address this problem, we propose our FedCIL model with two simple but effective solutions: model consolidation and consistency enforcement. Our experimental results on multiple benchmark datasets demonstrate that our method significantly outperforms baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2302_13001
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Better Generative Replay for Continual Federated Learning
Qi, Daiqing
Zhao, Handong
Li, Sheng
Machine Learning
Artificial Intelligence
Federated learning is a technique that enables a centralized server to learn from distributed clients via communications without accessing the client local data. However, existing federated learning works mainly focus on a single task scenario with static data. In this paper, we introduce the problem of continual federated learning, where clients incrementally learn new tasks and history data cannot be stored due to certain reasons, such as limited storage and data retention policy. Generative replay based methods are effective for continual learning without storing history data, but adapting them for this setting is challenging. By analyzing the behaviors of clients during training, we find that the unstable training process caused by distributed training on non-IID data leads to a notable performance degradation. To address this problem, we propose our FedCIL model with two simple but effective solutions: model consolidation and consistency enforcement. Our experimental results on multiple benchmark datasets demonstrate that our method significantly outperforms baselines.
title Better Generative Replay for Continual Federated Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2302.13001