paper-with-me

홈 › Papers

Federated In-Context LLM Agent Learning

2024-12-11 · Panlong Wu, Kangshuo Li, Junbao Nan, Fangxin Wang

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.

📄 PDF Abstract BibTeX arXiv:2412.08054

Code (0)

등록된 구현이 없습니다.

Tasks

Federated LearningIn-Context LearningLogical ReasoningPrivacy PreservingRAGRetrieval-augmented Generation

Similar Papers 제목 키워드 기반

A Simple and Provably Efficient Algorithm for Asynchronous Federated Contextual Linear Bandits

2022-07-07 · Jiafan He, Tianhao Wang, Yifei Min, Quanquan Gu

We study federated contextual linear bandits, where $M$ agents cooperate with each other to solve a global contextual linear bandit problem with the help of a central server. We consider the asynchronous setting, where a…

Federated Learning for Heterogeneous Bandits with Unobserved Contexts

2023-03-29 · Jiabin Lin, Shana Moothedath

We study the problem of federated stochastic multi-arm contextual bandits with unknown contexts, in which M agents are faced with different bandits and collaborate to learn. The communication model consists of a central …

Federated LearningMulti-Armed Bandits

Federated Neural Bandits

2022-05-28 · Zhongxiang Dai, Yao Shu, Arun Verma, Flint Xiaofeng Fan 외

Recent works on neural contextual bandits have achieved compelling performances due to their ability to leverage the strong representation power of neural networks (NNs) for reward prediction. Many applications of contex…

Multi-Armed Bandits

FedFormer: Contextual Federation with Attention in Reinforcement Learning

2022-05-27 · Liam Hebert, Lukasz Golab, Pascal Poupart, Robin Cohen

A core issue in multi-agent federated reinforcement learning is defining how to aggregate insights from multiple agents. This is commonly done by taking the average of each participating agent's model weights into one co…

Federated Learningreinforcement-learningReinforcement LearningReinforcement Learning (RL)

Differentially-Private Federated Linear Bandits

2020-10-22 · NeurIPS 2020 12 · Abhimanyu Dubey, Alex Pentland

The rapid proliferation of decentralized learning systems mandates the need for differentially-private cooperative learning. In this paper, we study this in context of the contextual linear bandit: we consider a collecti…

Federated Learning