GAMF

February 15, 2023 ยท View on GitHub

This repo provides code for 'Deep Neural Network Fusion via Graph Matching with Applications to Model Ensemble and Federated Learning', ICML 2022.

Run federated_learning_with_FedML.py to test the performance of our proposed GAMF.

Our code is based on the code of FedML and OTFusion, thanks for their open-source code.

For a simpler version, please refer to here.

Performance on Cifar-10 with 5 clients. Performance on Cifar-10 with 10 clients.