Learning to Collaborate: An Orchestrated-Decentralized Framework for Peer-to-Peer LLM Federation
Not on the map. Why?
Abstract (arXiv)
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 approx. 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.
Key facts. Cited by 2 papers in the map (data through 13 Oct 2026). Builds on 16 of its 34 references that are in the map. First posted on arXiv 23 Jan 2026.
Builds on
16 of 34 references.
References · 16
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,231
- 2021Evaluating Large Language Models Trained on Codecited by 8,205
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,653
- 2021Program Synthesis with Large Language Modelscited by 3,573
- 2018Federated Optimization in Heterogeneous Networkscited by 2,981
- 2020Inverting Gradients -- How easy is it to break privacy in federated learning?cited by 711
- 2020CodeBLEU: a Method for Automatic Evaluation of Code Synthesiscited by 475
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 459
- 2019BrainTorrent: A Peer-to-Peer Environment for Decentralized Federated Learningcited by 117
- 2023Federated Learning of Large Language Models with Parameter-Efficient Prompt Tuning and Adaptive Optimizationcited by 65
- 2022FedBalancer: Data and Pace Control for Efficient Federated Learning on Heterogeneous Clientscited by 46
- 2021Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devicescited by 36
4 more
- 2021IPLS : A Framework for Decentralized Federated Learningcited by 26
- 2023Personalized Decentralized Federated Learning with Knowledge Distillationcited by 21
- 2025Coral Protocol: Open Infrastructure Connecting The Internet of Agentscited by 9
- 2021Evaluating the Cybersecurity Risk of Real World, Machine Learning Production Systemscited by 6
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Agentic Federated Learning: The Future of Distributed Training Orchestration
- 2024Federated In-Context LLM Agent Learning
- 2026Federated Agent Optimization
- 2024Large Language Model Federated Learning with Blockchain and Unlearning for Cross-Organizational Collaboration
- 2025Can Federated Learning Safeguard Private Data in LLM Training? Vulnerabilities, Attacks, and Defense Evaluation
- 2026FL-MAESTRO: Multi-Agent LLM Orchestration for Resource-Constrained Federated Learning
- 2025Scaling Decentralized Learning with FLock
- 2024Recent Advances in Federated Learning Driven Large Language Models: A Survey on Architecture, Performance, and Security