arXiv:2410.17783cs.CLcs.HC2024-10被引 11

通过领域自适应提升RAG模型问答能力并减少幻觉

Leveraging the Domain Adaptation of Retrieval Augmented Generation Models for Question Answering and Reducing Hallucination

  • 用酒店对话数据微调RAG模型,实现领域适配
  • 领域自适应使问答准确率提升,幻觉率显著下降
  • 适合构建可靠客服类对话系统的研究者参考

尽管大语言模型在自然语言处理任务中表现卓越,检索增强生成模型(RAG)在问答等下游应用中尤为有效。近期的端到端RAG模型进一步优化架构,在领域自适应方面取得显著性能提升。然而,这些RAG架构在客户服务等专业领域微调后的表现仍缺乏深入研究,且如何在保持高领域准确性的同时减少幻觉仍是关键挑战。本文通过领域自适应评估多种RAG及类似架构在问答任务中的表现,并构建了新数据集HotelConvQA,该数据集源自大量酒店相关对话,用于模型微调。我们还首次系统性探究了领域自适应对不同RAG架构幻觉抑制的影响,填补了先前研究的空白。实验结果表明,领域自适应在所有指标上均取得正向效果,不仅提升了问答性能,也显著降低了各类RAG模型的幻觉发生率。

原文摘要 · Abstract (English)

While ongoing advancements in Large Language Models have demonstrated remarkable success across various NLP tasks, Retrieval Augmented Generation Model stands out to be highly effective on downstream applications like Question Answering. Recently, RAG-end2end model further optimized the architecture and achieved notable performance improvements on domain adaptation. However, the effectiveness of these RAG-based architectures remains relatively unexplored when fine-tuned on specialized domains such as customer service for building a reliable conversational AI system. Furthermore, a critical challenge persists in reducing the occurrence of hallucinations while maintaining high domain-specific accuracy. In this paper, we investigated the performance of diverse RAG and RAG-like architectures through domain adaptation and evaluated their ability to generate accurate and relevant response grounded in the contextual knowledge base. To facilitate the evaluation of the models, we constructed a novel dataset HotelConvQA, sourced from wide range of hotel-related conversations and fine-tuned all the models on our domain specific dataset. We also addressed a critical research gap on determining the impact of domain adaptation on reducing hallucinations across different RAG architectures, an aspect that was not properly measured in prior work. Our evaluation shows positive results in all metrics by employing domain adaptation, demonstrating strong performance on QA tasks and providing insights into their efficacy in reducing hallucinations. Our findings clearly indicate that domain adaptation not only enhances the models' performance on QA tasks but also significantly reduces hallucination across all evaluated RAG architectures.

RAG问答系统幻觉抑制领域自适应

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