最近一直在做检索增强和问答系统的内容。今天给大家分享一些关于LLM和知识库相结合的技巧和使用场景,提供一些信息增量。江湖经验,欢迎有相关经验和感兴趣的伙伴一起交流学习。
一、问答系统简介
问答系统(Question Answer system,简称 QA system),其核心就是用简要的自然文本回答用户提出的问题。
QA 的难度主要在于回答这个问题所依赖的信息在长文档中的分布情况,具体可以大致分为下面三种情况:
(L1)相关信息可以出现在不超过一个固定上下文长度(512)限制的 block 以内
(L2)相关信息出现在多个 block 里,出现的总数不超过 5-10 个,最好是能在大模型支持的有效长度内。
(L3)需要考虑多个片段,并综合结合上下文信息甚至全部内容才能得到结果
受限于数据集的大小和规模以及问题的难度,目前主要研究偏向于 L1 和 L2 。由于高质量的中文数据集整理会耗费很大的人力、物力、现在 QA 问题的数据集,文档长度普遍偏低,而且规模较小和实际应用相差甚远。当然,现在借助大模型的能力,能比较好的整合和回答 L1 类问题,L2 类问题也有比较不错的结果,但对于 L3 类问题,如果所涉及到的片段长度过长,还是无法做到有效的回答。
二、基于LLM的问答系统架构
目前比较常见的开源 LLM 的问答系统都会遵循下图这种结构去进行设计,整个流程用文字表述如下:加载文件 -> 读取文本 -> 文本分割 -> 文本向量化 -> 问句向量化 -> 在文本向量中匹配出与问句向量最相似的 top k
个 -> 匹配出的文本作为上下文和问题一起添加到 prompt
中 -> 提交给 LLM
生成回答。
LangChain+ChatGLM 框架示意图
参照之前,我们将之前的 QA 问题的过程进行细化,梳理出更详细的解决方案:
1)将长文档以某种方式切分为若干 block,每个 block 大小一般会小于向量模型能处理的最大上下文。切分方式可以简单也可以是一个复杂的策略。比如,相邻的 block 之间可以保留重复的内容,用来降低简单切分方式造成的信息损失,并进一步补充 block 的上下文信息。
2)将每个 block 映射为一个vector。对于每个问题,也通过相同的方式与 block 进行相关性匹配,或者其他方式进行匹配。选择最相关的 1个 或者多个 block,加上问题,组合为 prompt 输入给 LLM,让其进行回答。
对于L2的问题,所需的 block 数量比较多,再加上提示词模板,很容易超过 LLM 的长度,因此需要进行信息的压缩和回溯。具体一个可选方案就是:
1)每一个 block 和提问的问题进行组合输入给 LLM,然后返回相关信息。
2)将多个相关信息进行汇总和筛选,总结,将相关性较高的内容作为新的 context 输入给 LLM 进行回答。
三、问题分析与解决方案
根据上述的方案设计,我们会很容易发现几个问题。虽然L1 类问题解决比较容易,但是 L2 类问题涉及到了信息的压缩和整合,如果多次调用 LLM,则代价很大,而且时间开销不低。
整个向量检索到最后 LLM 生成的整体链路很长。需要对文本切分,然后向量化之后存储到向量数据库中。然后需要进行 KNN 检索最相似的 K 个文本片段,拼接成提示词去调用 LLM。这几个步骤都有可能出现问题,导致带来连锁反应,使得最后结果的可信度满足不了客户效果。
从整体流程梳理,细节的问题主要集中在下面几点:
1)文本切分,切分策略需要按数据集进行适配和微调,没有低成本、高质量的方案
2)向量模型的语义表达能力问题,长文本的召回率和相关性一般
3)作为基座的 LLM,能力不够,天花板太低
针对出现的问题,提供几个可能的解决方案和优化措施,经过我们实验,能在一定程度上缓解这些问题。
一)文本切分
目前对文本切片的方案,主要有两种,一种就是基于策略规则,另外一种是基于算法模型。同时,文本切分的策略也和向量模型息息相关,建议同时考虑向量模型的建模能力去设定切分的方案。例如,BERT 模型能处理的最大文本长度为 512,但切分时并不一定按照 512 去切分。
基于策略规则,主要有几种不同的策略可以供选择:
- 截断 截取前 510 个或后 510 个或前 128 + 后 382
- 分段 分段 k=L/510, 然后各段可以求平均、求 max
- 滑动窗口 (Sliding window),即把文档分成有重叠的若干段,然后每一段都当作独立的文档送入BERT进行处理。最后再对于这些独立文档得到的结果进行整合
Sliding window 可以使用 LangChain 里的 RecursiveCharacterTextSplitter 或者 LangChain-Chatchat 实现的ChineseRecursiveTextSplitter 进行切分,对中文相比较友好一些。重叠的长度和窗口长度需要根据实际进行调整,没有特别明确的规则,可以根据文档的平均长度为参考基准。核心可以参考的几个指标有,响应时间、检索相关性以及产生幻觉的次数。
基于算法模型,主要是使用类似 BERT 结构的语义段落分割模型,能够较好的对段落进行切割,并获取尽量完整的上下文语义。需要微调,上手难度高,而且切分出的段落有可能大于向量模型所支持的长度,这样就还需要进行切分。
二)向量模型
向量模型核心是召回大量和查询的 query 相似的语义片段,如果想减少或者优化召回后的语义片段数量,还需要和排序模型相结合使用。现在常用的向量模型结构都比较类似,存在语义表示不足的问题。
目前,向量模型都会基于 BERT 作为基座模型再进行微调,而 BERT 最大长度仅支持到 512;超过 512 的时候,性能会急剧下降,再使用的意义不大,也制约了语义表示的质量。比较主流的向量模型,例如 text2vec 支持的长度仅为128;bge、m3e 等语义模型支持长度均为 512。
还需要注意的是,有的向量模型使用的数据集,文档长度都没有达到 512,这就意味着这种模型实际有效的文档长度并不是 512,因此做文本切分的时候还是需要根据实际情况进行确定。
向量模型的优化一般有两个思路,从适用性和实现难度上来讲,选择对文本进行切分的代价和成本都比较低,这种方法也是目前主流的 LLM+QA 系统常用的方案。第二种就是直接对向量模型进行改进,优化模型结构使得可以支持更大的上下文信息,以便生成更高质量的语义向量。
文本切分的策略可以参考第一小节,对文本切分之后的处理方式一般有两种,一种是将不同组块视为不同的向量,然后进行召回;另外一种是,将不同组块拼接为一个向量,可以取平均、取最值或者拼接向量。这两种方案在实现上难度比较接近,如果后续接入 LLM,一般会为了召回更相关的内容,用方式一的比较多。如果是做检索和排序任务,一般会检索整篇文章,方式二可能会更好一些。
直接对向量模型底层的 BERT 或者 Transformer 结构进行改进,会使向量模型支持更长的上下文,简单来说会有下面几种不同的方式,在具体任务上的表现也需要视情况而定。
- 修改模型结构,比如Transformer-XL,魔改了 Transformer 的注意力机制,天然支持长文本序列。
- 层次编码机制,由句子生成段落表示,然后由段落生成文章的表示。
- 利用模型进行信息压缩和筛选,比如CogLTX。CogLTX 的关键步骤是将文本拆分成多个 block,然后识别关键的文本 block,在 CogLTX 中叫做 MemRecall。CogLTX 中有另一个 BERT 模型,叫做 judge,用来给 block 的相关性进行打分。
目前主流的中文向量模型有 m3e、bge、text2vec 等,从实际使用效果来看,可以优先选择 bge,由于整个链路的性能瓶颈,主要在 LLM 一侧,向量模型耗时可以忽略不计。bge-v1.5 的效果没有原来的版本高,不建议使用。
如果有自己的数据集,建议按照官方说明,微调 bge。相关实验参数建议自己根据实际进行修改,难负例的挖掘十分重要,数量控制在 2-8 个为宜;batch size 尽可能的越大越好,微调的 epoch 不建议过大,1-2个 epoch 为宜。
如果,直接用 bge 召回效果不太满意,可以考虑后续用 BM25 算法去进一步优化召回的得分。 BM25 实现简单,可以作为 baseline,后续有能力再接入排序模型也是很好的选择。
三)基座LLM选择
LLM 是基石,选择一个质量较高的模型是关键。从成本和使用效果两个方面来分析,如果不考虑成本因素,首选GPT4,其次 GPT3.5。根据我们实际测试的几个大模型的效果,和 GPT3.5 还是有所差距。
考虑有本地部署的需求,我们使用了 ChatGLM 和 baichuan,实测下来两者效果差不太多。6B 参数量级的模型差强人意,去微调模型或者是优化提示词能带来的提升比较有限,有能力的最好部署参数量更大的模型(大于等于 13B)。
四)效果评价
目前去评价整体的 QA 系统是有相关测试数据集可以评价的,但是对于 LLM 生成的回答用机器的方法去评价会有遗漏的地方,还是需要人工介入更好。建议整体上对效果的评价分三步走,第一步选出最好的向量模型,第二步选择一个最优的基座 LLM 模型,第三步优化提示词。
第一步,对于向量模型的效果,可以使用专门用于检索的数据去评价,也可以自己制作相关的数据集去评选出最优的向量模型。
第二步,使用相同的问题和检索的上下文构建提示词,分别对不同的 LLM 模型输出的结果进行盲审,评选出最优的LLM 模型。
第三步,使用向量模型召回的 context 和人工筛选出的 context 去利用 LLM 生成回答,并对回答的结果进行验证,定位效果瓶颈。如果人工筛选的 context 结果明显好于向量模型的,则瓶颈在向量召回上,应该去优化向量模型。如果人工筛选的 context 和向量召回的 context 效果差距不大,则瓶颈在 LLM,应该去使用更好的 LLM 或者尝试优化提示词。
如何系统的去学习大模型LLM ?
作为一名热心肠的互联网老兵,我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。
但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的 AI大模型资料
包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。
😝有需要的小伙伴,可以V扫描下方二维码免费领取🆓
一、全套AGI大模型学习路线
AI大模型时代的学习之旅:从基础到前沿,掌握人工智能的核心技能!
二、640套AI大模型报告合集
这套包含640份报告的合集,涵盖了AI大模型的理论研究、技术实现、行业应用等多个方面。无论您是科研人员、工程师,还是对AI大模型感兴趣的爱好者,这套报告合集都将为您提供宝贵的信息和启示。
三、AI大模型经典PDF籍
随着人工智能技术的飞速发展,AI大模型已经成为了当今科技领域的一大热点。这些大型预训练模型,如GPT-3、BERT、XLNet等,以其强大的语言理解和生成能力,正在改变我们对人工智能的认识。 那以下这些PDF籍就是非常不错的学习资源。
四、AI大模型商业化落地方案
阶段1:AI大模型时代的基础理解
- 目标:了解AI大模型的基本概念、发展历程和核心原理。
- 内容:
- L1.1 人工智能简述与大模型起源
- L1.2 大模型与通用人工智能
- L1.3 GPT模型的发展历程
- L1.4 模型工程
- L1.4.1 知识大模型
- L1.4.2 生产大模型
- L1.4.3 模型工程方法论
- L1.4.4 模型工程实践 - L1.5 GPT应用案例
阶段2:AI大模型API应用开发工程
- 目标:掌握AI大模型API的使用和开发,以及相关的编程技能。
- 内容:
- L2.1 API接口
- L2.1.1 OpenAI API接口
- L2.1.2 Python接口接入
- L2.1.3 BOT工具类框架
- L2.1.4 代码示例 - L2.2 Prompt框架
- L2.2.1 什么是Prompt
- L2.2.2 Prompt框架应用现状
- L2.2.3 基于GPTAS的Prompt框架
- L2.2.4 Prompt框架与Thought
- L2.2.5 Prompt框架与提示词 - L2.3 流水线工程
- L2.3.1 流水线工程的概念
- L2.3.2 流水线工程的优点
- L2.3.3 流水线工程的应用 - L2.4 总结与展望
- L2.1 API接口
阶段3:AI大模型应用架构实践
- 目标:深入理解AI大模型的应用架构,并能够进行私有化部署。
- 内容:
- L3.1 Agent模型框架
- L3.1.1 Agent模型框架的设计理念
- L3.1.2 Agent模型框架的核心组件
- L3.1.3 Agent模型框架的实现细节 - L3.2 MetaGPT
- L3.2.1 MetaGPT的基本概念
- L3.2.2 MetaGPT的工作原理
- L3.2.3 MetaGPT的应用场景 - L3.3 ChatGLM
- L3.3.1 ChatGLM的特点
- L3.3.2 ChatGLM的开发环境
- L3.3.3 ChatGLM的使用示例 - L3.4 LLAMA
- L3.4.1 LLAMA的特点
- L3.4.2 LLAMA的开发环境
- L3.4.3 LLAMA的使用示例 - L3.5 其他大模型介绍
- L3.1 Agent模型框架
阶段4:AI大模型私有化部署
- 目标:掌握多种AI大模型的私有化部署,包括多模态和特定领域模型。
- 内容:
- L4.1 模型私有化部署概述
- L4.2 模型私有化部署的关键技术
- L4.3 模型私有化部署的实施步骤
- L4.4 模型私有化部署的应用场景
学习计划:
- 阶段1:1-2个月,建立AI大模型的基础知识体系。
- 阶段2:2-3个月,专注于API应用开发能力的提升。
- 阶段3:3-4个月,深入实践AI大模型的应用架构和私有化部署。
- 阶段4:4-5个月,专注于高级模型的应用和部署。
这份完整版的大模型 LLM 学习资料已经上传CSDN,朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费
】
😝有需要的小伙伴,可以Vx扫描下方二维码免费领取🆓