生成 · 2023

REPLUG

不改黑盒模型的参数,把同一问题配上不同文档分别计算,再合并概率。

演进中 · 提示或小模型训练

要解决的问题

RETRO 和 Atlas 都要训练模型。GPT 一类接口看不到隐藏状态,也改不了权重。

相对前一代的变化

kNN-LM 需要内部表示。REPLUG 只使用黑盒模型对「文档加问题」算出的输出概率,把多篇文档的结果按检索分数合并。

做法

Shi 等人先检索文档。每篇文档单独和问题一起送给冻结模型,再把这些输出分布按文档概率加权。另一个版本会反过来用模型偏好调整检索器。

局限

文档数量增加时,模型要跑多次。合并的是输出分布,长答案并不总有可合并的词概率。

在链路中的位置

面向只能调用接口的模型,位于检索之后的生成融合。