Code to reproduce our paper: FedNAR: Federated Optimization with Normalized Annealing Regularization

Requirements

pip install -r requirements

Training

To access all the training scripts, you can find them in the ./scripts directory. For all the scripts, you can enable --use_nar to use FedNAR variants. For the Shakespeare dataset, please generate the required data by following the instructions provided in ./LEAF/shakespeare.

Acknowledgements

We would like to extend our gratitude to the authors of the following works: FedExP, FedDyn, and LEAF. Our codes are built upon their open-source projects.