Skip to content
/ FedL2G Public

FedL2G: Learning to Guide Local Training in Heterogeneous Federated Learning

License

Notifications You must be signed in to change notification settings

TsingZ0/FedL2G

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This is the implementation of our paper: FedL2G: Learning to Guide Local Training in Heterogeneous Federated Learning.

Datasets and Baselines

Due to the file size limitation, we only upload the statistics (config.json) of the Cifar10 dataset in the Dirichlet setting. All the code for the baselines, datasets, and environments is publicly available in the popular repository HtFLlib.

System

  • main.py: System configurations.
  • total.sh: Command lines to run experiments for FedL2G with default hyperparameter settings on Linux.
  • flcore/:
    • clients/: The code on clients for both FedL2G-l and FedL2G-f.
    • servers/: The code on servers for both FedL2G-l and FedL2G-f.
    • trainmodel/: The code for some heterogeneous client models.
  • utils/:
    • data_utils.py: The code to read the dataset.
    • mem_utils.py: The code to record memory usage.
    • result_utils.py: The code to save results to files.