arXiv:2410.23605cs.CL2024-10NAACL被引 6

通过动态排序检索结果提升大模型对冷门知识的问答准确率。

Dynamic Uncertainty Ranking: Enhancing Retrieval-Augmented In-Context Learning for Long-Tail Knowledge in LLMs

  • 用强化学习动态调整检索样本优先级,选更可靠的信息
  • 在长尾问题上准确率提升5.96%,整体比基线高2.76%
  • 适合需要精准冷门知识推理的应用场景

大语言模型虽在预训练中学习了广泛知识,但冷门领域知识仍稀缺且未充分记忆。已有研究显示,结合检索器的上下文学习(ICL)可帮助模型更好捕捉冷门知识,减少对预训练数据的依赖。然而我们发现,模型对冷门问题的回答仍受检索样本变化影响而不确定。为此,提出基于强化学习的动态不确定性排序方法,考虑每个检索样本对模型预测的影响差异,优先选择信息量高、稳定的样本,淘汰误导性内容,并根据模型反馈实时更新排序。为提高训练效率并降低查询成本,引入可学习的动态阈值,在预测出现负向偏移时自动调整。在多个跨领域问答数据集上的实验表明,该方法优于最佳基线2.76%,尤其在零样本无法覆盖的长尾问题上提升5.96%。

原文摘要 · Abstract (English)

Large language models (LLMs) can learn vast amounts of knowledge from diverse domains during pre-training. However, long-tail knowledge from specialized domains is often scarce and underrepresented, rarely appearing in the models' memorization. Prior work has shown that in-context learning (ICL) with retriever augmentation can help LLMs better capture long-tail knowledge, reducing their reliance on pre-trained data. Despite these advances, we observe that LLM predictions for long-tail questions remain uncertain to variations in retrieved samples. To take advantage of the uncertainty in ICL for guiding LLM predictions toward correct answers on long-tail samples, we propose a reinforcement learning-based dynamic uncertainty ranking method for ICL that accounts for the varying impact of each retrieved sample on LLM predictions. Our approach prioritizes more informative and stable samples while demoting misleading ones, updating rankings based on the feedback from the LLM w.r.t. each retrieved sample. To enhance training efficiency and reduce query costs, we introduce a learnable dynamic ranking threshold, adjusted when the model encounters negative prediction shifts. Experimental results on various question-answering datasets from different domains show that our method outperforms the best baseline by $2.76\%$, with a notable $5.96\%$ boost in accuracy on long-tail questions that elude zero-shot inference.

知识增强上下文学习动态排序

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