Search papers, labs, and topics across Lattice.
This paper introduces One-Shot Incremental Federated Learning (OSI-FL), a novel federated learning framework designed to mitigate communication overhead and catastrophic forgetting in incremental learning scenarios. OSI-FL leverages a frozen vision-language model (VLM) to generate category-specific embeddings at each client, which are then used by a pre-trained diffusion model at the server to synthesize new data. To combat catastrophic forgetting, the framework employs Selective Sample Retention (SSR), retaining the most informative samples based on loss during training.
By synthesizing data with a diffusion model and selectively retaining informative samples, OSI-FL achieves state-of-the-art performance in class-incremental and domain-incremental federated learning scenarios while minimizing communication costs.
Modern big-data systems generate massive, heterogeneous, and geographically dispersed streams that are large-scale and privacy-sensitive, making centralization challenging. While federated learning (FL) provides a privacy-enhancing training mechanism, it assumes a static data flow and learns a collaborative model over multiple rounds, making learning with \textit{incremental} data challenging in limited-communication scenarios. This paper presents One-Shot Incremental Federated Learning (OSI-FL), the first FL framework that addresses the dual challenges of communication overhead and catastrophic forgetting. OSI-FL communicates category-specific embeddings, devised by a frozen vision-language model (VLM) from each client in a single communication round, which a pre-trained diffusion model at the server uses to synthesize new data similar to the client's data distribution. The synthesized samples are used on the server for training. However, two challenges still persist: i) tasks arriving incrementally need to retrain the global model, and ii) as future tasks arrive, retraining the model introduces catastrophic forgetting. To this end, we augment training with Selective Sample Retention (SSR), which identifies and retains the top-p most informative samples per category and task pair based on sample loss. SSR bounds forgetting by ensuring that representative retained samples are incorporated into training in further iterations. The experimental results indicate that OSI-FL outperforms baselines, including traditional and one-shot FL approaches, in both class-incremental and domain-incremental scenarios across three benchmark datasets.
