Faithfulness-QA: A Counterfactual Entity Substitution Dataset for Training Context-Faithful RAG Models
本文介绍了 Faithfulness-QA,这是一个包含近 10 万个经反事实修改的问答样本的大规模数据集,旨在训练和评估检索增强生成模型,使其优先利用检索到的上下文而非内部参数记忆,从而解决生成不忠实这一关键问题。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
想象一下,你有一位非常聪明的学生,他读过数百万本书,并记住了海量的事实。这位学生非常擅长回答问题,但他有一个坏习惯:当你给他一篇新文章阅读并就其提问时,他往往会完全忽略这篇文章。相反,他只是脱口而出自己记忆中的内容,即使这些记忆是错误的或过时的。
在人工智能领域,这被称为RAG(检索增强生成)模型。RAG 的目标是让 AI 阅读新文章(即“上下文”),并仅基于该文章进行回答。但通常,AI 过于固执,反而依赖其“参数化记忆”(即其预训练的大脑)。
本文介绍了一种名为**Faithfulness-QA(忠实性问答)**的解决方案。你可以将其想象为一个特殊的训练营,旨在教导 AI 真正去倾听新文章。
问题所在:“固执的学生”
通常,当我们训练这些 AI 模型时,文章中的问题和答案往往与 AI 已经知道的内容相匹配。因此,AI 能答对问题,但我们不知道它为什么答对。它是阅读了文章,还是仅仅因为它已经知道这个事实而猜对了?
这就像问一个学生:“谁是美国第一任总统?”同时给他们看一张乔治·华盛顿的照片。如果他们回答“乔治·华盛顿”,他们可能是看了照片,也可能只是复述了他们在幼儿园时背诵的内容。我们无法区分这两种情况。
解决方案:“陷阱问题”法
为了解决这个问题,研究人员创建了一个包含99,094 个“陷阱问题”的大型数据集。他们使用了一种巧妙的方法,称为反事实实体替换。
以下是他们构建该数据集的过程,使用一个简单的类比:
- 源材料:他们从两个现有的巨大问答库(SQuAD 和 TriviaQA)开始。
- “实体库”:他们建立了一个巨大的数字目录,包含近 77,000 个名称、地点和日期(如“巴黎”、“玛丽·居里”或"7 月 4 日”)。
- 替换:对于每个问题,他们找到故事中特定的名称或地点(即“承载答案的实体”),并将其与目录中属于同一类别的另一个不同的名称或地点进行交换。
- 原始故事:“圣母玛利亚于 1858 年向圣伯尔纳定显现。”
- 替换操作:他们将“圣伯尔纳定”替换为“平清盛”(另一个人名)。
- 新故事:“圣母玛利亚于 1858 年向平清盛显现。”
- 冲突:现在,AI 的记忆会说“是圣伯尔纳定!”,但新故事却说“是平清盛!”
为什么这有效
这制造了一个知识冲突。AI 被迫做出选择:
- 选项 A:忽略故事,给出其记忆中的答案(圣伯尔纳定)。
- 选项 B:忽略其记忆,信任故事(平清盛)。
如果 AI 是“忠实的”,它必须选择选项 B。通过在成千上万个此类冲突中进行训练,AI 学会了当存在冲突时,新故事才是主导者。
质量控制
研究人员并非随机替换词语;他们建立了一个严格的“质量过滤器”(就像一位非常挑剔的编辑)。
- 他们检查了新名称是否真正符合句子语境。
- 他们确保故事不过短或破碎。
- 他们验证了新名称是否原本就不在故事中。
他们在 200 个故事的随机样本上进行了测试,100% 的样本都通过了这些检查。最终的数据集非常庞大(99,000 个样本),涵盖了八种不同类型的实体,如人物、组织、日期和地点。
他们提供了什么
该论文向公众发布了三样主要成果:
- 数据集:准备好用于训练的 99,000 个陷阱问题。
- 流水线:自动生成这些陷阱问题的代码。
- 实体库:用于替换的 77,000 个名称和地点列表。
核心结论
该论文声称,通过使用此数据集,我们可以训练 AI 模型,使其不再成为那些忽略新信息的“固执学生”。相反,它们将成为“忠实读者”,将眼前的文本置于其自以为已知的知识之上。作者指出,虽然该系统在替换名称方面非常有效,但它无法修正代词(例如,如果名称改变,无法将“他”改为“她”),并且目前仅适用于英语。但就其特定目的——教导 AI 信任上下文而非其记忆而言——这是一个巨大且高质量的新工具。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。