用自然语言知识替代模型参数,实现私密环境下大模型协作训练。
Federated In-Context LLM Agent Learning
- 以知识摘要代替模型参数传输,降低通信开销
- 通过教师引导的工具学习模块提升智能体能力
- 在保护隐私前提下实现跨客户端大模型协同优化
大语言模型(LLM)通过逻辑推理、工具使用和外部系统交互展现出强大智能。然而,高质量数据稀缺且常具敏感性,限制了模型发展。联邦学习(FL)可在保护隐私的前提下实现分布式模型协同训练,但面临带宽与计算资源压力,以及数据分布异构等挑战。新兴的上下文学习能力可通过传递自然语言而非庞大的模型参数来缓解问题,但存在隐私泄露风险,因需收集并展示各客户端的数据样本。本文提出首个将上下文学习应用于联邦学习的隐私保护框架——FICAL。其核心是通过新型LLM增强的知识摘要生成(KCG)模块,将客户端生成的知识摘要上传至服务器,而非传统模型参数。此外,设计基于检索增强生成(RAG)的工具学习与使用(TLU)模块,并利用聚合后的全局知识摘要作为教师指导模型学习工具使用。实验表明,相比现有最优方法,FICAL性能相当,通信成本降低3.33×10⁵倍。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have revolutionized intelligent services by enabling logical reasoning, tool use, and interaction with external systems as agents. The advancement of LLMs is frequently hindered by the scarcity of high-quality data, much of which is inherently sensitive. Federated learning (FL) offers a potential solution by facilitating the collaborative training of distributed LLMs while safeguarding private data. However, FL frameworks face significant bandwidth and computational demands, along with challenges from heterogeneous data distributions. The emerging in-context learning capability of LLMs offers a promising approach by aggregating natural language rather than bulky model parameters. Yet, this method risks privacy leakage, as it necessitates the collection and presentation of data samples from various clients during aggregation. In this paper, we propose a novel privacy-preserving Federated In-Context LLM Agent Learning (FICAL) algorithm, which to our best knowledge for the first work unleashes the power of in-context learning to train diverse LLM agents through FL. In our design, knowledge compendiums generated by a novel LLM-enhanced Knowledge Compendiums Generation (KCG) module are transmitted between clients and the server instead of model parameters in previous FL methods. Apart from that, an incredible Retrieval Augmented Generation (RAG) based Tool Learning and Utilizing (TLU) module is designed and we incorporate the aggregated global knowledge compendium as a teacher to teach LLM agents the usage of tools. We conducted extensive experiments and the results show that FICAL has competitive performance compared to other SOTA baselines with a significant communication cost decrease of $\mathbf{3.33\times10^5}$ times.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。