fedpa

Decentralized optimizer

A modular JAX implementation of federated learning via posterior averaging for decentralized optimization

Federated posterior averaging implemented in JAX

GitHub

50 stars
2 watching
16 forks
Language: Jupyter Notebook
last commit: over 3 years ago
federated-learningjaxpython

Related projects:

RepositoryDescriptionStars
divyansh03/fedexpAn implementation of a federated averaging algorithm with an extrapolation approach to speed up distributed machine learning training on client-held data.9
mc-nya/fednestAn implementation of a federated optimization algorithm for distributed machine learning6
debcaldarola/fedsamImproving generalization in federated learning by seeking flat minima through optimization techniques82
litian96/fedproxAn optimization framework designed to address heterogeneity in federated learning across distributed networks655
lyn1874/fedpvrAn implementation of a federated learning algorithm for handling heterogeneous data6
optimization-ai/icml2023_fedxlAn implementation of a federated learning algorithm for optimization problems with compositional pairwise risk optimization.2
zackzikaixiao/fedgrabA tool for training federated learning models with adaptive gradient balancing to handle class imbalance in multi-client scenarios.14
yuetan031/fedprotoAn implementation of federated learning with prototype-based methods across heterogeneous clients134
harliwu/fedamdThis project presents an approach to federated learning with partial client participation by optimizing anchor selection for improving model accuracy and convergence.2
zfancy/sfatCombating heterogeneity in federated learning by combining adversarial training with client-wise slack during aggregation28
ignavierng/notears-admmAn implementation of Bayesian network structure learning with continuous optimization for federated learning.10
illidanlab/splitmixAn algorithm for distributed learning with flexible model customization during training and testing40
dos-group/fedzeroAn implementation of federated learning optimized for training on renewable energy sources and spare compute capacity to minimize carbon emissions.19
omarfoq/fedemDevelops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions.157
yamingguo98/fediirAn implementation of a federated learning algorithm that generalizes to out-of-distribution scenarios using implicit invariant relationships10