FedJAX: Federated learning simulation with JAX

4 August 2021

Papers citing "FedJAX: Federated learning simulation with JAX"

31 / 31 papers shown

Title
Decoding FL Defenses: Systemization, Pitfalls, and Remedies M. A. Khan Virat Shejwalkar Yasra Chandio Amir Houmansadr Fatima M. Anwar AAML 38 0 0 03 Feb 2025
Parameter-Efficient Transfer Learning under Federated Learning for Automatic Speech Recognition Xuan Kan Yonghui Xiao Tien-Ju Yang Nanxin Chen Rajiv Mathews FedML 21 0 0 19 Aug 2024
One-Shot Federated Learning with Bayesian Pseudocoresets Tim d'Hondt Mykola Pechenizkiy Robert Peharz FedML 21 0 0 04 Jun 2024
Efficient Language Model Architectures for Differentially Private Federated Learning Jae Hun Ro Srinadh Bhojanapalli Zheng Xu Yanxiang Zhang A. Suresh FedML 31 2 0 12 Mar 2024
Training Machine Learning models at the Edge: A Survey Aymen Rayane Khouas Mohamed Reda Bouadjenek Hakim Hacid Sunil Aryal 24 9 0 05 Mar 2024
How to Privately Tune Hyperparameters in Federated Learning? Insights from a Benchmark Study Natalija Mitic Apostolos Pyrgelis Sinem Sav FedML 47 1 0 25 Feb 2024
Heterogeneous Federated Learning Using Knowledge Codistillation Jared Lichtarge Ehsan Amid Shankar Kumar Tien-Ju Yang Rohan Anil Rajiv Mathews FedML 18 0 0 04 Oct 2023
Window-based Model Averaging Improves Generalization in Heterogeneous Federated Learning Debora Caldarola Barbara Caputo Marco Ciccone FedML 13 7 0 02 Oct 2023
Federated Deep Equilibrium Learning: A Compact Shared Representation for Edge Communication Efficiency Long Tan Le Tuan Dung Nguyen Tung-Anh Nguyen Choong Seon Hong Nguyen H. Tran FedML 19 0 0 27 Sep 2023
A Comprehensive Empirical Study of Bugs in Open-Source Federated Learning Frameworks Weijie Shao Yuyang Gao Fu Song Sen Chen Lingling Fan JingZhu He FedML 16 0 0 09 Aug 2023
Towards Federated Foundation Models: Scalable Dataset Pipelines for Group-Structured Learning Zachary B. Charles Nicole Mitchell Krishna Pillutla Michael Reneer Zachary Garrett FedML AI4CE 23 28 0 18 Jul 2023
Explicit Personalization and Local Training: Double Communication Acceleration in Federated Learning Kai Yi Laurent Condat Peter Richtárik FedML 24 5 0 22 May 2023
Flame: Simplifying Topology Extension in Federated Learning Harshit Daga Jae-Kwang Shin D. Garg Ada Gavrilovska Myungjin Lee Ramana Rao Kompella AI4CE 18 10 0 09 May 2023
JaxPruner: A concise library for sparsity research Jooyoung Lee Wonpyo Park Nicole Mitchell Jonathan Pilault J. Obando-Ceron ... Hong-Seok Kim Yann N. Dauphin Karolina Dziugaite P. S. Castro Utku Evci 31 13 0 27 Apr 2023
Federated Learning as Variational Inference: A Scalable Expectation Propagation Approach Han Guo P. Greengard Hongyi Wang Andrew Gelman Yoon Kim Eric P. Xing FedML 13 20 0 08 Feb 2023
Federated Automatic Differentiation Keith Rush Zachary B. Charles Zachary Garrett FedML 32 1 0 18 Jan 2023
Encrypted machine learning of molecular quantum properties J. Weinreich Guido Falk von Rudorff O. Anatole von Lilienfeld 26 1 0 05 Dec 2022
Federated Hypergradient Descent A. K. Kan FedML 11 3 0 03 Nov 2022
Rieoptax: Riemannian Optimization in JAX Saiteja Utpala Andi Han Pratik Jawanpuria Bamdev Mishra 8 3 0 10 Oct 2022
Protea: Client Profiling within Federated Systems using Flower Wanru Zhao Xinchi Qiu Javier Fernandez-Marques Pedro Porto Buarque de Gusmão Nicholas D. Lane 8 6 0 03 Jul 2022
Towards Fair Federated Recommendation Learning: Characterizing the Inter-Dependence of System and Data Heterogeneity Kiwan Maeng Haiyu Lu Luca Melis John Nguyen Michael G. Rabbat Carole-Jean Wu FedML 24 30 0 30 May 2022
Scaling Language Model Size in Cross-Device Federated Learning Jae Hun Ro Theresa Breiner Lara McConnaughey Mingqing Chen A. Suresh Shankar Kumar Rajiv Mathews FedML 21 24 0 31 Mar 2022
Improving Generalization in Federated Learning by Seeking Flat Minima Debora Caldarola Barbara Caputo Marco Ciccone FedML 14 109 0 22 Mar 2022
Correlated quantization for distributed mean estimation and optimization A. Suresh Ziteng Sun Jae Hun Ro Felix X. Yu 15 12 0 09 Mar 2022
FedLite: A Scalable Approach for Federated Learning on Resource-constrained Clients Jianyu Wang Qi A. S. Rawat Sashank J. Reddi Sagar M. Waghmare Felix X. Yu Gauri Joshi FedML 11 22 0 28 Jan 2022
FLIX: A Simple and Communication-Efficient Alternative to Local Methods in Federated Learning Elnur Gasanov Ahmed Khaled Samuel Horváth Peter Richtárik FedML 14 16 0 22 Nov 2021
A Field Guide to Federated Optimization Jianyu Wang Zachary B. Charles Zheng Xu Gauri Joshi H. B. McMahan ... Mi Zhang Tong Zhang Chunxiang Zheng Chen Zhu Wennan Zhu FedML 167 410 0 14 Jul 2021
FedScale: Benchmarking Model and System Performance of Federated Learning at Scale Fan Lai Yinwei Dai Sanjay Sri Vallabh Singapuram Jiachen Liu Xiangfeng Zhu H. Madhyastha Mosharaf Chowdhury FedML 27 194 0 24 May 2021
Communication-Efficient Agnostic Federated Averaging Jae Hun Ro Mingqing Chen Rajiv Mathews M. Mohri A. Suresh FedML 9 17 0 06 Apr 2021
FedML: A Research Library and Benchmark for Federated Machine Learning Chaoyang He Songze Li Jinhyun So Xiao Zeng Mi Zhang ... Yang Liu Ramesh Raskar Qiang Yang M. Annavaram Salman Avestimehr FedML 156 553 0 27 Jul 2020
IBM Federated Learning: an Enterprise Framework White Paper V0.1 Heiko Ludwig Nathalie Baracaldo Gegi Thomas Yi Zhou Ali Anwar ... Sean Laguna Mikhail Yurochkin Mayank Agarwal Ebube Chuba Annie Abay FedML 124 155 0 22 Jul 2020