该研究针对检索增强生成(RAG)系统中检索文档可能不足或相互冲突的问题,提出利用语言模型内部信号进行三分类判断:信息充分、不充分或冲突。研究者构建了包含虚构信息的受控基准数据集,并提取模型的隐藏层激活和注意力特征,训练轻量级线性分类器作为路由机制。实验覆盖16种不同架构和规模的语言模型,结果显示该特征路由方法在分类性能上始终优于基于提示的基线方法及专门的RAG模型。进一步分析表明,模型中间层包含最丰富的判别信号,且隐藏激活状态在多数模型中比注意力值或MLP特征输出更有效。该研究证实语言模型内部确实编码了关于检索证据充分性的信息,且这些信号可被可靠解码用于RAG系统的质量把关,为提升RAG系统的可靠性提供了新思路。
| Retrieval-Augmented Generation (RAG) | 检索增强生成,一种结合信息检索和语言生成的技术,用于提升回答的准确性和时效性。 |
| three-way classification | 三分类问题,将输入信息分为充分、不足或冲突三类。 |
| hidden activations | 隐藏层激活,语言模型内部神经元的输出值,用于捕捉模型处理信息时的状态。 |
| attention-derived features | 注意力派生特征,从模型的注意力机制中提取的特征,反映模型对输入不同部分的关注程度。 |
| feature-based router | 基于特征的分类器,使用模型内部特征进行决策的轻量级模型,用于判断RAG中的信息状态。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅