Fast and Differentiable Geometric Median in PyTorch and NumPy. Install as
pip install geom-median.
A toolbox of selected optimization algorithms for unstructured tasks such as binary classification, and structured prediction tasks such as visual object localization and named entity recognition.
RFA TensorFlow Federated Implementation of robust aggregation for federated learning using the geometric median.
Code to reproduce results from papers
Federated Learning with Partial Personalization: PyTorch implementation of various personalized federated learning algorithms and experiments on text, vision, and speech data. Reproduce results from this ICML 2022 paper.
RFA and PyTorch port tRFA: Implementation of RFA, a robust aggregation algorithm for federated learning, in simulation. Reproduce results from this paper published in IEEE Transactions on Signal Processing 2022.