提升扩散语言模型的生成精度,解决检索增强中的语义漂移问题
Unlocking the Potentials of Retrieval-Augmented Generation for Diffusion Language Models
- 引入查询相关性引导的去噪策略,保持生成与查询语义一致
- 实验显示新方法显著提升生成精度,有效缓解语义漂移
- 适合关注生成质量与检索增强融合的研究者和应用开发者
扩散语言模型(DLMs)在自然语言处理任务中展现出强大能力,但其在检索增强生成(RAG)框架下的潜力尚未充分探索,原因在于LLM与DLM解码机制的根本差异。本文系统评估了DLM在RAG框架下的表现,发现其对上下文依赖性强,但生成精度有限。我们识别出关键问题:响应语义漂移(RSD),即生成内容在迭代去噪过程中逐渐偏离原始查询语义,导致低精度输出。根源在于现有去噪策略无法在全过程维持与查询的语义对齐。为此,我们提出语义保持型检索增强扩散模型(SPREAD),通过查询相关性引导的去噪策略,主动约束生成轨迹,确保结果始终锚定于查询语义。实验表明,SPREAD显著提升了生成精度,有效抑制了语义漂移。
原文摘要 · Abstract (English)
Diffusion Language Models (DLMs) have recently demonstrated remarkable capabilities in natural language processing tasks. However, the potential of Retrieval-Augmented Generation (RAG), which shows great successes for enhancing large language models (LLMs), has not been well explored, due to the fundamental difference between LLM and DLM decoding. To fill this critical gap, we systematically test the performance of DLMs within the RAG framework. Our findings reveal that DLMs coupled with RAG show promising potentials with stronger dependency on contextual information, but suffer from limited generation precision. We identify a key underlying issue: Response Semantic Drift (RSD), where the generated answer progressively deviates from the query's original semantics, leading to low precision content. We trace this problem to the denoising strategies in DLMs, which fail to maintain semantic alignment with the query throughout the iterative denoising process. To address this, we propose Semantic-Preserving REtrieval-Augmented Diffusion (SPREAD), a novel framework that introduces a query-relevance-guided denoising strategy. By actively guiding the denoising trajectory, SPREAD ensures the generation remains anchored to the query's semantics and effectively suppresses drift. Experimental results demonstrate that SPREAD significantly enhances the precision and effectively mitigates RSD of generated answers within the RAG framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。