IJCAIJun, 2024

大型语言模型的指导下的文档记忆

TL;DR利用大数据训练的 AI 模型在参数中存储着与文档相关的内容。本研究提出了一种基于文档记忆的架构,通过映射文档表示到记忆条目来追踪训练中的文档记忆。同时,引入了文档引导损失,增加带有文档记忆的文本的可能性,减少其他文档记忆的文本的可能性。实验结果表明,所提出的方法在生成文档相关内容时,在记忆条目和高召回率方面具有优势。