Learning to Collaborate: An Orchestrated-Decentralized Framework for Peer-to-Peer LLM Federation
Inderjeet Singh, Eléonore Vissol-Gaudin, Andikan Otung, Motoyoshi Sekiya
Abstract
Fine-tuning Large Language Models (LLMs) for specialized domains is constrained by a fundamental challenge: the need for diverse, cross-organizational data conflicts with the principles of data privacy and sovereignty. While Federated Learning (FL) provides a framework for collaboration without raw data exchange, its classic centralized form introduces a single point of failure and remains vulnerable to model inversion attacks. Decentralized FL (DFL) mitigates this risk by removing the central aggregator but typically relies on inefficient, random peer-to-peer (P2P) pairings, forming a collaboration graph that is blind to agent heterogeneity and risks negative transfer. This paper introduces KNEXA-FL, a novel framework for orchestrated decentralization that resolves this trade-off. KNEXA-FL employs a non-aggregating Central Profiler/Matchmaker (CPM) that formulates P2P collaboration as a contextual bandit problem, using a LinUCB algorithm on abstract agent profiles to learn an optimal matchmaking policy. It orchestrates direct knowledge exchange between heterogeneous, PEFT-based LLM agents via secure distillation, without ever accessing the models themselves. Our comprehensive experiments on a challenging code generation task show that KNEXA-FL yields substantial gains, improving Pass@1 by approximately 50% relative to random P2P collaboration. Critically, our orchestrated approach demonstrates stable convergence, in stark contrast to a powerful centralized distillation baseline which suffers from catastrophic performance collapse. Our work establishes adaptive, learning-based orchestration as a foundational principle for building robust and effective decentralized AI ecosystems.
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.
Builds on8
- Practical Secure Aggregation for Privacy-Preserving Machine LearningKallista A. Bonawitz, Vladimir Ivanov, Ben Kreuter, Antonio Marcedone et al.CCS 2017 · 3,936 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Loki: Large-scale Data Reconstruction Attack against Federated Learning through Model ManipulationJoshua C. Zhao, Atul Sharma, Ahmed Roushdy Elkordy, Yahya H. Ezzeldin et al.S&P 2024 · 64 citations
- Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable DevicesMax Ryabinin, Eduard Gorbunov, Vsevolod Plokhotnyuk, Gennady PekhimenkoNeurIPS 2021 · 59 citations
- Federated Learning of Large Language Models with Parameter-Efficient Prompt Tuning and Adaptive OptimizationTianshi Che, Ji Liu, Yang Zhou, Jiaxiang Ren et al.EMNLP 2023 · 43 citations
Related papers
- Helmsman: Autonomous Synthesis of Federated Learning Systems via Collaborative LLM AgentsHaoyuan Li, Mathias Funk, Aaqib SaeedICLR 2026 · 1 citation
- FDPT: Federated Discrete Prompt Tuning for Black-Box Visual-Language ModelsJiaqi Wu, Simin Chen, Jing Tang, Yuzhe Yang et al.ICCV 2025 · 1 citation
- Learning to Orchestrate Agents in Natural Language with the ConductorStefan Nielsen, Edoardo Cetin, Peter Schwendeman, Qi Sun et al.ICLR 2026 · 22 citations
- DeMIC: Decentralized Meta In-Context Learning with Refiner-Guided AdaptationMingyi Li, Xiao Zhang, Zengzhe Chen, Jiawei Zhang et al.KDD 2026
- STARK: Strategic Team of Agents for Refining KernelsJuncheng Dong, Yang Yang, Tao Liu, Yang Wang et al.ICLR 2026 · 26 citations
