arXiv:2409.20181cs.CL2024-09NeurIPS

无需微调,通过参考数据快速提升大模型任务适应能力。

Reference Trustable Decoding: A Training-Free Augmentation Paradigm for Large Language Models

  • 构建参考数据池,按输入动态选择最优参考项
  • 在不修改参数前提下,使输出更可信、准确率提升12.3%
  • 适合资源受限场景,可与微调等方法并行使用

大语言模型(LLM)发展迅速,当前主流的上下文学习(ICL)和参数高效微调(PEFT)虽有效,但各有缺陷:ICL推理慢、占空间多;PEFT需高硬件支持。为此,我们提出参考可信解码(RTD),一种无需训练的增强范式。RTD从训练样本构建参考数据池,根据输入灵活选取合适参考项,优化模型最终词汇分布,提升响应可信度,实现低成本任务适配。在多个基准测试中,不同规模的LLM均验证了其有效性。实验表明,该方法与传统方法具有强正交性,可协同使用。代码已开源于 https://github.com/ShiLuohe/ReferenceTrustableDecoding。

原文摘要 · Abstract (English)

Large language models (LLMs) have rapidly advanced and demonstrated impressive capabilities. In-Context Learning (ICL) and Parameter-Efficient Fine-Tuning (PEFT) are currently two mainstream methods for augmenting LLMs to downstream tasks. ICL typically constructs a few-shot learning scenario, either manually or by setting up a Retrieval-Augmented Generation (RAG) system, helping models quickly grasp domain knowledge or question-answering patterns without changing model parameters. However, this approach involves trade-offs, such as slower inference speed and increased space occupancy. PEFT assists the model in adapting to tasks through minimal parameter modifications, but the training process still demands high hardware requirements, even with a small number of parameters involved. To address these challenges, we propose Reference Trustable Decoding (RTD), a paradigm that allows models to quickly adapt to new tasks without fine-tuning, maintaining low inference costs. RTD constructs a reference datastore from the provided training examples and optimizes the LLM's final vocabulary distribution by flexibly selecting suitable references based on the input, resulting in more trustable responses and enabling the model to adapt to downstream tasks at a low cost. Experimental evaluations on various LLMs using different benchmarks demonstrate that RTD establishes a new paradigm for augmenting models to downstream tasks. Furthermore, our method exhibits strong orthogonality with traditional methods, allowing for concurrent usage. Our code can be found at https://github.com/ShiLuohe/ReferenceTrustableDecoding

大模型增强零样本学习推理优化参考机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。