Shypd.aiThe largest open index of AI toolsFederated-Learning-PyTorch is an open-source tool that implements communication-efficient federated learning for deep networks. It supports experiments on various datasets like MNIST and CIFAR10, with both IID and non-IID data distributions.
Trending
Also listed in
About
Federated-Learning-PyTorch provides an open-source implementation of the vanilla federated learning paradigm, as described in the paper 'Communication-Efficient Learning of Deep Networks from Decentralized Data'. This tool is built using PyTorch and allows researchers and developers to conduct experiments on popular datasets such as MNIST, Fashion MNIST, and CIFAR10. It supports both independent and identically distributed (IID) and non-IID data distributions, with options for equal or unequal data splits among users. The implementation focuses on simple models like MLP and CNN to illustrate the effectiveness of federated learning, making it a valuable resource for understanding and experimenting with this distributed machine learning approach.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
