arXiv:2503.02846cs.CL2025-03ICLR被引 28

通过细粒度事实对齐,让大模型回答更真实可信。

Mask-DPO: Generalizable Fine-grained Factuality Alignment of LLMs

  • 用句子级事实性作为掩码,只学习正确内容,避免训练噪声。
  • 在ANAH测试集上准确率从49.19%提升至77.53%,超越70B模型。
  • 适用于未见过的领域和主题,尤其适合提升小模型事实性。

大语言模型在充当AI助手时常出现幻觉(即不真实或无意义的信息)。由于幻觉总是与真实内容共存,以往基于响应级别的偏好学习方法在训练中不可避免引入噪声。为此,本文提出一种基于直接偏好优化(DPO)的细粒度事实对齐方法——Mask-DPO。该方法引入句子级事实性作为掩码信号,仅在优选样本中学习正确句子,防止对非优选样本中真实内容的惩罚,解决了偏好学习中的歧义问题。大量实验表明,即使在训练中未见过的领域和主题下,Mask-DPO也能显著提升模型回答的事实性。仅在ANAH训练集上训练后,Llama3.1-8B-Instruct在ANAH测试集上的得分从49.19%提升至77.53%,甚至超过Llama3.1-70B-Instruct的53.44%;在跨域传记数据集上的FactScore也从30.29%提升至39.39%。进一步研究发现,扩大数据集中主题数量比增加问题数量更有利于泛化。本文还提出了关于事实对齐机制的假设,并通过概念验证实验加以验证,期望为未来事实对齐的规模化研究提供基础。

原文摘要 · Abstract (English)

Large language models (LLMs) exhibit hallucinations (i.e., unfaithful or nonsensical information) when serving as AI assistants in various domains. Since hallucinations always come with truthful content in the LLM responses, previous factuality alignment methods that conduct response-level preference learning inevitably introduced noises during training. Therefore, this paper proposes a fine-grained factuality alignment method based on Direct Preference Optimization (DPO), called Mask-DPO. Incorporating sentence-level factuality as mask signals, Mask-DPO only learns from factually correct sentences in the preferred samples and prevents the penalty on factual contents in the not preferred samples, which resolves the ambiguity in the preference learning. Extensive experimental results demonstrate that Mask-DPO can significantly improve the factuality of LLMs responses to questions from both in-domain and out-of-domain datasets, although these questions and their corresponding topics are unseen during training. Only trained on the ANAH train set, the score of Llama3.1-8B-Instruct on the ANAH test set is improved from 49.19% to 77.53%, even surpassing the score of Llama3.1-70B-Instruct (53.44%), while its FactScore on the out-of-domain Biography dataset is also improved from 30.29% to 39.39%. We further study the generalization property of Mask-DPO using different training sample scaling strategies and find that scaling the number of topics in the dataset is more effective than the number of questions. We provide a hypothesis of what factual alignment is doing with LLMs, on the implication of this phenomenon, and conduct proof-of-concept experiments to verify it. We hope the method and the findings pave the way for future research on scaling factuality alignment.

事实对齐大模型幻觉偏好学习模型泛化

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