RAG 检索质量从 60% 到 90%:混合检索 + 重排序的完整实践

27 阅读4分钟

纯向量检索用 Embedding 模型,用户问「振动异常」,知识库写「振动超标」——搜不到。混合检索就是答案。


单一检索的局限

上篇我搭好了 RAG 管道:文本分块 → Embedding → Milvus → 检索。跑 10 条测试查询,纯向量检索的命中率约 60%。

60% 的意思是:10 个问题中有 4 个搜不到正确答案。

为什么搜不到?

用户问:「传送带跑偏了怎么调?」
知识库写:「传送带跑偏调整方法:1)调节张紧装置...」

Embedding 模型觉得这两句话不太像 → 向量距离远 → 检索失败。

Embedding 模型理解「语义」,但它不完全懂中文口语和专业术语的细微差异。这是所有 Embedding 的共性问题——尤其在使用多语言通用模型时。


方案:向量 + 关键词双路召回

Dense(向量检索) 擅长语义匹配——「温度高」能匹配「发热异常」。 Sparse(关键词检索) 擅长精确匹配——「传送带」必须命中「传送带」。

两者互补。这就是混合检索(Hybrid Search)。

步骤 1:加 BM25 稀疏检索

BM25 是经典的 TF-IDF 变体,核心逻辑是「词在少数文档中出现频率高 = 该文档和这个词相关」。

我在 Bm25Retriever 中实现了一个简化版:

Bm25Retriever bm25 = new Bm25Retriever();
bm25.index(documents);

// 用户问 "传送带跑偏"
List<ScoredDoc> results = bm25.search("传送带跑偏怎么调", 10);
// → 能精确匹配到包含「传送带」和「调」的文档

步骤 2:RRF 融合

两路检索结果怎么合并?我用了 RRF(Reciprocal Rank Fusion)——一种简单但有效的融合算法:

RRF_score(d) = sum over rankings: 1 / (k + rank(d))
k = 60(经典取值)

每个文档在两个排行榜中的排名越靠前,RRF 分数越高。

RrfFusion fusion = new RrfFusion();
List<String> fused = fusion.fuse(denseResults, sparseResults, 5);

步骤 3:LLM 重排序

拿到融合后的 Top-5 候选,再用 LLM 精排——这是最后一公里。虽然这次编码里没上专门的 Cross-Encoder,但原理一样:候选文档 + 用户问题 → LLM 判断相关性 → 选 Top-3


实验数据:三种策略对比

我在 10 条测试查询上跑了三种策略:

策略命中率MRR
Dense Only(纯向量)~60%0.42
BM25 Only(纯关键词)~70%0.48
RRF Fused(混合检索)~90%0.68

Dense Only 输在哪?「传感器信号漂移」这条,Embedding 模型没理解「漂移」=「零点漂移」,搜到了不相关的电气故障文档。

BM25 Only 输在哪?「CNC 主轴异响」这条,知识库里写的是「CNC主轴异响」,BM25 对「异响」和「异常噪音」这种同义词完全没辙。

RRF Fused 为什么赢? 两种检索互补——向量错了的,关键词能兜底;关键词漏了的,向量能补上。

hybrid-retrieval.png

代码:在 Agent 中启用混合检索

我把整个混合检索管线封装成了带评估的端到端实验:

# 1. 摄入知识库
curl -X POST http://localhost:8080/api/rag/ingest/500

# 2. 运行评估(Dense vs BM25 vs Fused)
curl -X POST http://localhost:8080/api/rag/evaluate

# 返回:
# {
#   "DENSE_hitRate": "60.0%",  "DENSE_MRR": "0.420",
#   "BM25_hitRate": "70.0%",   "BM25_MRR": "0.480",
#   "FUSED_hitRate": "90.0%",  "FUSED_MRR": "0.680"
# }

什么时候需要混合检索?

场景推荐策略
知识库 < 100 条,查询单一纯 Dense 就够了
包含专业术语/编号(如 CNC-001)加 BM25
用户用口语提问加 BM25 + Query Rewriting
多语言混用纯 Dense(BM25 分语言要调分词器)

判断标准很简单:跑 10 条测试查询看命中率。如果 Dense Only 低于 70%,加 BM25。加了之后还低于 80%,加 Query Rewriting。


一句话总结

向量检索懂语义但不懂关键词,BM25 懂关键词但不懂语义。RRF 融合让两个互相兜底,90% 命中率是混合检索的基本盘。剩下的 10%,靠 Query Rewriting 和更精准的分块。


代码仓库:github.com/LaoLiang-ag…


本文由 LaoLiang 原创,首发于掘金/知乎/微信公众号。转载请联系作者。