├── LICENSE ├── README.md ├── algorithms └── basePS │ ├── ps_aggregator.py │ └── ps_client_trainer.py ├── algorithms_standalone ├── basePS │ ├── aggregator.py │ ├── basePSmanager.py │ └── client.py ├── fedavg │ ├── FedAVGManager.py │ ├── aggregator.py │ └── client.py └── fednova │ ├── FedNovaManager.py │ ├── aggregator.py │ └── client.py ├── asset ├── FedFed.png └── tab1.jpg ├── config.yaml ├── configs ├── __init__.py ├── config.py └── default.py ├── data_preprocessing ├── FashionMNIST │ ├── __init__.py │ ├── data_loader.py │ └── datasets.py ├── SVHN │ ├── __init__.py │ ├── data_loader.py │ └── datasets.py ├── __init__.py ├── build.py ├── cifar10 │ ├── __init__.py │ ├── centralized_loader.py │ ├── data_loader.py │ ├── datasets.py │ └── iid_data_loader.py ├── cifar100 │ ├── __init__.py │ └── datasets.py ├── loader.py └── utils │ ├── data_distribution.py │ ├── data_distribution_visualize.py │ ├── data_distribution_visualize_plt.py │ ├── data_visualize_manual.py │ ├── imbalance_data.py │ ├── stats.py │ ├── transform_utils.py │ └── utils.py ├── environment.yaml ├── loss_fn ├── __init__.py └── build.py ├── lr_scheduler ├── __init__.py ├── base_lr_scheduler.py ├── build.py ├── consine_lr_scheduler.py ├── multisteplr_scheduler.py ├── steplr_scheduler.py └── torch_scheduler.py ├── main.py ├── model ├── FL_VAE.py ├── __init__.py ├── build.py └── cv │ ├── others.py │ └── resnet_v2.py ├── optim ├── AdamW.py ├── __init__.py └── build.py ├── requirements.txt ├── trainers ├── averager.py ├── build.py └── normal_trainer.py └── utils ├── __init__.py ├── data_utils.py ├── log_info.py ├── logger.py ├── meter.py ├── normalize.py ├── randaugment4fixmatch.py ├── set.py └── tool.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/README.md -------------------------------------------------------------------------------- /algorithms/basePS/ps_aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms/basePS/ps_aggregator.py -------------------------------------------------------------------------------- /algorithms/basePS/ps_client_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms/basePS/ps_client_trainer.py -------------------------------------------------------------------------------- /algorithms_standalone/basePS/aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/basePS/aggregator.py -------------------------------------------------------------------------------- /algorithms_standalone/basePS/basePSmanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/basePS/basePSmanager.py -------------------------------------------------------------------------------- /algorithms_standalone/basePS/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/basePS/client.py -------------------------------------------------------------------------------- /algorithms_standalone/fedavg/FedAVGManager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fedavg/FedAVGManager.py -------------------------------------------------------------------------------- /algorithms_standalone/fedavg/aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fedavg/aggregator.py -------------------------------------------------------------------------------- /algorithms_standalone/fedavg/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fedavg/client.py -------------------------------------------------------------------------------- /algorithms_standalone/fednova/FedNovaManager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fednova/FedNovaManager.py -------------------------------------------------------------------------------- /algorithms_standalone/fednova/aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fednova/aggregator.py -------------------------------------------------------------------------------- /algorithms_standalone/fednova/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/algorithms_standalone/fednova/client.py -------------------------------------------------------------------------------- /asset/FedFed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/asset/FedFed.png -------------------------------------------------------------------------------- /asset/tab1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/asset/tab1.jpg -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/config.yaml -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | from .config import * -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/configs/default.py -------------------------------------------------------------------------------- /data_preprocessing/FashionMNIST/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_preprocessing/FashionMNIST/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/FashionMNIST/data_loader.py -------------------------------------------------------------------------------- /data_preprocessing/FashionMNIST/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/FashionMNIST/datasets.py -------------------------------------------------------------------------------- /data_preprocessing/SVHN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_preprocessing/SVHN/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/SVHN/data_loader.py -------------------------------------------------------------------------------- /data_preprocessing/SVHN/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/SVHN/datasets.py -------------------------------------------------------------------------------- /data_preprocessing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_preprocessing/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/build.py -------------------------------------------------------------------------------- /data_preprocessing/cifar10/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_preprocessing/cifar10/centralized_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/cifar10/centralized_loader.py -------------------------------------------------------------------------------- /data_preprocessing/cifar10/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/cifar10/data_loader.py -------------------------------------------------------------------------------- /data_preprocessing/cifar10/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/cifar10/datasets.py -------------------------------------------------------------------------------- /data_preprocessing/cifar10/iid_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/cifar10/iid_data_loader.py -------------------------------------------------------------------------------- /data_preprocessing/cifar100/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_preprocessing/cifar100/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/cifar100/datasets.py -------------------------------------------------------------------------------- /data_preprocessing/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/loader.py -------------------------------------------------------------------------------- /data_preprocessing/utils/data_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/data_distribution.py -------------------------------------------------------------------------------- /data_preprocessing/utils/data_distribution_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/data_distribution_visualize.py -------------------------------------------------------------------------------- /data_preprocessing/utils/data_distribution_visualize_plt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/data_distribution_visualize_plt.py -------------------------------------------------------------------------------- /data_preprocessing/utils/data_visualize_manual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/data_visualize_manual.py -------------------------------------------------------------------------------- /data_preprocessing/utils/imbalance_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/imbalance_data.py -------------------------------------------------------------------------------- /data_preprocessing/utils/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/stats.py -------------------------------------------------------------------------------- /data_preprocessing/utils/transform_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/transform_utils.py -------------------------------------------------------------------------------- /data_preprocessing/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/data_preprocessing/utils/utils.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/environment.yaml -------------------------------------------------------------------------------- /loss_fn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /loss_fn/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/loss_fn/build.py -------------------------------------------------------------------------------- /lr_scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lr_scheduler/base_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/base_lr_scheduler.py -------------------------------------------------------------------------------- /lr_scheduler/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/build.py -------------------------------------------------------------------------------- /lr_scheduler/consine_lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/consine_lr_scheduler.py -------------------------------------------------------------------------------- /lr_scheduler/multisteplr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/multisteplr_scheduler.py -------------------------------------------------------------------------------- /lr_scheduler/steplr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/steplr_scheduler.py -------------------------------------------------------------------------------- /lr_scheduler/torch_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/lr_scheduler/torch_scheduler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/main.py -------------------------------------------------------------------------------- /model/FL_VAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/model/FL_VAE.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/model/build.py -------------------------------------------------------------------------------- /model/cv/others.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/model/cv/others.py -------------------------------------------------------------------------------- /model/cv/resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/model/cv/resnet_v2.py -------------------------------------------------------------------------------- /optim/AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/optim/AdamW.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /optim/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/optim/build.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/requirements.txt -------------------------------------------------------------------------------- /trainers/averager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/trainers/averager.py -------------------------------------------------------------------------------- /trainers/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/trainers/build.py -------------------------------------------------------------------------------- /trainers/normal_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/trainers/normal_trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/log_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/log_info.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/normalize.py -------------------------------------------------------------------------------- /utils/randaugment4fixmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/randaugment4fixmatch.py -------------------------------------------------------------------------------- /utils/set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/set.py -------------------------------------------------------------------------------- /utils/tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tmlr-group/FedFed/HEAD/utils/tool.py --------------------------------------------------------------------------------