「Datawhale」RAG技术全栈指南 Task 3

向量嵌入

概念

embedding是将真实世界中复杂、高维的数据对象转换为数学上易于处理的、低维、稠密的连续数值向量的技术。embedding的核心意义在于,它所构建的向量空间中,语义相似的对象得到的向量距离会更近,语义上不相关的对象向量距离会更远。
理解:

embedding是将文本/图像等人类能看懂的数据转换成计算中可量化的数据。

如何衡量一个embedding模型的好坏,是看相近词的向量相似度是否更高,无关词的向量相似度更低。
如何度量相似度?

  • 余弦相似度/点积/欧氏距离
    如何选择不同的度量方法?

  • 余弦相似度(Cosine)

适合:语义相似、主题相近、同义表达匹配

○ 文本检索(FAQ/客服知识库/代码搜索/文档搜索)

○ 聚类/去重(同一语义不同写法)

○ 跨域长度差异大(短 query vs 长文档)

  • 点积

○ 在做L2归一后,点积 = 余弦相似度

  • 欧式距离

适合:向量在空间里的"位置差"有明确意义

○ 图像特征、度量学习(某些 triplet loss / contrastive loss 直接用 L2)

○ 下游算法天然基于距离(k-means 常用欧氏)

○ 在做异常检测/漂移检测,关心"偏离中心点有多远"

Embedding在RAG中的作用

  • 语义检索的基础:所有需要和用户输入问题相关的查询/检索都是需要用embedding来进行表示和计算。

多模态embedding

  • 多模态embedding表现形式是一致的,主要是将图片和文字的向量对齐在同一空间内。

向量数据库

解决的问题:如何快速、准确地从海量向量中找到与用户查询最相似的几个。

索引机制详解:

HNSW(Hierarchical Navigable Small World)

核心思想:把向量构造成多层的"近邻图"(小世界图)。查询时从顶层快速跳转到接近目标的区域,再在底层做更精细的图搜索。

适用场景:通用的向量检索,尤其是需要高召回 + 低延迟、数据规模中到大(百万级常见)。

优点

召回率高、延迟低,通常是工程上最"省心"的默认选择

支持多种距离(L2、inner product、cosine(通常通过归一化+IP))

增量插入相对友好(动态更新比 IVF 稳)

缺点

内存占用较大(图结构需要额外边)

构建时间可能较长;参数(如 M, efConstruction, efSearch)需要调优。

IVF(Inverted File Index,倒排文件/聚类分桶)

核心思想:先用 k-means 把向量分成很多"簇"(centroids),查询时先找离 query 最近的若干簇(nprobe),只在这些簇的候选集合里做精搜/重排。

适用场景:超大规模(百万到十亿级),希望用"先粗后细"控制计算量。

优点

可扩展性强,适合大数据量

通过 nlist(簇数)和 nprobe(探测簇数)在速度/召回之间做权衡

缺点

需要训练聚类中心(训练成本 + 数据分布变化会影响效果)

对动态更新/频繁插入不如 HNSW 友好(实现相关)

召回对 nprobe 很敏感:nprobe 小容易漏,nprobe 大延迟上升

练习:

  1. LlamaIndex默认会将数据存储为透明可读的JSON格式,运行03_llamaindex_vector.py文件,查看保存的json文件内容。

• docstore.json

作用:保存内容本体。主要包括:

传入的 Document(text=...)

以及由 Document 转换/切分出来的 Node(例如 TextNode)

node_id → node 的映射(文本、metadata、关系)

理解:检索到候选 node_id 后,最终要用它来取回原文片段,作为上下文喂给 LLM。

• default__vector_store.json

作用:保存向量索引"真正用于相似度检索的数据",通常包括:embedding 向量(浮点数组)和 node_id(向量对应哪个节点)

• index_store.json

作用:保存"索引的结构与配置",例如:index_id、索引类型(VectorStoreIndex)索引里包含哪些 node_id,一些索引的组织结构信息(不同 Index 类型差异很大。

  1. 新建一个代码文件实现对LlamaIndex存储数据的加载和相似性搜索。

API网址:https://llamaindex.readthedocs.io/zh/latest/reference/query/retrievers/vector_store.html

有两种写法:

复制代码
retriever = index.as_retriever(similarity_top_k=3)
nodes = retriever.retrieve("张三是谁?")

for n in nodes:
    print(n.score, n.node.text)

直接写retriever用的是Llamaindex里面python实现的类似向量库查询的办法。

Faiss的写法:

复制代码
from llama_index.core import VectorStoreIndex, StorageContext, Settings
from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.vector_stores.faiss import FaissVectorStore
import faiss

texts = [
    "张三是法外狂徒",
    "FAISS是一个用于高效相似性搜索和密集向量聚类的库。",
    "LangChain是一个用于开发由语言模型驱动的应用程序的框架。"
]

Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-zh-v1.5")
dim = len(Settings.embed_model.get_text_embedding("测试"))

faiss_index = faiss.IndexFlatIP(dim)
vector_store = FaissVectorStore(faiss_index=faiss_index)
storage_context = StorageContext.from_defaults(vector_store=vector_store)

index = VectorStoreIndex.from_documents(texts, storage_context=storage_context)

retriever = index.as_retriever(similarity_top_k=1)
nodes = retriever.retrieve("FAISS是做什么的?")
for n in nodes:
    print(n.score, n.node.get_content())

Milvus介绍及多模态检索实践

直接看对应章节,本章写的很好。

https://github.com/datawhalechina/all-in-rag/blob/main/docs/chapter3/09_milvus.md

注意事项⚠️:

现在docker启动Milvus后再运行代码,不然就会

索引优化

上下文扩展(句子窗口检索)

思路:为检索精确性而索引小块,为上下文丰富性而检索大块。

工作流程:

索引阶段:用句子作为索引,同时将句子的上下文窗口都存到元数据中,即该句子原文中的前N个和后N个句子。窗口文本不会作为索引,只是会作为元数据存储。

检索阶段:在单一句子节点上执行相似度搜索。

后处理阶段:用元数据,也就是包含检索句子的上下文数据来替换节点中原来单一的句子内容,传递给LLM。

我理解句子窗口检索的问题是根据位置来获取的上下文信息不一定是靠谱的,可以看一些其他论文看一下改进。

FreeChunker: A Cross-Granularity Chunking Framework https://arxiv.org/pdf/2510.20356

Reconstructing Context: Evaluating Advanced Chunking Strategies for Retrieval-Augmented Generation https://arxiv.org/abs/2504.19754

结构化索引

为了解决文档库中存在大量文档的情况,利用结构化索引。其原理是在索引文本块的同时,为其附加结构化的元数据(Metadata)

这些元数据可以是任何有助于筛选和定位信息的标签,例如:

• 文件名

• 文档创建日期

• 章节标题

• 作者

• 任何自定义的分类标签

重点就是先元数据过滤,再在过滤后的数据后做向量搜索。

相关推荐
那个村的李富贵19 小时前
光影魔术师:CANN加速实时图像风格迁移,让每张照片秒变大师画作
人工智能·aigc·cann
腾讯云开发者20 小时前
“痛点”到“通点”!一份让 AI 真正落地产生真金白银的实战指南
人工智能
CareyWYR20 小时前
每周AI论文速递(260202-260206)
人工智能
hopsky21 小时前
大模型生成PPT的技术原理
人工智能
禁默1 天前
打通 AI 与信号处理的“任督二脉”:Ascend SIP Boost 加速库深度实战
人工智能·信号处理·cann
心疼你的一切1 天前
昇腾CANN实战落地:从智慧城市到AIGC,解锁五大行业AI应用的算力密码
数据仓库·人工智能·深度学习·aigc·智慧城市·cann
AI绘画哇哒哒1 天前
【干货收藏】深度解析AI Agent框架:设计原理+主流选型+项目实操,一站式学习指南
人工智能·学习·ai·程序员·大模型·产品经理·转行
数据分析能量站1 天前
Clawdbot(现名Moltbot)-现状分析
人工智能
那个村的李富贵1 天前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
二十雨辰1 天前
[python]-AI大模型
开发语言·人工智能·python