免费获取学习方案
ARTICLE DETAIL

资讯详情

深耕编程基础知识与建站技术分享的一线实战洞察。

向量检索与标量过滤混合查询:PostgreSQL pgvector 与 Milvus 的过滤下推实操

向量检索与标量过滤混合查询:PostgreSQL pgvector 与 Milvus 的过滤下推实操 向量检索与标量过滤混合查询PostgreSQL pgvector 与 Milvus 的过滤下推实操在构建企业级 RAG 知识库与智能体记忆检索时极少有纯粹的“全局高维向量相似度检索”。绝大多数真实的业务查询都带有极其严格的标量业务过滤条件Scalar Filtering“在[租户ID tenant_001]且[文档分类 财务制度]且[发布时间 2026-01-01]的范围内检索与‘发票报销时限’最相似的 Top-5 段落”。如果在架构设计与中间件选型时不理解“标量过滤”与“向量索引”的结合机制系统在海量数据下面临灾难性的性能坍塌前置过滤Pre-filtering全表扫描如果先按标量过滤出符合条件的 500 条记录再在这些记录上逐一做暴力向量距离计算当候选集大时耗时极大且无法利用 HNSW 索引加速后置过滤Post-filtering结果真空如果先用 HNSW 捞出全局 Top-100 相似向量再在内存中过滤tenant_id若该租户的数据在全局占比仅为 0.1%过滤后最终可能只剩下 0 条有效结果导致严重的“检索结果凭空消失”。如何在PostgreSQL (pgvector)与Milvus / Qdrant中实现真正的单阶段原生过滤下推Single-Stage Filter Pushdown一、三大混合过滤机制的底层执行对比┌────────────────────────────────────────────────────────┐ │ 模式 1: 后置过滤 (Post-Filtering - 极易漏检) │ │ 流程: 全局 HNSW Top-K ──► 内存标量过滤 ──► 剩余结果极少│ ├────────────────────────────────────────────────────────┤ │ 模式 2: 前置过滤 (Pre-Filtering - 无法利用图索引) │ │ 流程: 标量索引查出 ID ──► 暴力计算余弦相似度 ──► 延迟高│ ├────────────────────────────────────────────────────────┤ │ 模式 3: 单阶段原生过滤下推 (Filter Pushdown - 推荐) │ │ 流程: 在 HNSW 图游走遍历的每一步动态跳过不符合标量条件的│ │ 节点兼顾图索引的极速与标量过滤的 100% 准确性 │ └────────────────────────────────────────────────────────┘二、PostgreSQL pgvector 的混合查询生产实操在 PostgreSQL 16 与 pgvector 0.7 中通过引入迭代式索引扫描Iterative Index Scan机制完美解决了后置过滤的漏检问题-- 1. 创建包含业务标量与向量的复合数据表 CREATE TABLE enterprise_documents ( id BIGSERIAL PRIMARY KEY, tenant_id VARCHAR(64) NOT NULL, category VARCHAR(32) NOT NULL, created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(), content TEXT NOT NULL, embedding vector(1536) NOT NULL ); -- 2. 创建标量复合索引 CREATE INDEX idx_docs_tenant_cat ON enterprise_documents(tenant_id, category); -- 3. 创建针对 1536 维向量的 HNSW 索引 CREATE INDEX idx_docs_hnsw ON enterprise_documents USING hnsw (embedding vector_cosine_ops) WITH (m 32, ef_construction 128); -- 4. 生产级单阶段混合查询 (过滤下推生效) EXPLAIN ANALYZE SELECT id, content, 1 - (embedding [0.012, -0.045, ...]) AS similarity_score FROM enterprise_documents WHERE tenant_id tenant_001 AND category finance_policy AND created_at 2026-01-01 ORDER BY embedding [0.012, -0.045, ...] LIMIT 5;在 PG 执行计划中Query Optimizer 会自动识别并将tenant_id过滤条件下推至 HNSW 索引扫描循环内部保证既利用了图索引加速又精准满足多租户隔离。三、Milvus 分布式向量数据库过滤下推实战Milvus 在底层架构中原生支持标量与向量的联合执行计划Bitset 机制from pymilvus import Collection collection Collection(enterprise_kb_collection) # 1. 构造结构化标量布尔过滤表达式 filter_expr tenant_id tenant_001 and category in [finance, tax] and created_timestamp 1767225600 # 2. 执行联合检索 search_params { metric_type: COSINE, params: {ef: 96} # HNSW 在线遍历深度 } results collection.search( data[query_embedding], anns_fieldvector, paramsearch_params, limit5, exprfilter_expr, # 标量过滤表达式直接下推至底层的 QueryNode output_fields[doc_id, title, content] ) for hits in results: for hit in hits: print(fID: {hit.entity.get(doc_id)} | Score: {hit.distance:.4f})Milvus 在检索前会预先通过标量索引生成一个Bitset位图在 HNSW 图游走过程中通过极速的位运算Bitwise AND直接判定某个向量节点是否合法处理 1000 万条数据仅需 8~15ms。四、生产选型与调优总结评估维度PostgreSQL pgvector专用 Milvus / Qdrant 集群数据量规模100 万条以内500 万 ~ 1 亿条以上标量过滤灵活性极高支持复杂 SQL JOIN 与子查询较高支持常见比较与 IN 表达式运维成本极低复用现有 PG 关系库中等需维护独立分布式集群推荐场景中小企业一体化 RAG、单库多租户超大规模企业级数据中台与智能体大脑搞懂过滤下推的底层执行链路告别低效的前后置暴力过滤才能为多租户、强权限的企业级智能体构建起既安全合规又极速响应的检索中枢。
返回列表