Federated Learning Under Temporal Drift -- Mitigating Catastrophic Forgetting via Experience Replay

Sahasra Kokkula, Daniel David, Aaditya Baruah
1/19/2026
cs.LGcs.DC

No confident code implementation found for this paper.

Abstract

Federated Learning struggles under temporal concept drift where client data distributions shift over time. We demonstrate that standard FedAvg suffers catastrophic forgetting under seasonal drift on Fashion-MNIST, with accuracy dropping from 74% to 28%. We propose client-side experience replay, where each client maintains a small buffer of past samples mixed with current data during local training. This simple approach requires no changes to server aggregation. Experiments show that a 50-sample-per-class buffer restores performance to 78-82%, effectively preventing forgetting. Our ablation study reveals a clear memory-accuracy trade-off as buffer size increases.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{kokkula2026federated,
  title  = {Federated Learning Under Temporal Drift -- Mitigating Catastrophic Forgetting via Experience Replay},
  author = {Sahasra Kokkula and Daniel David and Aaditya Baruah},
  year   = {2026},
  eprint = {2601.13456},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.13456v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.13456.svg)](https://codeofpaper.com/paper/2601.13456)

Discussion