arXiv · 1910.07796
Overcoming Forgetting in Federated Learning on Non-IID Data
Abstract
We tackle the problem of Federated Learning in the non i.i.d. case, in which local models drift apart, inhibiting learning. Building on an analogy with Lifelong Learning, we adapt a solution for catastrophic forgetting to Federated Learning. We add a penalty term to the loss function, compelling all local models to converge to a shared optimum. We show that this can be done efficiently for communication (adding no further privacy risks), scaling with the number of nodes in the distributed setting. Our experiments show that this method is superior to competing ones for image recognition on the MNIST dataset.
Explore related subjects
Keep this discovery
Neta Shoham, Tomer Avidor, Aviv Keren, Nadav Israel, Daniel Benditkis, Liron Mor-Yosef, Itai Zeitak. 2019-10-17. Overcoming Forgetting in Federated Learning on Non-IID Data. https://arxiv.org/abs/1910.07796
Cite the original work for its findings. Save a collection to share your selection of sources.