May, 2024

可证明鲁棒检索错误的 RAG 模型

TL;DR在本文中,我们提出了 RobustRAG 作为对抗检索污染攻击的第一个防御框架,通过采用独立后聚合策略,我们设计了基于关键词和解码的算法,以安全地聚合非结构化文本响应,并证明了 RobustRAG 在某些查询上能始终返回准确的响应,即使攻击者对我们的防御有完全了解并能随意注入少量恶意段落。我们在开放域问题回答和长文本生成数据集上评估了 RobustRAG,展示了其在各种任务和数据集上的有效性和泛化能力。