Proceedings

A Survey on Efficient Federated Learning Methods for Foundation Model Training

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Survey Track. Pages 8317-8325.

https://doi.org/10.24963/ijcai.2024/919

Abstract

Federated Learning (FL) has become an established technique to facilitate privacy-preserving collaborative training across a multitude of clients. However, new approaches to FL often discuss their contributions involving small deep-learning models only and focus on training full models on clients. In the wake of Foundation Models (FM), the reality is different for many deep learning applications. Typically, FMs have already been pre-trained across a wide variety of tasks and can be fine-tuned to specific downstream tasks over significantly smaller datasets than required for full model training. However, access to such datasets is often challenging. By its design, FL can help to open data silos. With this survey, we introduce a novel taxonomy focused on computational and communication efficiency, the vital elements to make use of FMs in FL systems. We discuss the benefits and drawbacks of parameter-efficient fine-tuning (PEFT) for FL applications, elaborate on the readiness of FL frameworks to work with FMs and provide future research opportunities on how to evaluate generative models in FL as well as the interplay of privacy and PEFT.

  1. [Aledhari et al., 2020] Mohammed Aledhari, Rehma Razzak, Reza M. Parizi, and Fahad Saeed. Federated learning: A survey on enabling technologies, protocols, and applications. IEEE Access, 8:140699-140725, 2020.
  2. [Alistarh et al., 2017] Dan Alistarh, Demjan Grubic, et al. Qsgd: Communication-efficient sgd via gradient quantization and en- coding. In Advances in Neural Information Processing Systems, volume 30, 2017.
  3. [Babakniya et al., 2023] Sara Babakniya, Ahmed R. Elkordy, et al. Slora: Federated parameter efficient fine-tuning of language models, 2023.
  4. [Banabilah et al., 2022] Syreen Banabilah, Moayad Aloqaily, et al. Federated learning review: Fundamentals, enabling technologies, and future applications. Information Processing & Management, 59(6), 2022.
  5. [Beutel et al., 2020] Daniel J. Beutel, Taner Topal, et al. Flower: A friendly federated learning research framework, 2020.
  6. [Bommasani et al., 2021] Rishi Bommasani, Drew A. Hudson, et al. On the opportunities and risks of foundation models, 2021.
  7. [Chen et al., 2023] Yuanyuan Chen, Zichen Chen, et al. Fedobd: Opportunistic block dropout for efficiently training large-scale neural networks through federated learning. In Proceedings of the Thirty-Second International Joint Conference on Artificial In- telligence, IJCAI-23. IJCAI Org., 2023. IJCAI
  8. [Diao et al., 2020] Enmao Diao, Jie Ding, et al. Heterofl: Compu- tation and communication efficient federated learning for hetero- geneous clients, 2020.
  9. [Ding et al., 2023] Ning Ding, Yujia Qin, et al. Parameter-efficient fine-tuning of large-scale pre-trained language models. Nature Machine Intelligence, 5(3):220-235, March 2023.
  10. [Dosovitskiy et al., 2020] Alexey Dosovitskiy, Lucas Beyer, et al. An image is worth 16x16 words: Transformers for image recog- nition at scale, 2020.
  11. [Erben et al., 2023] Alexander Erben, Ruben Mayer, and Hans- Arno Jacobsen. How can we train deep learning models across clouds and continents? an experimental study, 2023.
  12. [Fan et al., 2023] Tao Fan, Yan Kang, et al. Fate-llm: A industrial grade federated learning framework for large language models, 2023.
  13. [Foley et al., 2022] Patrick Foley, Micah J Sheller, et al. Openfl: the open federated learning library. Physics in Medicine & Biology, 67(21), 2022.
  14. [Frankle and Carbin, 2019] Jonathan Frankle and Michael Carbin. The lottery ticket hypothesis: Finding sparse, trainable neural networks. In ICLR, 2019.
  15. [Galtier and Marini, 2019] Mathieu N Galtier and Camille Marini. Substra: a framework for privacy-preserving, traceable and col- laborative machine learning, 2019.
  16. Inc. Google. Tensorflow federated: Machine learn- ing on decentralized data. https://www.tensorflow.org/federated, 2019. Accessed: 2023-12-01. Link
  17. [Habib et al., 2023] Gousia Habib, Tausifa Jan Saleem, and Brejesh Lall. Knowledge distillation in vision transformers: A critical review, 2023.
  18. [He et al., 2020] Chaoyang He, Songze Li, et al. Fedml: A research library and benchmark for federated machine learning, 2020.
  19. [He et al., 2021] Haoyu He, Xingjian Shi, et al. Distiller: A sys- tematic study of model distillation methods in natural language processing, 2021.
  20. [Horváth et al., 2021] Samuel Horváth, Stefanos Laskaridis, et al. FjORD: Fair and accurate federated learning under heteroge- neous targets with ordered dropout. In Advances in Neural In- formation Processing Systems, 2021.
  21. [Houlsby et al., 2019] Neil Houlsby, Andrei Giurgiu, et al. Parameter-efficient transfer learning for nlp, 2019.
  22. [Hu et al., 2021] Edward J. Hu, Yelong Shen, et al. Lora: Low-rank adaptation of large language models, 2021.
  23. [Huang et al., 2022] Hong Huang, Lan Zhang, et al. Distributed pruning towards tiny neural networks in federated learning, 2022.
  24. [Isik et al., 2022] Berivan Isik, Francesco Pase, et al. Sparse ran- dom networks for communication-efficient federated learning, 2022.
  25. [Jia et al., 2022] Menglin Jia, Luming Tang, et al. Visual prompt tuning, 2022.
  26. [Jiang et al., 2022] Yuang Jiang, Shiqiang Wang, et al. Model prun- ing enables efficient federated learning on edge devices. IEEE Transactions on Neural Networks and Learning Systems, 2022.
  27. [Kuang et al., 2023] Weirui Kuang, Bingchen Qian, et al. Federatedscope-llm: A comprehensive package for fine-tuning large language models in federated learning, 2023.
  28. [Lagunas et al., 2021] Francois Lagunas, Ella Charlaix, et al. Block pruning for faster transformers. In Proceedings of the 2021 Con- ference on EMNLP, Online and Punta Cana, Dominican Repub- lic, November 2021. ACL.
  29. [Lester et al., 2021] Brian Lester, Rami Al-Rfou, and Noah Con- stant. The power of scale for parameter-efficient prompt tuning, 2021.
  30. [Li et al., 2020] Ang Li, Jingwei Sun, et al. Lotteryfl: Personal- ized and communication-efficient federated learning with lottery ticket hypothesis on non-iid datasets, 2020.
  31. [Li et al., 2022] Zhize Li, Haoyu Zhao, et al. Soteriafl: A uni- fied framework for private federated learning with communica- tion compression. In Advances in Neural Information Processing Systems, volume 35, 2022.
  32. [Li et al., 2023] Qinbin Li, Zeyi Wen, Zhaomin Wu, Sixu Hu, Naibo Wang, Yuan Li, Xu Liu, and Bingsheng He. A survey on federated learning systems: Vision, hype and reality for data privacy and protection. IEEE Transactions on Knowledge and Data Engineering, 35(4):3347-3366, April 2023.
  33. [Liu et al., 2021] Yang Liu, Tao Fan, et al. Fate: An industrial grade platform for collaborative learning with data protection. JMLR, 22(1), 2021.
  34. [Liu et al., 2022] Ji Liu, Jizhou Huang, et al. From distributed ma- chine learning to federated learning: a survey. Knowledge and Information Systems, 64(4), 2022.
  35. [Longpre et al., 2023] Shayne Longpre, Le Hou, et al. The flan col- lection: Designing data and methods for effective instruction tun- ing, 2023.
  36. [Lu et al., 2023] Wang Lu, Xixu Hu, et al. Fedclip: Fast general- ization and personalization for clip in federated learning, 2023.
  37. [Ludwig et al., 2020] Heiko Ludwig, Nathalie Baracaldo, et al. Ibm federated learning: an enterprise framework white paper v0.1, 2020.
  38. [McMahan et al., 2017] Brendan McMahan, Eider Moore, et al. Communication-Efficient Learning of Deep Networks from De- centralized Data. In Proceedings of the 20th International Con- ference on Artificial Intelligence and Statistics, volume 54 of Pro- ceedings of Machine Learning Research. PMLR, 2017.
  39. [Mishchenko et al., 2019] Konstantin Mishchenko, Eduard Gor- bunov, et al. Distributed learning with compressed gradient dif- ferences, 2019.
  40. [Mitra et al., 2021] Aritra Mitra, Rayana Jaafar, George J. Pappas, and Hamed Hassani. Linear convergence in federated learning: Tackling client heterogeneity and sparse gradients. In Advances in Neural Information Processing Systems, 2021.
  41. [Nguyen et al., 2021] Dinh C. Nguyen, Ming Ding, et al. Federated learning for internet of things: A comprehensive survey. IEEE Communications Surveys & Tutorials, 23(3), 2021.
  42. [OpenAI, 2023] OpenAI. GPT-4 technical report, 2023.
  43. [Penedo et al., 2023] Guilherme Penedo, Quentin Malartic, et al. The refinedweb dataset for falcon llm: Outperforming curated corpora with web data, and web data only, 2023.
  44. [Peng et al., 2020] Xingchao Peng, Zijun Huang, et al. Federated adversarial domain adaptation. In ICLR, 2020.
  45. [Perez et al., 2020] Andres Perez, Valentina Sanguineti, et al. Audio-visual model distillation using acoustic images. In Pro- ceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV), 2020.
  46. [Qin et al., 2023] Zhen Qin, Daoyuan Chen, et al. Federated full- parameter tuning of billion-sized language models with commu- nication cost under 18 kilobytes, 2023.
  47. [Radford et al., 2019] Alec Radford, Jeff Wu, et al. Language mod- els are unsupervised multitask learners. 2019.
  48. [Reisizadeh et al., 2020] Amirhossein Reisizadeh, Aryan Mokhtari, et al. Fedpaq: A communication-efficient feder- ated learning method with periodic averaging and quantization. In Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, volume 108. PMLR, 2020.
  49. [Roth et al., 2022] Holger R. Roth, Yan Cheng, et al. Nvidia flare: Federated learning from simulation to real-world. 2022.
  50. [Ryabinin et al., 2023] Max Ryabinin, Tim Dettmers, et al. Swarm parallelism: Training large models can be surprisingly communication-efficient, 2023.
  51. [Sanh et al., 2020] Victor Sanh, Thomas Wolf, and Alexander Rush. Movement pruning: Adaptive sparsity by fine-tuning. In Advances in Neural Information Processing Systems, volume 33.
  52. Curran Associates, Inc., 2020.
  53. [Sun et al., 2023] Guangyu Sun, Matias Mendieta, et al. Explor- ing parameter-efficient fine-tuning for improving communication efficiency in federated learning, 2023.
  54. [Taori et al., 2023] Rohan Taori, Ishaan Gulrajani, et al. Stanford alpaca: An instruction-following llama model. https://github. com/tatsu-lab/stanford alpaca, 2023. Link
  55. [Tian et al., 2022] Yuanyishu Tian, Yao Wan, et al. Fedbert: When federated learning meets pre-training. ACM Transactions on In- telligent Systems and Technology, 13(4), 2022.
  56. [Woisetschläger et al., 2023] Herbert Woisetschläger, Alexander Erben, et al. Federated fine-tuning of llms on the very edge: The good, the bad, the ugly, 2023.
  57. [Xie et al., 2023] Yuexiang Xie, Zhen Wang, et al. Federatedscope: A flexible federated learning platform for heterogeneity. Pro- ceedings of the VLDB Endowment, 16(5), 2023.
  58. [Xu et al., 2023] Zheng Xu, Yanxiang Zhang, et al. Federated learning of gboard language models with differential privacy. 2023.
  59. [Yang et al., 2023] Dongchao Yang, Jinchuan Tian, et al. Uniaudio: An audio foundation model toward universal audio generation, 2023.
  60. [Yang et al., 2024] Yiyuan Yang, Guodong Long, Taoshu Shen, Jing Jiang, and Michael Blumenstein. Dual-personalizing adapter for federated foundation models. ArXiv, abs/2403.19211, 2024.
  61. [Yousefpour et al., 2023] Ashkan Yousefpour, Shen Guo, et al. Green federated learning, 2023.
  62. [Yu et al., 2023] Sixing Yu, J. Pablo Muñoz, and Ali Jannesari. Federated foundation models: Privacy-preserving and collabo- rative learning for large models, 2023.
  63. [Zaken et al., 2021] Elad Ben Zaken, Shauli Ravfogel, and Yoav Goldberg. Bitfit: Simple parameter-efficient fine-tuning for transformer-based masked language-models, 2021.
  64. [Zhang et al., 2021] Chen Zhang, Yu Xie, et al. A survey on feder- ated learning. Knowledge-Based Systems, 216, 2021.
  65. [Zhang et al., 2023a] Jianyi Zhang, Saeed Vahidian, et al. Towards building the federated gpt: Federated instruction tuning, 2023.
  66. [Zhang et al., 2023b] Zhuo Zhang, Yuanhang Yang, et al. Fed- PETuning: When federated learning meets the parameter- efficient tuning methods of pre-trained language models. In Findings of the Association for Computational Linguistics: ACL 2023. ACL, 2023.
  67. [Zhao et al., 2022] Haodong Zhao, Wei Du, et al. Fedprompt: Communication-efficient and privacy preserving prompt tuning in federated learning, 2022.
  68. [Zheng et al., 2023] Rui Zheng, Shihan Dou, et al. Secrets of rlhf in large language models part i: Ppo, 2023.
  69. [Zhou et al., 2018] Yiren Zhou, Seyed Moosavi-Dezfooli, et al. Adaptive quantization for deep neural network. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1), 2018.
  70. [Zhu and Gupta, 2017] Michael Zhu and Suyog Gupta. To prune, or not to prune: exploring the efficacy of pruning for model com- pression, 2017.
  71. [Zhu et al., 2019] Maohua Zhu, Tao Zhang, et al. Sparse tensor core: Algorithm and hardware co-design for vector-wise sparse neural networks on modern gpus. In Proceedings of the 52nd An- nual IEEE/ACM International Symposium on Microarchitecture, MICRO '52. ACM, 2019.
  72. [Zhuang et al., 2023] Weiming Zhuang, Chen Chen, and Lingjuan Lyu. When foundation model meets federated learning: Motiva- tions, challenges, and future directions, 2023.
  73. [Ziller et al., 2021] Alexander Ziller, Andrew Trask, et al. PySyft: A Library for Easy Federated Learning. Springer International Publishing, 2021.

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

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