Proceedings

FediLoRA: Practical Federated Fine-Tuning of Foundation Models Under Missing-Modality Constraints

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Health. Pages 6966-6974.

https://doi.org/10.24963/ijcai.2026/775

Abstract

Federated Learning with LoRA fine-tuning offers an efficient and privacy-aware solution for institutions to collaboratively leverage their large datasets to train VLLMs. However, participating institutions often possess heterogeneous computational resources, resulting in imbalanced LoRA ranks, which pose a major challenge for effective collaboration. In addition, real-world applications in domains such as healthcare and transportation frequently suffer from missing modalities due to user mistakes or device failures, which significantly degrade global model performance in federated settings. To the best of our knowledge, no prior work has addressed these two challenges simultaneously in federated VLLMs. To tackle these issues, we propose FediLoRA, a lightweight federated LoRA aggregation framework that effectively mitigates the impact of missing modalities in heterogeneous environment. FediLoRA is explicitly motivated by the observation that simple averaging and structured editing can jointly benefit both global and personalized models. Our approach achieves strong performance across multiple general-domain and medical-domain benchmark datasets. Additional experiments on healthcare data further demonstrate that FediLoRA is well-suited for practical, real-world deployment scenarios. Our code is released at https://github.com/gotobcn8/FediLoRA.

  1. [Bai et al., 2024] Jiamu Bai, Daoyuan Chen, Bingchen Qian, Liuyi Yao, and Yaliang Li. Federated fine-tuning of large language models under heterogeneous tasks and client re- sources. In Proceedings of the NeurIPS, 2024, 2024.
  2. [Bao et al., 2024] Guangyin Bao, Qi Zhang, Duoqian Miao, Zixuan Gong, Liang Hu, Ke Liu, Yang Liu, and Chongyang Shi. Multimodal federated learning with miss- ing modality via prototype mask and contrast, 2024.
  3. [Bian et al., 2025] Jieming Bian, Lei Wang, Letian Zhang, and Jie Xu. Lora-fair: Federated lora fine-tuning with ag- gregation and initialization refinement. In Proceedings of the CVPR, 2025.
  4. [Byun and Lee, 2025] Yuji Byun and Jaeho Lee. Towards federated low-rank adaptation of language models with rank heterogeneity. In Proceedings of the 2025 NAACL Short Papers, 2025.
  5. [Chen and Li, 2022] Sijia Chen and Baochun Li. Towards optimal multi-modal federated learning on non-iid data with hierarchical gradient blending. In IEEE INFOCOM, 2022, 2022.
  6. [Chen et al., 2023] Junsong Chen, Jincheng Yu, Chongjian Ge, Lewei Yao, Enze Xie, Yue Wu, Zhongdao Wang, James Kwok, Ping Luo, Huchuan Lu, and Zhenguo Li. Pixart-α: Fast training of diffusion transformer for photo- realistic text-to-image synthesis, 2023.
  7. [Cho et al., 2024] Yae Jee Cho, Luyang Liu, Zheng Xu, Aldi Fahrezi, and Gauri Joshi. Heterogeneous LoRA for Fed- erated Fine-tuning of On-Device Foundation Models. In Proceedings of the 2024 EMNLP), 2024.
  8. [Chu et al., 2025] Xiaoli Chu, Yiheng Ye, Siqiao Tang, Miaoru Han, Guowei Wang, Shuai Lin, Bingzhen Sun, Qingchun Huang, Yan Zhang, Xiaodong Chu, et al. Per- sonalized medication for chronic diseases using multi- modal data-driven chain-of-decisions. Advanced Science, 2025.
  9. [Cismondi et al., 2013] Federico Cismondi, André S Fialho, Susana M Vieira, Shane R Reti, João MC Sousa, and Stan N Finkelstein. Missing data in medical databases: Im- pute, delete or classify? Artificial intelligence in medicine, 2013.
  10. [Collins et al., 2022] Liam Collins, Hamed Hassani, Aryan Mokhtari, and Sanjay Shakkottai. Fedavg with fine tuning: Local updates lead to representation learning. Proceedings of the NeurIPS, 2022, 2022.
  11. [De Zarzà et al., 2023] I De Zarzà, J De Curtò, Gemma Roig, and Carlos T Calafate. Optimized financial plan- ning: integrating individual and cooperative budgeting models with llm recommendations. AI, 2023. Feng et al., 2023]
  12. Tiantian Feng, Digbalay Bose, Tuo Zhang, Rajat Hebbar, Anil Ramakrishna, Rahul Gupta, Mi Zhang, Salman Avestimehr, and Shrikanth Narayanan. Fedmultimodal: A benchmark for multimodal federated learning. In Proceedings of the 29th KDD, KDD 2023, Long Beach, CA, USA, August 6-10, 2023, 2023.
  13. [Guo et al., 2024] Pengxin Guo, Shuang Zeng, Yanran Wang, Huijie Fan, Feifei Wang, and Liangqiong Qu. Se- lective aggregation for low-rank adaptation in federated learning. arXiv preprint arXiv:2410.01463, 2024.
  14. [Houlsby et al., 2019] Neil Houlsby, Andrei Giurgiu, Stanis- law Jastrzebski, Bruna Morrone, Quentin De Laroussilhe, Andrea Gesmundo, Mona Attariyan, and Sylvain Gelly. Parameter-efficient transfer learning for nlp. In ICML, 2019.
  15. [Hu et al., 2022a] Edward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, and Weizhu Chen. Lora: Low-rank adapta- tion of large language models. In Proceedings of the 10th ICLR, 2022, 2022.
  16. [Hu et al., 2022b] Edward J Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu, Yuanzhi Li, Shean Wang, Lu Wang, Weizhu Chen, et al. Lora: Low-rank adaptation of large language models. ICLR, 2022.
  17. [Karimireddy et al., 2020] Sai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank Reddi, Sebastian Stich, and Ananda Theertha Suresh. SCAFFOLD: Stochastic controlled averaging for federated learning. In Proceedings of the 37th International Conference on Machine Learning, 2020.
  18. [Kuang et al., 2024] Weirui Kuang, Bingchen Qian, Zitao Li, Daoyuan Chen, Dawei Gao, Xuchen Pan, Yuexi- ang Xie, Yaliang Li, Bolin Ding, and Jingren Zhou. FederatedScope-LLM: A Comprehensive Package for Fine-tuning Large Language Models in Federated Learn- ing. In Proceedings of the 30th KDD, 2024, 2024.
  19. [Li and Liang, 2021] Xiang Lisa Li and Percy Liang. Prefix- tuning: Optimizing continuous prompts for generation. In ACL. Association for Computational Linguistics, 2021.
  20. [Li et al., 2019] Xiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang, and Zhihua Zhang. On the convergence of fedavg on non-iid data. In International Conference on Learning Representations, 2019.
  21. [Lin, 2004] Chin-Yew Lin. Rouge: A package for automatic evaluation of summaries. In Text Summarization Branches Out. Association for Computational Linguistics, 2004.
  22. [Liu et al., 2021] Bo Liu, Li-Ming Zhan, Li Xu, Lin Ma, Yan Yang, and Xiao-Ming Wu. Slake: A semantically-labeled knowledge-enhanced dataset for medical visual question answering. In 2021 IEEE 18th international symposium on biomedical imaging (ISBI). IEEE, 2021.
  23. [Liu et al., 2022] Xiao Liu, Heyan Huang, Ge Shi, and Bo Wang. Dynamic prefix-tuning for generative template- based event extraction. arXiv preprint arXiv:2205.06166, 2022.
  24. [Liu et al., 2023] Haotian Liu, Chunyuan Li, Qingyang Wu, and Yong Jae Lee. Visual instruction tuning, 2023.
  25. [Liu et al., 2025] Xiao-Yang Liu, Rongyi Zhu, Daochen Zha, Jiechao Gao, Shan Zhong, Matt White, and Meikang Qiu. Differentially Private Low-Rank Adaptation of Large Language Model Using Federated Learning. ACM Trans. Manag. Inf. Syst., 2025.
  26. [McMahan et al., 2017] Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, and Blaise Agüera y Ar- cas. Communication-Efficient Learning of Deep Networks from Decentralized Data. In Proceedings of the 20th AIS- TATS, 2017.
  27. [Mishra, 2024] Sandeep Mishra. battlemaster/llava-next- preference-dataset-20k. https://huggingface.co/datasets/ battleMaster/llava-next-Preference-dataset-20k, 2024. Accessed: 2026-05-11. Link
  28. [Papineni et al., 2002] Kishore Papineni, Salim Roukos, Todd Ward, and Wei-Jing Zhu. Bleu: A method for au- tomatic evaluation of machine translation. In Proceedings of the 40th Annual Meeting of the Association for Compu- tational Linguistics, 2002.
  29. [Reddi et al., 2021] Sashank J. Reddi, Zachary Charles, Manzil Zaheer, Zachary Garrett, Keith Rush, Jakub Konečný, Sanjiv Kumar, and Hugh Brendan McMahan. Adaptive federated optimization. In Proceedings of the 9th ICLR, 2021, 2021.
  30. [Savage et al., 2025] Thomas Savage, Stephen P Ma, Ab- dessalem Boukil, Ekanath Rangan, Vishwesh Patel, Ivan Lopez, and Jonathan Chen. Fine-tuning methods for large language models in clinical medicine by supervised fine- tuning and direct preference optimization: Comparative evaluation. Journal of Medical Internet Research, 2025.
  31. [Sun et al., 2024a] Youbang Sun, Zitao Li, Yaliang Li, and Bolin Ding. Improving lora in privacy-preserving feder- ated learning. arXiv preprint arXiv:2403.12313, 2024.
  32. [Sun et al., 2024b] Youbang Sun, Zitao Li, Yaliang Li, and Bolin Ding. Improving LoRA in Privacy-preserving Fed- erated Learning. In Proceedings of the 12th ICLR, 2024, 2024.
  33. [Wang et al., 2022] Zifeng Wang, Zhenbang Wu, Dinesh Agarwal, and Jimeng Sun. Medclip: Contrastive learning from unpaired medical images and text. In Proceedings of the EMNLP, 2022.
  34. [Wang et al., 2023] Ziao Wang, Yuhang Li, Junda Wu, Jae- hyeon Soon, and Xiaofeng Zhang. Finvis-gpt: A multi- modal large language model for financial chart analysis. arXiv preprint arXiv:2308.01430, 2023.
  35. [Wang et al., 2024a] Ziyao Wang, Zheyu Shen, Yexiao He, Guoheng Sun, Hongyi Wang, Lingjuan Lyu, and Ang Li. FLoRA: Federated Fine-Tuning Large Language Models with Heterogeneous Low-Rank Adaptations. In Proceed- ings of the NeurIPS, 2024, 2024.
  36. [Wang et al., 2024b] Ziyao Wang, Zheyu Shen, Yexiao He, Guoheng Sun, Hongyi Wang, Lingjuan Lyu, and Ang Li. Flora: Federated fine-tuning large language models with heterogeneous low-rank adaptations. arXiv preprint arXiv:2409.05976, 2024.
  37. [Wen et al., 2024] Qingsong Wen, Jing Liang, Carles Sierra, Rose Luckin, Richard Tong, Zitao Liu, Peng Cui, and Jil- iang Tang. Ai for education (ai4edu): Advancing person- alized education with llm and adaptive learning. In Pro- ceedings of the 30th KDD, 2024.
  38. [Wu et al., 2024] Xun Wu, Shaohan Huang, and Furu Wei. Mixture of lora experts. arXiv preprint arXiv:2404.13628, 2024.
  39. [Xiong et al., 2023] Baochen Xiong, Xiaoshan Yang, Yaguang Song, Yaowei Wang, and Changsheng Xu. Client-adaptive cross-model reconstruction network for modality-incomplete multimodal federated learning. In Proceedings of the 31st ACM International Conference on Multimedia (MM), MM 2023, Ottawa, ON, Canada, 29 October 2023-3 November 2023, 2023.
  40. [Yang et al., 2024a] Lishan Yang, Alireza Seyed Shakeri, Liangxi Pu, Weitong Chen, and Yanjun Shu. Efficient clus- tered federated learning by locality sensitive hashing. In International Conference on Advanced Data Mining and Applications. Springer, 2024.
  41. [Yang et al., 2024b] Yiyuan Yang, Guodong Long, Tao Shen, Jing Jiang, and Michael Blumenstein. Dual- Personalizing Adapter for Federated Foundation Models. In Proceedings of the NeurIPS, 2024, 2024.
  42. [Yang et al., 2025] Lishan Yang, Wei Emma Zhang, Quan Z Sheng, Lina Yao, Weitong Chen, and Ali Shakeri. Mmic: Mitigating modality incompleteness in clustered federated learning. CIKM, 2025.
  43. [Yu et al., 2023] Qiying Yu, Yang Liu, Yimu Wang, Ke Xu, and Jingjing Liu. Multimodal federated learning via con- trastive representation ensemble. In Proceedings of the 11th ICLR, 2023, Kigali, Rwanda, May 1-5, 2023, 2023.
  44. [Yuan et al., 2024] Haolin Yuan, William Paul, John Aucott, Philippe Burlina, and Yinzhi Cao. Pfededit: Personalized federated learning via automated model editing. In Euro- pean Conference on Computer Vision, 2024.
  45. [Zeng et al., 2024] Huimin Zeng, Zhenrui Yue, and Dong Wang. Open-vocabulary federated learning with multi- modal prototyping. In Proceedings of the 2024 Conference of the NAACL: Human Language Technologies (Long Pa- pers), 2024.
  46. [Zhang et al., 2024] Jianyi Zhang, Saeed Vahidian, Martin Kuo, Chunyuan Li, Ruiyi Zhang, Tong Yu, Guoyin Wang, and Yiran Chen. Towards Building The Federatedgpt: Federated Instruction Tuning. In Proceedings of the 2024 IEEE ICASSP, 2024.
  47. [Zong et al., 2021] Linlin Zong, Qiujie Xie, Jiahui Zhou, Peiran Wu, Xianchao Zhang, and Bo Xu. Fedcmr: Fed- erated cross-modal retrieval. In Proceedings of the 44th SIGIR, Virtual Event, Canada, July 11-15, 2021, 2021.

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

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