Proceedings

Federated Learning with Fair Averaging

Download BibTeX Cite

Proceedings of the Thirtieth International Joint Conference on Artificial Intelligence. Main Track. Pages 1615-1623.

https://doi.org/10.24963/ijcai.2021/223

Abstract

Fairness has emerged as a critical problem in federated learning (FL). In this work, we identify a cause of unfairness in FL -- conflicting gradients with large differences in the magnitudes. To address this issue, we propose the federated fair averaging (FedFV) algorithm to mitigate potential conflicts among clients before averaging their gradients. We first use the cosine similarity to detect gradient conflicts, and then iteratively eliminate such conflicts by modifying both the direction and the magnitude of the gradients. We further show the theoretical foundation of FedFV to mitigate the issue conflicting gradients and converge to Pareto stationary solutions. Extensive experiments on a suite of federated datasets confirm that FedFV compares favorably against state-of-the-art methods in terms of fairness, accuracy and efficiency. The source code is available at https://github.com/WwZzz/easyFL.

  1. [Abay et al., 2020] Annie Abay, Yi Zhou, Nathalie Bara- caldo, Shashank Rajamoni, Ebube Chuba, and Heiko Lud- wig. Mitigating bias in federated learning. arXiv e-prints, page arXiv:2012.02447, December 2020.
  2. [Chaudhry et al., 2018] Arslan Chaudhry, Marc'Aurelio Ranzato, Marcus Rohrbach, and Mohamed Elho- seiny. Efficient lifelong learning with A-GEM. CoRR, abs/1812.00420, 2018.
  3. [Cho et al., 2020] Yae Jee Cho, Samarth Gupta, Gauri Joshi, and Osman Yagan. Bandit-based communication-efficient client selection strategies for federated learning. arXiv e- prints, page arXiv:2012.08009, 2020.
  4. [Désidéri, 2012] Jean-Antoine Désidéri. Multiple-gradient descent algorithm (mgda) for multiobjective optimization. Comptes Rendus Mathematique, 350(5):313 -318, 2012.
  5. [Farajtabar et al., 2019] Mehrdad Farajtabar, Navid Azizan, Alex Mott, and Ang Li. Orthogonal gradient descent for continual learning. CoRR, abs/1910.07104, 2019.
  6. [Guo et al., 2019] Yunhui Guo, Mingrui Liu, Tianbao Yang, and Tajana Rosing. Learning with long-term remem- bering: Following the lead of mixed stochastic gradient. CoRR, abs/1909.11763, 2019.
  7. [Hu et al., 2020] Zeou Hu, Kiarash Shaloudegi, Guojun Zhang, and Yaoliang Yu. Fedmgda+: Federated learning meets multi-objective optimization. arXiv e-prints, page arXiv:2006.11489, 2020.
  8. [Huang et al., 2020] Wei Huang, Tianrui Li, Dexian Wang, Shengdong Du, and Junbo Zhang. Fairness and ac- curacy in federated learning. arXiv e-prints, page arXiv:2012.10069, 2020.
  9. [Kairouz et al., 2019] Peter Kairouz, H. Brendan McMahan, Brendan Avent, Aurélien Bellet, Mehdi Bennis, and et al. Advances and open problems in federated learning. CoRR, abs/1912.04977, 2019.
  10. [Krizhevsky, 2012] Alex Krizhevsky. Learning multiple lay- ers of features from tiny images. University of Toronto, 05 2012.
  11. [LeCun et al., 1998] Yann LeCun, Léon Bottou, Yoshua Bengio, and Patrick Haffner. Gradient-based learning ap- plied to document recognition. Proceedings of the IEEE, 86(11):2278-2324, 1998.
  12. [Li et al., 2019] Tian Li, Maziar Sanjabi, and Virginia Smith. Fair resource allocation in federated learning. CoRR, abs/1905.10497, 2019.
  13. [Li et al., 2020a] Tian Li, Shengyuan Hu, Ahmad Beirami, and Virginia Smith. Federated multi-task learning for com- peting constraints. arXiv e-prints, page arXiv:2012.04221, 2020.
  14. [Li et al., 2020b] Tian Li, Anit Kumar Sahu, Manzil Zaheer, Maziar Sanjabi, Ameet Talwalkar, and Virginia Smith. Federated optimization in heterogeneous networks. arXiv e-prints, page arXiv:1812.06127, 2020.
  15. [Li et al., 2020c] Xiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang, and Zhihua Zhang. On the conver- gence of fedavg on non-iid data. arXiv e-prints, page arXiv:1907.02189, 2020.
  16. [Lopez-Paz and Ranzato, 2017] David Lopez-Paz and Marc'Aurelio Ranzato. Gradient episodic memory for continuum learning. CoRR, abs/1706.08840, 2017.
  17. [Lyu et al., 2019] Lingjuan Lyu, Jiangshan Yu, Karthik Nan- dakumar, Yitong Li, Xingjun Ma, and Jiong Jin. Towards fair and decentralized privacy-preserving deep learning with blockchain. CoRR, abs/1906.01167, 2019.
  18. [Lyu et al., 2020] Lingjuan Lyu, Xinyi Xu, and Qian Wang. Collaborative fairness in federated learning. arXiv e- prints, page arXiv:2008.12161, 2020.
  19. [McMahan et al., 2017] Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, and Blaise Aguera y Ar- cas. Communication-Efficient Learning of Deep Networks from Decentralized Data. In Aarti Singh and Jerry Zhu, editors, Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, volume 54 of Pro- ceedings of Machine Learning Research, pages 1273- 1282, Fort Lauderdale, FL, USA, 20-22 Apr 2017. PMLR.
  20. [Mohri et al., 2019] Mehryar Mohri, Gary Sivek, and Ananda Theertha Suresh. Agnostic federated learning. CoRR, abs/1902.00146, 2019.
  21. [Xiao et al., 2017] Han Xiao, Kashif Rasul, and Roland Vollgraf. Fashion-mnist: a novel image dataset for benchmarking machine learning algorithms. CoRR, abs/1708.07747, 2017.
  22. [Xu and Lyu, 2020] Xinyi Xu and Lingjuan Lyu. Towards building a robust and fair federated learning system. arXiv e-prints, page arXiv:2011.10464, 2020.
  23. [Yu et al., 2020] Tianhe Yu, Saurabh Kumar, Abhishek Gupta, Sergey Levine, Karol Hausman, and Chelsea Finn. Gradient Surgery for Multi-Task Learning. arXiv e-prints, page arXiv:2001.06782, January 2020.
  24. [Zhao et al., 2018] Yue Zhao, Meng Li, Liangzhen Lai, Naveen Suda, Damon Civin, and Vikas Chandra. Feder- ated Learning with Non-IID Data. arXiv e-prints, page arXiv:1806.00582, June 2018.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards