让大模型和小模型在联邦学习中互相提升,保护隐私且开销小。
Federated Co-tuning Framework for Large and Small Language Models
- 用轻量适配器实现大模型与小模型协同优化。
- 小模型性能显著提升,大模型效果接近直接微调。
- 适合需要隐私保护的跨设备智能应用。
通过将大语言模型(LLMs)适配到特定领域任务或注入领域知识,可充分释放其潜力。然而,如何实现服务器端大模型与客户端小语言模型(SLMs)的同步相互增强仍存挑战。为此,我们提出FedCoLLM——一种新型高效的联邦学习框架,用于联合优化大模型与小模型。该方法通过在客户端部署轻量级适配器,实现服务器大模型与客户端小模型之间的知识双向迁移,在保障数据隐私的同时显著降低计算与通信开销。在多个公开大模型与小模型及多种NLP文本生成任务上的评估表明,借助大模型辅助,客户端小模型性能显著提升;同时,经联邦协同优化的大模型表现可媲美直接在客户端数据上微调的结果。代码已开源至FATE项目:https://github.com/FederatedAI/FATE-LLM/tree/main/python/fate_llm/algo/fedcollm。
原文摘要 · Abstract (English)
By adapting Large Language Models (LLMs) to domain-specific tasks or enriching them with domain-specific knowledge, we can fully harness the capabilities of LLMs. Nonetheless, a gap persists in achieving simultaneous mutual enhancement between the server's LLM and the downstream clients' Small Language Models (SLMs). To address this, we propose FedCoLLM, a novel and parameter-efficient federated framework designed for co-tuning LLMs and SLMs. This approach is aimed at adaptively transferring server-side LLMs knowledge to clients' SLMs while simultaneously enriching the LLMs with domain insights from the clients. To accomplish this, FedCoLLM utilizes lightweight adapters in conjunction with SLMs, facilitating knowledge exchange between server and clients in a manner that respects data privacy while also minimizing computational and communication overhead. Our evaluation of FedCoLLM, utilizing various public LLMs and SLMs across a range of NLP text generation tasks, reveals that the performance of clients' SLMs experiences notable improvements with the assistance of the LLMs. Simultaneously, the LLMs enhanced via FedCoLLM achieves comparable performance to that obtained through direct fine-tuning on clients' data. Our code has been contributed to the FATE open-source project and is now publicly accessible at https://github.com/FederatedAI/FATE-LLM/tree/main/python/fate_llm/algo/fedcollm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。