knn-per

Personalized FL framework

A federated learning framework with personalized memorization using deep neural networks and k-nearest neighbors for collaborative learning of statistical models

Official code for "Personalized Federated Learning through Local Memorization" (ICML'22)

GitHub

43 stars
2 watching
15 forks
Language: Python
last commit: over 1 year ago
deep-learningfederated-learningmachine-learningpersonalized-federated-learningpytorch

Related projects:

Repository Description Stars
omarfoq/fedem Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions. 157
krishnap25/fl_partial_personalization A framework for federated learning with partial model personalization 2
jinheonbaek/fed-pub Personalized Subgraph Federated Learning framework for distributed machine learning 45
substra/substra Enables the training and validation of machine learning models on distributed datasets in a secure and scalable manner. 274
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
litian96/ditto A framework for personalized federated learning to balance fairness and robustness in decentralized machine learning systems. 138
wenkehuang/fccl A framework for tackling heterogeneity and catastrophic forgetting in federated learning by leveraging cross-correlation and similarity learning 97
omarfoq/communication-in-cross-silo-fl A toolkit for optimizing federated learning in cross-silo settings by designing efficient communication topologies 30
kai-yue/ntk-fed A framework for federated learning that leverages the neural tangent kernel to address statistical heterogeneity in distributed machine learning. 3
codepothunter/fednp A framework for non-IID federated learning via neural propagation 6
dawenzi098/sfl-structural-federated-learning A Python implementation of Personalized Federated Learning with Graph using PyTorch. 49
ganyuwang/vfl-czofo A unified framework for improving privacy and reducing communication overhead in distributed machine learning models. 12
rong-dai/dispfl An implementation of a personalized federated learning framework with decentralized sparse training and peer-to-peer communication protocol. 72
hongyouc/fed-rod Develops a framework to balance competing goals in federated learning by decoupling generic and personalized prediction tasks. 14
astuary/flow A framework for personalized federated learning that creates dynamic models for each input instance and improves generalizability of global models. 9