首页 > Ai资讯 > Ai知识库 > 会“追问”的Agent，大模型知识库另类实践

会“追问”的Agent，大模型知识库另类实践

发布时间：2024年06月06日

最近帮朋友实现了一个公积金Agent，能够基于公积金政策回答用户的问题，踩了一些坑，分享出来，以飨读者。

在实际的生产环境中，此类知识库应用可能遇到两类问题：知识型和政策型（图1）。

图1: 知识型问题 vs 政策型问题

相较于知识型问题，政策型问题对于知识库应用判定用户意图的能力和回答的准确性上有着近乎苛刻的要求：

100%的准确。基于交互的严肃性，应用输出内容必须能够达到100%的准确，尤其是在针对政策类、法规类和操作手册类文档。“100%的准确”不仅仅包括输出内容的正确性，还包括内容的完整性和时效性。

判定用户的真实意图。很多情况下，用户初始的意图是模糊的，应用必须有能力“追问”或“应对追问”，引导用户找到目标答案。知识型问答中常见的“一问一答”显然无法满足该需求。

那么，基于大模型Finetune和Langchain的知识库应用是否可以满足政策类问题的需求呢？

经过实践，我们抛弃了这两种方案。

抛弃Finetune和Langchain

我们首先排除了Finetune（FT）方式。排除FT的原因不仅仅出于训练数据准备的繁琐和额外的训练成本，而更是因为FT在“Factual Recall”类型任务上的不可靠。实际上，OpenAI早在去年就抛出类似的观点：

GPT can learn knowledge in two ways:

Via model weights (i.e., fine-tune the model on a
training set)

Via model inputs (i.e., insert the knowledge into
an input message)

Although fine-tuning can feel like the more natural
option—training on data is how GPT learned all of its other knowledge, after
all—we generally do not recommend it as a way to
teach the model knowledge. Fine-tuning is
better suited to teaching specialized tasks or styles, and is less reliable for
factual recall.

对于“Factual Recall”类型任务，OpenAI更倾向于Emedding Search。显然FT不是回答政策类问题的最佳方案。

之后，我们“干掉了”Langchain。Langchain + Vectorstore + LLMs是当下最“主流”的知识库方案。但在提升回答的准确度方面，该方案存在一些致命的缺陷，通过分解Langchain的架构，我们可以看到（图2）：

图2: Langchain + Vectorstore + LLMs

首先，在第2步，Langchain的全自动文件切片（Chunk）无法保证语义的准确性。无论如何设置，Langchain的文件切片工具：RecursiveCharacterTextSplitter或CharacterTextSplitter都可能将位置上相邻、但语义上应当区隔的内容放到一个切片中，从而导致最终回答中出现不相干、甚至相反的内容。

其次，在第7步，Langchain无法依据不同的问题提取合适数量的切片。即使切片内容准确，不同问题对切片数量的要求也不同。例如，用1个切片就可以回答诸如“我有租房发票，如何在北京进行线下公积金提取？”等比较明确问题，而需要用更多的切片回答“我想买房，如何提取公积金?”之类较为概括性的问题。Langchain并不具备依据不同的问题来提取不同数量的切片的能力。

再次，在第8步，Langchain将已经很精准的答案送往大模型纯属“画蛇添足”。对于政策型问题，往往原文已经是最佳答案，而大模型对原文的“二次加工”将破坏其语义和完整性，导致准确度下降。

上述原因让我们“抛弃”了FT和Langchain，并改向基于大模型的Classification能力，通过“追问型”Agent满足“政策型”文档问答的需求。

“追问型”Agent

通常而言，对于政策型问题，用户的初始问题和最终其想要的答案之间可能存在鸿沟，而追问就是弥补鸿沟，并提升回答准确性的有效手段。例如，我和HR专员可能有这样的对话：

我：租房能取公积金吗？

HR专员：可以，租房提取公积金需要你的公积金缴存满三个月，且没有自有住房。满足条件的话，你可以申请。你在哪租房？

我：海淀。

HR专员：你有租房发票吗？

我：有。

HR专员：你打算网上申请还是去网点？

我：网申吧，省得麻烦...

HR专员：好，这是网申的方法....

实际上HR专员的脑中有一棵“决策树”，通过不断的“追问”，其可以在决策树中选择正确的路径并最终反馈给我准确的材料。

而模拟这个过程将能够有效提升Agent对政策型问题回答的准确性。受到Autonomous Agent架构的启发（图3），我们将模拟过程切分为三步：

图3. Agent架构

第一步，构建决策树（Planning）。政策型文档往往本身蕴含着知识图谱-决策树，而Agent的工作是将其清晰地描绘出来以应对和引导用户的意图，例如，对于公积金租房场景，Agent可以构建一棵这样的决策树（图4）：

图4:租房场景决策树

第二步，挂接知识（Memory）。对于决策树中的每一个节点，Agent都可以挂接一个或者多个知识（文档切片）。这些切片就像是Agent的Short-term Memory，帮助Agent回答问题。我们需要保证挂接切片的准确性，以实现“只要Agent能够正确判断用户的意图，就能反馈准确的内容”（图5）。