Weaving in the Clouds: Achieving Synergistic Collaboration among LLM Agents via Federated Learning
Jiaxing Zhao, Hongbin Xie, Yuzhen Lei, Xuan Song, Zhuoran Shi, Lianxin Li, Shuangxue Liu, Haoran Zhang
Abstract
Multi-Agent Systems (MAS) powered by Large Language Models (LLMs) have become a promising paradigm for solving workflow-structured tasks through expert collaboration. However, the data required for such collaboration are often distributed across organizations and cannot be centrally pooled due to privacy, intellectual property, and compliance constraints. Federated Learning preserves data locality, but existing federated paradigms typically treat clients as independent and fail to model workflow dependencies that are crucial for coherent multi-stage collaboration. We introduce FedWave , a federated and workflow-aware framework that enables LLM-based experts to solve sequential tasks under data-locality constraints. FedWave combines a Value Chain Layer to model inter-stage dependencies, federated LoRA adaptation to preserve role-specific expertise, a shared Mixture-of-Experts (MoE) router for input-conditioned expert fusion, and a Direct Preference Optimization (DPO) stage based on router-induced preferences. Experiments across workflow datasets and LLM backbones show that FedWave outperforms federated baselines and remains competitive with centralized multi-agent systems while preserving data locality.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0309e487-9e33-4a6b-aab9-4c33234d81ceBuilds on12
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning et al.NeurIPS 2023 · 10,924 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Tackling the Objective Inconsistency Problem in Heterogeneous Federated OptimizationJianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi et al.NeurIPS 2020 · 2,231 citations
- Adaptive Federated OptimizationSashank J. Reddi, Zachary Charles, Manzil Zaheer, Zachary Garrett et al.ICLR 2021 · 1,917 citations
Related papers
- Personalized Federated Fine-Tuning for LLMs via Data-Driven Heterogeneous Model ArchitecturesYicheng Zhang, Zhen Qin, Zhaomin Wu, Jian Hou et al.WWW 2026 · 9 citations
- DAWN: Distributed LLM Multi-Agent Workflow SynthesisGuancheng Wan, Mo Zhou, Ziyi Wang, Xiaoran Shang et al.AAAI 2026
- FedVLA: Federated Vision-Language-Action Learning with Dual Gating Mixture-of-Experts for Robotic ManipulationCui Miao, Tao Chang, Meihan Wu, Hongbin Xu et al.ICCV 2025 · 7 citations
- FedALT: Federated Fine-Tuning Through Adaptive Local Training with Rest-of-World LoRAJieming Bian, Lei Wang, Letian Zhang, Jie XuAAAI 2026 · 12 citations
- On-Device Collaborative Language Modeling via a Mixture of Generalists and SpecialistsDongyang Fan, Bettina Messmer, Nikita Doikov, Martin JaggiICML 2025
