FedBABU

FL algorithm

An implementation of federated learning for image classification tasks

ICLR 2022, "FedBABU: Toward Enhanced Representation for Federated Image Classification"

GitHub

51 stars
2 watching
11 forks
Language: Python
last commit: over 2 years ago
federated-learningmachine-learningpytorch

Related projects:

Repository Description Stars
mediabrain-sjtu/pfedgraph This project enables personalized federated learning with inferred collaboration graphs to improve the performance of machine learning models on non-IID (non-independent and identically distributed) datasets. 26
dawenzi098/sfl-structural-federated-learning A Python implementation of Personalized Federated Learning with Graph using PyTorch. 50
jinheonbaek/fed-pub Personalized Subgraph Federated Learning framework for distributed machine learning 44
shenzebang/federated-learning-pytorch A PyTorch-based framework for Federated Learning experiments 40
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
mediabrain-sjtu/fedgela Federated learning algorithm designed to handle partially class-disjoint data by utilizing bilateral curation and Dirichlet partitioning. 10
tfzhou/fedfa An ICLR 2023 paper implementation in PyTorch of Federated Feature Augmentation for federated learning with data augmentation and medical image analysis. 57
yamingguo98/fediir An implementation of a federated learning algorithm that generalizes to out-of-distribution scenarios using implicit invariant relationships 9
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 365
zhenqincn/fedapen An implementation of cross-silo federated learning with adaptability to statistical heterogeneity 10
charliedinh/pfedme An implementation of Personalized Federated Learning with Moreau Envelopes and related algorithms using PyTorch for research and experimentation. 289
yuetan031/fedproto An implementation of federated learning with prototype-based methods across heterogeneous clients 133
conditionwang/fcil Implementation of Federated Class-Incremental Learning for Continual Learning in Computer Vision 101
omarfoq/fedem Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions. 154
hongyouc/fed-rod Develops a framework to balance competing goals in federated learning by decoupling generic and personalized prediction tasks. 14