├── .gitignore ├── DataBalance.py ├── DataProcessor.py ├── DatasetDifference.ipynb ├── LICENSE ├── README.md ├── main_fed.py ├── models ├── Fed.py ├── Nets.py ├── Update.py ├── __init__.py └── test.py ├── save ├── .gitkeep ├── fed_mnist_cnn_2_C0.1_iidTrue.png ├── fed_mnist_mlp_10_C0.1_iidFalse.png └── fed_mnist_mlp_2_C0.1_iidFalse.png ├── test_figures ├── confusion_matrix_non-self_balanced_global.png ├── confusion_matrix_non-self_balanced_local.png ├── confusion_matrix_non-self_balanced_size.png ├── confusion_matrix_self_balanced_global.png ├── confusion_matrix_self_balanced_local.png └── confusion_matrix_self_balanced_size.png └── utils ├── __init__.py ├── options.py └── sampling.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/.gitignore -------------------------------------------------------------------------------- /DataBalance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/DataBalance.py -------------------------------------------------------------------------------- /DataProcessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/DataProcessor.py -------------------------------------------------------------------------------- /DatasetDifference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/DatasetDifference.ipynb -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/README.md -------------------------------------------------------------------------------- /main_fed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/main_fed.py -------------------------------------------------------------------------------- /models/Fed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/models/Fed.py -------------------------------------------------------------------------------- /models/Nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/models/Nets.py -------------------------------------------------------------------------------- /models/Update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/models/Update.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/models/test.py -------------------------------------------------------------------------------- /save/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /save/fed_mnist_cnn_2_C0.1_iidTrue.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/save/fed_mnist_cnn_2_C0.1_iidTrue.png -------------------------------------------------------------------------------- /save/fed_mnist_mlp_10_C0.1_iidFalse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/save/fed_mnist_mlp_10_C0.1_iidFalse.png -------------------------------------------------------------------------------- /save/fed_mnist_mlp_2_C0.1_iidFalse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/save/fed_mnist_mlp_2_C0.1_iidFalse.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_non-self_balanced_global.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_non-self_balanced_global.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_non-self_balanced_local.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_non-self_balanced_local.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_non-self_balanced_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_non-self_balanced_size.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_self_balanced_global.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_self_balanced_global.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_self_balanced_local.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_self_balanced_local.png -------------------------------------------------------------------------------- /test_figures/confusion_matrix_self_balanced_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/test_figures/confusion_matrix_self_balanced_size.png -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/utils/options.py -------------------------------------------------------------------------------- /utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mtang724/Self-Balancing-Federated-Learning/HEAD/utils/sampling.py --------------------------------------------------------------------------------