fedpa
by alshedivat
Federated posterior averaging implemented in JAX
AI summary
Decentralized optimizer
A modular JAX implementation of federated learning via posterior averaging for decentralized optimization
- stars
- 50
- forks
- 16
- watching
- 2
Similar projects
Found by comparing what the projects do, not just their names.
Federated Averaging Optimizer
An implementation of a federated averaging algorithm with an extrapolation approach to speed up distributed machine learning training on client-held data.
Federated Optimizer
An implementation of a federated optimization algorithm for distributed machine learning
Flat minimization optimizer
Improving generalization in federated learning by seeking flat minima through optimization techniques
Federated optimizer
An optimization framework designed to address heterogeneity in federated learning across distributed networks
Federated learner
An implementation of a federated learning algorithm for handling heterogeneous data
Federated learner
An implementation of a federated learning algorithm for optimization problems with compositional pairwise risk optimization.
Federated learner optimizer
A tool for training federated learning models with adaptive gradient balancing to handle class imbalance in multi-client scenarios.
Federated learning library
An implementation of federated learning with prototype-based methods across heterogeneous clients
Federated Learning Algorithm
This project presents an approach to federated learning with partial client participation by optimizing anchor selection for improving model accuracy and convergence.
Heterogeneity fixer
Combating heterogeneity in federated learning by combining adversarial training with client-wise slack during aggregation
Federated learning algorithm
An implementation of Bayesian network structure learning with continuous optimization for federated learning.
Federated learning optimizer
An algorithm for distributed learning with flexible model customization during training and testing
Federated Learning Optimizer
An implementation of federated learning optimized for training on renewable energy sources and spare compute capacity to minimize carbon emissions.
Federated learning framework
Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions.
Federated Learner
An implementation of a federated learning algorithm that generalizes to out-of-distribution scenarios using implicit invariant relationships