构建稳健RAG应用:文档索引与存储的深度解析

这部分笔记是关于RAG系统中文档索引与存储部分(下一篇关于检索与生成),感觉挺适合作为checklist用于对自己的系统查漏补缺的,同样先放一张整体的架构图:

image.png

1、Encoder

Encoder将查询输入的文本转换为向量(其实就是embedding),在RAG应用中这可能是最重要的一个部分,在embedding模型的选择和评估上可以参考下面的一些方法

  • MTEB Benchmark:这个benchmark对很多embedding模型进行了打分,并且列出来了这些模型的大小、输出维度等(包括openai的模型),可以作为一个参考:[huggingface.co/spaces/mteb…]
  • 自定义评估:如果数据集包含敏感数据,或者是很小众的领域,可能需要自定义一些评估手段以获得比较好的效果,有几种比较常用的方法: 通过标记来评估:生成一个数据集并对其打标签,使用检索的指标来定量评估(例如MRR和NDCG) 通过模型来评估:同样生成数据集并打标签,但是用LLM来评估embedding模型的相对性能,再对排名最高的三个进行手动筛选

除了效果之外,还有一些因素是需要关注的:

  • 查询和索引成本:例如早期用openai等第三方的api会更加简单且便宜,自己部署模型需要考虑gpu成本和延时,还需要自己操心运维
  • 向量数据库成本:如果文本量比较大,需要存储大量的embedding数据和文档片段,在选择embedding维度的时候需要仔细计算这方面的成本,向量数据库通常是按存储大小收费的(血泪教训+1)
  • 语言:如果用户会使用多语言,那么embedding最好也能支持多种语言
  • 搜索延迟:搜索的延迟与embedding的维度是线性增长的,如果对延迟要求比较高最好用输出维度比较低的embedding
  • 隐私:如果输入数据涉及到用户隐私,那么openai等公开api则不够安全

2、文档解析

在RAG应用中我们需要将外部的文档索引后进行存储,以便在查询的时候进行检索,一个好的RAG系统应该支持多种格式输入,例如pdf/word/excel等,除了文档本身的内容之外,还需要提取其他相关的信息:例如表格数据、图像的ocr识别、文件的meta信息、甚至文档中超链接的内容等,总之这部分内容挺繁琐的。

3、文档分块

选择合适的文档分块方法是至关重要的,如果太小则无法正确回答问题,如果太大则会有很多噪声,并且难以检索,这个视频在这方面做了很详细的介绍:[youtube.com/watch?v=qaP…]。对于特定领域的数据,还需要领域知识来进行调整,例如对于python的代码会基于def/class关键字来进行分块,可以参考这个文章:[rungalileo.io/blog/master…]

image.png

4、Indexer

Indexer负责管理文档的索引,和数据库系统很像,有很多类似的软件工程上的挑战:

  • Scale的问题:随着文档的增加,依旧要能快速的进行检索
  • 实时、一致性与原子性:如果一套系统会有大量并发的增删改查,这里会有挑战
  • 优化存储空间:存储大量的索引会有很大的存储成本,如何保持服务可用性的同时减少存储空间也是一种挑战
  • 监控:对于索引相关的问题需要有全面的监控,例如索引失败、资源瓶颈、索引过期等问题要及时发现

5、数据存储

系统中的数据存储包括以下几个部分:

  • embedding:单独存储在向量数据库中
  • 文档:pdf文件等,用s3等存储系统
  • 聊天记录/用户反馈等:存储在sql表中

6、向量数据库

向量数据库在整个系统中是很关键的一环,目前市面上有很多产品,这个网站对不同的向量数据库进行了详细的比较,可以参考下:superlinked.com/vector-db-c…。在选择向量数据库的时候需要考量下面这些因素:

  • recall和latency的tradeoff:不同的索引方式(Flat, HNSW, PQ等)有不同的倾向,根据自己的需求来选择

  • 成本:SaaS服务的向量数据库通常根据存储量大小和检索量来收费,同样也可以自己部署开源产品,但需要操心运维

  • 插入/查询速度的tradeoff:大多数情况下会优先考虑查询速度,需要对二者都做压测

  • 内存/硬盘存储:同样也是速度与成本之间的tradeoff,一些新的索引可以实现在硬盘上的高效索引(例如DiskANN中的Vamana)

  • 混合检索:单独的向量检索可能效果会不够好,有的向量数据库可以实现sparse和dense的混合检索,并提供一个超参数用于调整两边的权重,例如Pinecone,可以参考这篇文章:pinecone.io/learn/hybri… GKownpDbYAAULTe.jpg

  • 过滤:通常我们在检索的时候,会需要根据metadata进行一些过滤,这种过滤可分为检索前过滤和检索后过滤,检索前过滤可能会丢失一些相关结果,检索后过滤则会提高计算成本,尤其是满足过滤条件的数据占比较小的时候,有的向量数据库可以自定义过滤方式,例如Weaviate。

如何学习AI大模型?

我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述

第一阶段: 从大模型系统设计入手,讲解大模型的主要方法;

第二阶段: 在通过大模型提示词工程从Prompts角度入手更好发挥模型的作用;

第三阶段: 大模型平台应用开发借助阿里云PAI平台构建电商领域虚拟试衣系统;

第四阶段: 大模型知识库应用开发以LangChain框架为例,构建物流行业咨询智能问答系统;

第五阶段: 大模型微调开发借助以大健康、新零售、新媒体领域构建适合当前领域大模型;

第六阶段: 以SD多模态大模型为主,搭建了文生图小程序案例;

第七阶段: 以大模型平台应用与开发为主,通过星火大模型,文心大模型等成熟大模型构建大模型行业应用。

在这里插入图片描述

👉学会后的收获:👈
• 基于大模型全栈工程实现(前端、后端、产品经理、设计、数据分析等),通过这门课可获得不同能力;

• 能够利用大模型解决相关实际项目需求: 大数据时代,越来越多的企业和机构需要处理海量数据,利用大模型技术可以更好地处理这些数据,提高数据分析和决策的准确性。因此,掌握大模型应用开发技能,可以让程序员更好地应对实际项目需求;

• 基于大模型和企业数据AI应用开发,实现大模型理论、掌握GPU算力、硬件、LangChain开发框架和项目实战技能, 学会Fine-tuning垂直训练大模型(数据准备、数据蒸馏、大模型部署)一站式掌握;

• 能够完成时下热门大模型垂直领域模型训练能力,提高程序员的编码能力: 大模型应用开发需要掌握机器学习算法、深度学习框架等技术,这些技术的掌握可以提高程序员的编码能力和分析能力,让程序员更加熟练地编写高质量的代码。

在这里插入图片描述

1.AI大模型学习路线图
2.100套AI大模型商业化落地方案
3.100集大模型视频教程
4.200本大模型PDF书籍
5.LLM面试题合集
6.AI产品经理资源合集

👉获取方式:
😝有需要的小伙伴,可以保存图片到wx扫描二v码免费领取【保证100%免费】🆓

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值