01 双编码器与交叉编码器

自己的感悟:RAG,检索增强生成并非一个新技术,只是对于原有搜索算法在不同场景和技术的升级。举个例子,百度和gogle作为一个搜索引擎,同样采用了bm25和tf-idf等传统匹配算法。只是随着最近大语言模型的发展,多了一个增强生成的阶段和更细粒度的拆分。

Bi-Encoder vs Cross-Encoder

这里需要理解两种编码器的区别:

  • Bi-Encoder(双编码器):query 和 chunk 分别编码成向量,然后计算向量相似度。这就是 Embedding 模型的工作方式。优点是速度快,可以提前把所有 chunk 编码好存起来,查询时只需要编码 query;缺点是精度有限,因为 query 和 chunk 是独立编码的,无法捕捉它们之间的细粒度交互关系。

  • Cross-Encoder(交叉编码器):把 query 和 chunk 拼接在一起(比如 [CLS] query [SEP] chunk [SEP]),一起输入模型,模型能看到 query 和 chunk 的完整交互,输出一个相关性分数。优点是精度更高,能捕捉更细粒度的语义关系;缺点是速度慢,每个 (query, chunk) 对都要过一遍模型。

重排序通常用 Cross-Encoder,因为候选集已经很小了(比如 20~50 个),可以接受慢一点的速度,换取更高的精度。

实际上双编码器是将数据隐射到一个高维的坐标系中,通过一个多维度向量表示,然后通过一些距离算法计算匹配度。

而交叉编码器是通过数据进入 Transformer 编码器后,Attention 机制会计算序列中每一个 Token 与其他所有 Token 的关联权重 。Query 这边的"苹果"可以直接"看到" Chunk 那边的"水果"和"一种",同时 Chunk 那边的"水果"也能看到 Query 这边的"好吃"。它捕捉了所有词与词之间两两的交互关系。

所以交叉编码器比双编码器计算的相似度更加准确。

  • 双编码器先压缩(降维成向量),后交互(算距离)

    交互发生在压缩之后,用的是残缺的、丢失了细节的"压缩包"去碰运气。

  • 交叉编码器先交互(全注意力),后压缩(输出分数)

    在交互阶段,512 个 Token 完整保留,带着所有细节去和 Query 的每一个字进行"全场一对一辩论"。直到辩论结束,模型掌握了所有细微线索后,才在最后一层通过 [CLS] 将其压缩成一个分数。

为什么不直接用 Cross-Encoder 做检索

你可能会问:既然 Cross-Encoder 精度更高,为什么不直接用它做检索,还要搞两阶段?

因为太慢了。假设你的知识库有 100 万个 chunk,用户提问时,你需要把这 100 万个 chunk 逐个和 query 拼接起来过 Cross-Encoder,这需要 100 万次模型推理,延迟和成本都不可接受。

所以工程上一定是两阶段策略:

  • 1.粗检索(Bi-Encoder):快速从 100 万个 chunk 中召回 Top-20 或 Top-50,延迟低,覆盖面广
  • 2.精排序(Cross-Encoder):对这 20~50 个候选逐个打分,延迟可接受,精度高

这就是快召回 + 慢精排的核心思想。

注意嵌入向量生成其实比rerank慢很多,两三倍的差距,但是生成嵌入向量后再进行相似度计算,所需要的时间只有交叉编码器生成的百分之一

相关推荐
阿拉斯攀登21 小时前
AI数据助手:从文档问答到智能数据分析
人工智能·数据分析·embedding·知识库·rag·企业知识库·增强检索
AlfredZhao2 天前
一篇搞定:用 curl 测试私有部署模型联通性
llm·embedding·model·curl
闵孚龙19 天前
常用网络层:Linear、Conv、RNN、Embedding、Transformer
rnn·transformer·embedding
老刘说AI19 天前
类Sora模型:解锁动态视觉艺术的密码
人工智能·stable diffusion·架构·embedding
沪漂阿龙20 天前
Embedding:文本怎么变成向量?语义检索为什么能工作?
人工智能·python·embedding
大模型任我行21 天前
谷歌:多模态嵌入Gemini Embedding 2
人工智能·语言模型·embedding·论文笔记
kisdiem22 天前
Embeddings 到底是什么
embedding·rag
me83222 天前
【AI面试】小白理解大模型:关于RoPE 旋转位置嵌入
人工智能·ai·embedding
大模型推理23 天前
《Nano-vLLM 源码解读》第 18 篇 · Embedding 与 LM Head
embedding·vllm