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 1 year ago
federated-learningjaxpython

Related projects:

Repository Description Stars
divyansh03/fedexp An implementation of a federated averaging algorithm with an extrapolation approach to speed up distributed machine learning training on client-held data. 9
mc-nya/fednest An implementation of a federated optimization algorithm for distributed machine learning 6
debcaldarola/fedsam Improving generalization in federated learning by seeking flat minima through optimization techniques 82
litian96/fedprox An optimization framework designed to address heterogeneity in federated learning across distributed networks 655
lyn1874/fedpvr An implementation of a federated learning algorithm for handling heterogeneous data 6
optimization-ai/icml2023_fedxl An implementation of a federated learning algorithm for optimization problems with compositional pairwise risk optimization. 2
zackzikaixiao/fedgrab A tool for training federated learning models with adaptive gradient balancing to handle class imbalance in multi-client scenarios. 14
yuetan031/fedproto An implementation of federated learning with prototype-based methods across heterogeneous clients 134
harliwu/fedamd This project presents an approach to federated learning with partial client participation by optimizing anchor selection for improving model accuracy and convergence. 2
zfancy/sfat Combating heterogeneity in federated learning by combining adversarial training with client-wise slack during aggregation 28
ignavierng/notears-admm An implementation of Bayesian network structure learning with continuous optimization for federated learning. 10
illidanlab/splitmix An algorithm for distributed learning with flexible model customization during training and testing 40
dos-group/fedzero An implementation of federated learning optimized for training on renewable energy sources and spare compute capacity to minimize carbon emissions. 19
omarfoq/fedem Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions. 157
yamingguo98/fediir An implementation of a federated learning algorithm that generalizes to out-of-distribution scenarios using implicit invariant relationships 10