arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.08054cs.LGcs.AIcs.CLcs.CR

联邦上下文学习大语言模型智能体学习

Federated In-Context LLM Agent Learning

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳))

Panlong Wu, Kangshuo Li, Junbao Nan, Fangxin Wang

AI总结:

针对大语言模型训练数据稀缺且敏感、联邦学习带宽计算需求高及上下文学习存隐私泄露风险的问题,提出FICAL算法,通过知识纲要传输与RAG工具学习模块,在保持性能的同时大幅降低通信成本。

摘要:

大语言模型(LLMs)作为智能体能够实现逻辑推理、工具使用以及与外部系统的交互,已彻底革新了智能服务领域。然而,高质量数据的稀缺性常常阻碍LLMs的发展,而这类数据大多本身具有敏感性。联邦学习(FL)通过支持分布式LLMs的协同训练同时保护隐私数据,提供了一种潜在解决方案。但FL框架面临着巨大的带宽和计算需求,以及数据分布异构带来的挑战。LLMs新兴的上下文学习能力通过聚合自然语言而非庞大的模型参数,提供了一种有前景的方法。然而,这种方法存在隐私泄露风险,因为它需要在聚合过程中收集并展示来自不同客户端的数据样本。\n在本文中,我们提出了一种新颖的隐私保护型联邦上下文LLM智能体学习(FICAL)算法,据我们所知,这是首个利用上下文学习的力量通过联邦学习训练多样化LLM智能体的工作。在我们的设计中,由一种新颖的LLM增强型知识纲要生成(KCG)模块生成的知识纲要在客户端与服务器之间传输,而非以往FL方法中的模型参数。除此之外,我们设计了一个出色的基于检索增强生成(RAG)的工具学习与利用(TLU)模块,并将聚合后的全局知识纲要作为教师,来教授LLM智能体工具的使用方法。我们开展了大量实验,结果表明,与其他SOTA基线相比,FICAL具有极具竞争力的性能,同时通信成本显著降低了3.33×10^5倍。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑