Official implementation for our TMLR 2026 paper: FedProTIP
FedProTIP is a federated continual learning method designed for task-agnostic settings, where clients learn from sequentially arriving tasks without relying on explicit task identities during inference.
The method combines replay-free gradient projection with task identity prediction to mitigate catastrophic forgetting while preserving privacy constraints in federated learning.
This repository provides code to reproduce experiments on:
- 10-split CIFAR-100
- 6-split DomainNet
- ImageNet-R with 5, 10, and 20 task splits
For more details, please refer to our paper:
OpenReview: https://openreview.net/forum?id=GW4aw0fUKC
Install the required dependencies with:
pip install -r requirements.txtpython main.py --config configs/cifar100/fedprotip.jsonpython main.py --config configs/domainnet/dom_fedprotip.jsonpython main.py --config configs/imagenetr/imagenet-r_fedprotip.json --n_tasks 5 --increment 40python main.py --config configs/imagenetr/imagenet-r_fedprotip.json --n_tasks 10 --increment 20python main.py --config configs/imagenetr/imagenet-r_fedprotip.json --n_tasks 20 --increment 10This repository builds on and adapts components from the following open-source implementations:
We sincerely thank the authors for making their code publicly available.
If you find this repository useful, please cite our paper!
@article{cha2025task,
title={Task-Agnostic Federated Continual Learning via Replay-Free Gradient Projection},
author={Cha, Seohyeon and Chen, Huancheng and Vikalo, Haris},
journal={arXiv preprint arXiv:2509.21606},
year={2025}
}