arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.02412cs.LGcs.AIcs.CLcs.CV
收藏

LLM增强LLM:通过组合扩展能力

LLM Augmented LLMs: Expanding Capabilities through Composition

Google Research(谷歌研究院);Google DeepMind(谷歌DeepMind)

Rachit Bansal, Bidisha Samanta, Siddharth Dalmia, Nitish Gupta, Shikhar Vashishth, Sriram Ganapathy, Abhishek Bapna, Prateek Jain, Partha Talukdar

AI总结:

本文提出CALM方法,通过引入交叉注意力组合现有基础模型与特定任务模型,在保留原有能力的同时实现新任务的高效扩展,显著提升低资源语言和代码任务的性能。

摘要:

在大量数据语料库上训练的具有数十亿参数的基础模型在多个领域展示了非凡的能力。然而,由于其单体结构,增强它们或赋予新技能具有挑战性且成本高昂。另一方面,由于它们的适应能力,这些模型的新实例正被训练以适应新的领域和任务。在本研究中,我们探讨了将现有基础模型与更具体的模型进行高效且实用的组合以实现新能力的问题。为此,我们提出了CALM(Composition to Augment Language Models,组合增强语言模型),该方法在模型之间引入交叉注意力来组合它们的表示并实现新能力。CALM的主要特点是:(i)通过“重用”现有LLM以及少量额外参数和数据,在新任务上扩展LLM;(ii)保持现有模型权重不变,从而保留现有能力;(iii)适用于多样化的领域和设置。我们展示了将PaLM2-S与在低资源语言上训练的较小模型相结合,在翻译成英语和低资源语言算术推理等任务上实现了13%的绝对提升。类似地,当PaLM2-S与特定代码模型结合时,在代码生成和解释任务上比基础模型实现了40%的相对提升——与完全微调的对应模型相当。

英文摘要:

Foundational models with billions of parameters which have been trained on large corpora of data have demonstrated non-trivial skills in a variety of domains. However, due to their monolithic structure, it is challenging and expensive to augment them or impart new skills. On the other hand, due to their adaptation abilities, several new instances of these models are being trained towards new domains and tasks. In this work, we study the problem of efficient and practical composition of existing foundation models with more specific models to enable newer capabilities. To this end, we propose CALM -- Composition to Augment Language Models -- which introduces cross-attention between models to compose their representations and enable new capabilities. Salient features of CALM are: (i) Scales up LLMs on new tasks by 're-using' existing LLMs along with a few additional parameters and data, (ii) Existing model weights are kept intact, and hence preserves existing capabilities, and (iii) Applies to diverse domains and settings. We illustrate that augmenting PaLM2-S with a smaller model trained on low-resource languages results in an absolute improvement of up to 13\% on tasks like translation into English and arithmetic reasoning for low-resource languages. Similarly, when PaLM2-S is augmented with a code-specific model, we see a relative improvement of 40\% over the base model for code generation and explanation tasks -- on-par with fully fine-tuned counterparts.

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

↑