RAG系统的混合检索工程:向量搜索与关键词搜索的最优融合

news2026/4/29 20:59:18
纯向量RAG的致命盲区当大多数工程师谈起RAG脑子里浮现的是Embedding → 向量数据库 → 相似度搜索 → 召回文档。这个流程简洁优雅2022-2023年的大量RAG教程都是这样教的。然而在生产环境中纯向量RAG暴露出了一个系统性问题**它擅长语义相似但不擅长精确匹配。**考虑这个场景用户问Error code ERR_CONN_REFUSED_7749 怎么处理“。纯向量搜索会找出关于网络连接错误的文档”但很可能找不到恰好包含ERR_CONN_REFUSED_7749这个精确错误码的文档因为那个文档在语义空间里可能离得很远。或者“v2.3.1-hotfix 的changelog是什么”——精确的版本号向量搜索基本无法正确处理。反过来纯关键词检索如BM25对语义变体束手无策“如何让代码跑得更快” vs “代码性能优化方法”——在关键词层面几乎没有重叠但语义几乎相同。混合检索Hybrid Retrieval正是为了解决这个问题而生。## 混合检索的核心原理混合检索将向量检索和关键词检索的结果通过融合算法合并取两者的优势。### 两种检索器的特性对比| 维度 | 向量检索Dense | 关键词检索Sparse/BM25 ||------|-------------------|--------------------------|| 擅长 | 语义相似性、同义词 | 精确词匹配、专有名词 || 不擅长 | 精确字符串匹配 | 语义变体、多义词 || 典型用例 | “如何优化内存使用” | “OOMKilledError” || 计算成本 | 中等向量内积 | 低倒排索引 || 索引大小 | 大每个chunk一个向量 | 小倒排索引 |### 互补性验证python# 用一个简单实验验证两种方法的互补性test_queries [ # 语义查询向量胜 (如何提升系统响应速度, 性能优化方案文档), (模型预测不准确怎么办, 模型调优最佳实践), # 精确查询BM25胜 (GPT-4o-2024-11-20 API参数, API文档-2024-11-20版), (ValueError: list index out of range, Python异常处理文档), # 两者都需要的混合查询 (用RAG处理PDF文档时遇到OOM怎么办, RAG内存优化指南),]## 混合检索的工程实现### 方案一RRFReciprocal Rank Fusion融合RRF是目前最主流的混合检索融合算法思想简单有效对每个文档分别计算它在向量检索排名和关键词检索排名中的倒数然后相加。pythonfrom typing import List, Dict, Any, Tupleimport mathclass HybridRetriever: 混合检索器融合向量检索和BM25关键词检索 使用RRFReciprocal Rank Fusion算法合并结果 def __init__( self, vector_store, # 向量数据库如Qdrant/Weaviate/pgvector bm25_index, # BM25索引如rank_bm25或Elasticsearch vector_weight: float 0.5, keyword_weight: float 0.5, rrf_k: int 60 # RRF常数通常60效果较好 ): self.vector_store vector_store self.bm25 bm25_index self.vector_weight vector_weight self.keyword_weight keyword_weight self.rrf_k rrf_k def rrf_score(self, rank: int, k: int None) - float: RRF得分 1 / (k rank) k通常设为60rank从1开始 k k or self.rrf_k return 1.0 / (k rank) def retrieve(self, query: str, top_k: int 10) - List[Dict[str, Any]]: 执行混合检索并融合结果 # 1. 向量检索 vector_results self.vector_store.similarity_search_with_score(query, ktop_k * 2) # 2. BM25关键词检索 keyword_results self.bm25.search(query, top_ktop_k * 2) # 3. RRF融合 scores: Dict[str, float] {} doc_map: Dict[str, Any] {} # 处理向量检索结果rank从1开始 for rank, (doc, similarity) in enumerate(vector_results, 1): doc_id doc.metadata.get(id, doc.page_content[:50]) scores[doc_id] scores.get(doc_id, 0) self.vector_weight * self.rrf_score(rank) doc_map[doc_id] doc # 处理关键词检索结果 for rank, result in enumerate(keyword_results, 1): doc_id result[id] scores[doc_id] scores.get(doc_id, 0) self.keyword_weight * self.rrf_score(rank) if doc_id not in doc_map: doc_map[doc_id] result[document] # 4. 按融合得分排序 sorted_ids sorted(scores.keys(), keylambda x: scores[x], reverseTrue) return [ { document: doc_map[doc_id], rrf_score: scores[doc_id], doc_id: doc_id } for doc_id in sorted_ids[:top_k] ]# 使用Qdrant BM25的完整实现from qdrant_client import QdrantClientfrom rank_bm25 import BM25Okapiimport numpy as npclass ProductionHybridRAG: 生产级混合RAG系统 使用Qdrant作为向量数据库rank_bm25做关键词检索 def __init__(self, qdrant_url: str, collection_name: str, embedding_fn): self.qdrant QdrantClient(urlqdrant_url) self.collection_name collection_name self.embed embedding_fn # BM25内存索引适合中小规模大规模用Elasticsearch self.bm25_corpus [] # 存储分词后的文档 self.bm25_docs [] # 存储原始文档 self.bm25_index None def add_documents(self, documents: List[str], ids: List[str] None): 添加文档到向量库和BM25索引 import jieba # 中文分词 # 1. 向量索引 embeddings self.embed(documents) points [ {id: ids[i] if ids else i, vector: emb.tolist(), payload: {text: doc}} for i, (doc, emb) in enumerate(zip(documents, embeddings)) ] self.qdrant.upsert(collection_nameself.collection_name, pointspoints) # 2. BM25索引增量更新 for doc in documents: # 中文文档用jieba分词英文文档按空格分词 tokens list(jieba.cut(doc)) if any(\u4e00 c \u9fff for c in doc) else doc.split() self.bm25_corpus.append(tokens) self.bm25_docs.append(doc) self.bm25_index BM25Okapi(self.bm25_corpus) def search(self, query: str, top_k: int 5) - List[dict]: 混合检索 import jieba # 向量检索 query_vector self.embed([query])[0] qdrant_results self.qdrant.search( collection_nameself.collection_name, query_vectorquery_vector.tolist(), limittop_k * 2 ) # BM25检索 query_tokens list(jieba.cut(query)) if any(\u4e00 c \u9fff for c in query) else query.split() bm25_scores self.bm25_index.get_scores(query_tokens) bm25_top_indices np.argsort(bm25_scores)[::-1][:top_k * 2] # RRF融合 scores {} doc_contents {} for rank, hit in enumerate(qdrant_results, 1): doc_id str(hit.id) scores[doc_id] scores.get(doc_id, 0) 1.0 / (60 rank) doc_contents[doc_id] hit.payload[text] for rank, idx in enumerate(bm25_top_indices, 1): doc_id fbm25_{idx} scores[doc_id] scores.get(doc_id, 0) 1.0 / (60 rank) doc_contents[doc_id] self.bm25_docs[idx] sorted_ids sorted(scores.keys(), keylambda x: scores[x], reverseTrue) return [ {text: doc_contents[doc_id], score: scores[doc_id]} for doc_id in sorted_ids[:top_k] ]### 方案二动态权重调整不是所有查询都应该用固定的向量:关键词 5:5。根据查询类型动态调整权重更有效pythonclass AdaptiveHybridRetriever: 自适应混合检索根据查询类型动态调整向量/关键词权重 # 查询类型特征 EXACT_MATCH_SIGNALS [ r\b[A-Z_]\d[A-Z_]*\b, # 错误代码 如 ERR_7749 rv\d\.\d, # 版本号 如 v2.3.1 r[A-Za-z]Error, # 异常类型 如 ValueError r[^], # 精确引用 r[^], # 代码片段 ] def detect_query_type(self, query: str) - dict: 检测查询类型决定权重 import re exact_match_score 0 for pattern in self.EXACT_MATCH_SIGNALS: if re.search(pattern, query): exact_match_score 1 # 存在精确匹配信号时提高关键词权重 if exact_match_score 2: return {vector_weight: 0.2, keyword_weight: 0.8, type: exact} elif exact_match_score 1: return {vector_weight: 0.4, keyword_weight: 0.6, type: mixed} else: return {vector_weight: 0.7, keyword_weight: 0.3, type: semantic} def retrieve(self, query: str, top_k: int 5) - List[dict]: config self.detect_query_type(query) # 使用动态权重执行检索 return self._hybrid_search( queryquery, top_ktop_k, vector_weightconfig[vector_weight], keyword_weightconfig[keyword_weight] )## 查询改写增强混合检索在检索前对查询进行改写可以进一步提升召回率pythonclass QueryAugmentedRetriever: 查询增强检索先改写查询再执行混合检索 策略多路查询 结果去重 def __init__(self, hybrid_retriever: HybridRetriever, llm_client): self.retriever hybrid_retriever self.client llm_client def generate_query_variants(self, original_query: str, n: int 3) - List[str]: 生成查询变体覆盖不同的表达方式 response self.client.chat.completions.create( modelgpt-4o-mini, messages[{ role: user, content: f为以下查询生成{n}个不同的表达方式帮助从知识库中检索相关内容。每个变体占一行包含原始查询本身。原始查询{original_query}要求1. 保持语义相同改变表达方式2. 可以更具体也可以更宽泛3. 包含可能的同义词或技术术语 }] ) variants_text response.choices[0].message.content variants [v.strip() for v in variants_text.split(\n) if v.strip()] return [original_query] [v for v in variants if v ! original_query][:n-1] def retrieve_with_augmentation(self, query: str, top_k: int 5) - List[dict]: 多路查询 去重融合 query_variants self.generate_query_variants(query) all_results {} for variant_rank, variant in enumerate(query_variants): results self.retriever.retrieve(variant, top_ktop_k) for result_rank, result in enumerate(results, 1): doc_id result[doc_id] # 多路查询的分数融合 if doc_id not in all_results: all_results[doc_id] {document: result[document], score: 0} # 靠前的查询变体权重更高 variant_weight 1.0 / (1 variant_rank * 0.5) all_results[doc_id][score] result[rrf_score] * variant_weight # 按融合得分排序 sorted_results sorted( all_results.values(), keylambda x: x[score], reverseTrue ) return sorted_results[:top_k]## 性能评估体系混合检索的效果需要定量评估pythonclass HybridRAGEvaluator: 混合检索系统评估器 指标召回率K、MRR平均倒数排名、NDCG def evaluate( self, retriever, test_dataset: List[dict] # [{query: ..., relevant_doc_ids: [...]}] ) - dict: recall_at_5 [] mrr_scores [] for item in test_dataset: results retriever.retrieve(item[query], top_k5) retrieved_ids [r[doc_id] for r in results] relevant_ids set(item[relevant_doc_ids]) # 召回率5 hits len(set(retrieved_ids) relevant_ids) recall_at_5.append(hits / len(relevant_ids) if relevant_ids else 0) # MRR mrr 0 for rank, doc_id in enumerate(retrieved_ids, 1): if doc_id in relevant_ids: mrr 1.0 / rank break mrr_scores.append(mrr) return { recall5: sum(recall_at_5) / len(recall_at_5), mrr: sum(mrr_scores) / len(mrr_scores), test_size: len(test_dataset) } def compare_approaches(self, test_dataset, retrievers: dict) - dict: 对比多种检索策略的效果 comparison {} for name, retriever in retrievers.items(): comparison[name] self.evaluate(retriever, test_dataset) # 排名 sorted_by_recall sorted( comparison.items(), keylambda x: x[1][recall5], reverseTrue ) return { results: comparison, ranking: [name for name, _ in sorted_by_recall], best: sorted_by_recall[0][0] }## 生产部署建议Qdrant向量 Elasticsearch关键词是目前最成熟的混合检索技术栈yaml# docker-compose.ymlservices: qdrant: image: qdrant/qdrant:latest ports: [6333:6333] volumes: - qdrant_storage:/qdrant/storage environment: QDRANT__SERVICE__GRPC_PORT: 6334 elasticsearch: image: elasticsearch:8.12.0 environment: - discovery.typesingle-node - xpack.security.enabledfalse ports: [9200:9200] volumes: - es_data:/usr/share/elasticsearch/data## 总结混合检索是2026年RAG工程的标准配置不是可选项。核心要点1.RRF是默认选择简单、稳健、不需要调参K60开箱即用2.动态权重更优根据查询类型调整向量/关键词权重比固定权重效果更好3.查询改写增强多路查询去重能进一步提升召回率10-20%4.评估驱动优化用RecallK和MRR量化不同配置的效果差异5.QdrantES是黄金组合向量检索用Qdrant关键词检索用Elasticsearch两者都有成熟的生产方案纯向量RAG在简单Demo上表现不错但生产环境复杂查询的精确召回混合检索是必经之路。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2566648.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…