FedBABU

Image classification algorithms

An implementation of federated learning algorithms for image classification on multiple architectures.

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

GitHub

50 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 An implementation of federated learning with bilateral curation for partially class-disjoint data 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 10
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 366
zhenqincn/fedapen An implementation of cross-silo federated learning with adaptability to statistical heterogeneity 12
charliedinh/pfedme An implementation of Personalized Federated Learning with Moreau Envelopes and related algorithms using PyTorch for research and experimentation. 290
yuetan031/fedproto An implementation of federated learning with prototype-based methods across heterogeneous clients 132
conditionwang/fcil This implementation provides a framework for federated class-incremental learning on visual tasks. 102
omarfoq/fedem Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions. 155
hongyouc/fed-rod Develops a framework to balance competing goals in federated learning by decoupling generic and personalized prediction tasks. 14