登录
主页
向量检索底层算法原理
2026-07-15
  
1006
深数据
在人工智能、大模型RAG检索、图像音频匹配、推荐系统等场景中,非结构化数据(文本、图片、语音)无法通过传统关键词、结构化索引实现精准匹配。行业通用解决方案是将非结构化数据通过Embedding模型转化为高维稠密向量,依托向量检索技术挖掘向量间的语义相似性,从而完成内容匹配、召回与筛选。向量检索的核心价值,是在亿万级高维向量库中,以极低时延快速匹配与目标向量最相似的候选向量,是AI落地检索类业务的核心底层技术。
一、向量检索核心基础:相似度度量准则
高维向量本质是高维空间中的坐标点,向量检索的本质是在高维空间中寻找查询向量的最近邻向量,而判断向量远近、相似度高低的核心依据是标准化的距离/相似度计算公式,不同场景适配不同度量准则,也是算法优化的基础前提。
1.欧氏距离(L2距离)
欧式距离衡量高维空间中两个向量的直线绝对距离,距离越小,向量相似度越高,适用于图像特征、数值型向量匹配场景。
公式:d(A,B) = √[∑ⁿᵢ₌₁ (Aᵢ − Bᵢ)²]
优势:直观反映向量空间绝对差异;劣势:对高维向量计算开销大,易受维度冗余信息干扰。
2.余弦相似度
忽略向量模长,仅关注向量夹角,衡量向量的方向相似度,是文本Embedding检索、语义匹配的主流准则,完美适配语义相似但向量幅值不同的场景。
公式:cos(A,B) = (A · B) / (|A| × |B|)
取值范围[-1,1],数值越接近1,向量语义相似度越高,有效规避了文本长度、向量幅值对检索结果的干扰。
3.曼哈顿距离(L1距离)& 内积距离
曼哈顿距离计算向量各维度差值绝对值之和,计算轻量化,适用于低维快速检索场景;内积距离则在向量归一化后等价于余弦相似度,计算开销极低,是大规模向量检索的优选辅助度量方式。
二、向量检索两大范式:KNN精确检索与ANN近似检索
从检索逻辑划分,向量检索分为精确检索与近似检索两类,二者的取舍决定了检索系统的性能上限,也是所有底层算法的设计核心出发点。
1.KNN精确最近邻检索
KNN(K-Nearest Neighbors)是最基础的检索范式,核心逻辑为全库遍历计算:针对查询向量,逐一计算其与向量库中所有向量的距离,排序后选取距离最小的Top-K向量作为结果。
该算法优势为100%召回、无精度损耗,无需提前构建索引;但致命缺陷是时间复杂度极高,时间复杂度为O(N·D)(N为向量总数,D为向量维度)。在千万级、亿级高维向量场景下,单次检索耗时可达数十秒,完全无法满足线上实时业务的毫秒级响应需求,仅适用于小批量离线检索场景。
2.ANN近似最近邻检索
为解决KNN精确检索性能瓶颈,行业普遍采用ANN(Approximate Nearest Neighbor)近似最近邻检索算法,核心思想是以极小的精度损耗,换取指数级的检索性能提升。
ANN算法通过提前构建向量索引,对高维向量空间进行分块、映射、压缩或构图,检索时无需遍历全量向量,仅遍历局部高概率候选集,可将亿级向量检索耗时压缩至1~10ms,召回率稳定在92%~99%,完全适配线上实时检索、RAG问答、智能推荐等核心业务场景。当前主流的底层ANN算法可分为四大类:局部敏感哈希LSH、倒排索引IVF、乘积量化PQ、分层导航小世界HNSW。
三、主流ANN底层算法核心原理拆解
1.局部敏感哈希(LSH):哈希映射保相似性
LSH(Locality Sensitive Hashing)是经典的哈希类近似检索算法,颠覆了传统哈希“相似输入、不同输出”的散列特性,核心设计是让高维相似向量映射为相同/相近哈希码,不相似向量映射为完全不同哈希码,实现相似向量的快速聚类匹配。
索引构建阶段:通过随机投影生成多组哈希函数,将高维向量降维映射为固定长度的二进制哈希码;哈希码相同的向量归入同一个哈希桶,完成向量空间的粗分块存储。为降低哈希碰撞概率,通常会构建多组独立哈希表,提升候选集覆盖率。
检索查询阶段:对查询向量执行相同的哈希函数生成哈希码,仅定位对应哈希桶内的向量,仅对桶内少量向量做精准距离计算、排序,筛选Top-K结果。
LSH算法优势是无需训练、增量更新友好、天然支持分布式;劣势是高维向量场景下哈希冲突概率上升,索引存储空间大,召回率稳定性一般,目前多用于小规模、低时延容忍的检索场景。
2.倒排文件索引(IVF):聚类分治缩小检索范围
IVF(Inverted File)倒排索引借鉴传统文本检索的倒排思想,结合K-Means聚类算法,核心逻辑是空间分块、分而治之,将“全库遍历”转化为“局部池检索”,是工业界经典的轻量化检索算法。
索引训练与构建阶段:首先通过K-Means聚类算法,将全量高维向量聚类为nlist个簇,每个簇生成一个专属质心;随后遍历所有向量,计算每个向量与所有质心的距离,将向量归入距离最近的簇中,最终形成“质心-向量列表”的倒排索引结构,完成向量空间的结构化划分。
检索查询阶段:第一步,计算查询向量与所有簇质心的距离,筛选出距离最近的nprobe个候选簇(nprobe为检索超参);第二步,仅遍历这nprobe个簇内的所有向量,完成距离计算与排序,输出Top-K相似向量。
IVF算法核心优化点在于大幅减少检索遍历的向量数量,nprobe取值越小,检索速度越快,但召回率会略有下降,可通过超参灵活平衡性能与精度。该算法结构简单、构建速度快、内存占用低,适配大规模海量向量检索,是FAISS框架的核心基础算法之一。缺陷是聚类结果依赖K-Means效果,向量分布不均时易出现簇倾斜问题,影响检索稳定性。
3.乘积量化(PQ):向量压缩降低计算开销
PQ(Product Quantization)乘积量化是向量压缩与近似检索结合的经典算法,核心目标是解决高维向量存储占用高、距离计算开销大的问题,通过量化压缩实现极致的内存优化与提速。
核心原理:将原始D维高维向量均匀切分为m个子向量段,对每一段子向量单独做K-Means聚类,生成对应子向量量化中心;随后用子向量所属的量化中心ID替代原始子向量,最终将高维浮点向量压缩为短整型量化编码,向量存储体积可压缩至原来的1/10~1/20。
检索逻辑:查询向量同样执行分段量化,通过预计算各子量化中心的距离表,无需还原原始向量,直接通过量化编码快速计算向量相似度,完成检索排序。
PQ算法最大优势是内存占用极低、检索速度极快,适配超大规模亿级、十亿级向量库;缺点是量化压缩会带来不可逆的精度损耗,单独使用召回率偏低,工业界常将IVF与PQ结合(IVFPQ算法),兼顾空间分块与向量压缩,实现性能、内存、精度的均衡。
4.分层导航小世界(HNSW):当前最优综合性能算法
HNSW(Hierarchical Navigable Small World)分层导航小世界是目前工业界综合性能最优、应用最广泛的向量检索算法,广泛用于Milvus、Qdrant、Pinecone等主流向量数据库,核心优势是完美平衡检索速度、召回率、增量更新性能。其本质是基于小世界网络的多层图索引结构,模拟“高速路网+普通路网”的分层检索逻辑。
索引构建原理:HNSW构建多层稀疏网络图,层数自上而下递减,顶层为稀疏高速层,底层为稠密完整层。新向量插入时,首先随机分配所属层级,从顶层入口节点开始,逐层向下检索,在每一层中与邻近节点建立双向连接,同时冗余删除过远连接,保证每层节点连接数稳定在固定阈值,最终形成多层可导航的小世界拓扑网络。层级越高,节点越稀疏,检索跳转速度越快;层级越低,节点越稠密,向量匹配越精准。
检索查询原理:检索从顶层入口节点发起,在高层稀疏网络中快速跳转,快速逼近查询向量的目标区域,完成全局粗定位;随后逐层下沉至底层稠密网络,在局部精细范围内遍历候选节点,筛选出最优Top-K相似向量。
HNSW算法的核心亮点是无需聚类、无需量化,原生支持向量增量插入、更新与删除,检索时延极低且召回率极高,几乎无精度损耗。唯一短板是索引构建内存占用较高,适合对检索精度和实时性要求高的核心业务,如大模型RAG检索、精准推荐、人脸比对等场景。
四、主流检索算法核心参数对比与场景取舍
不同底层算法的设计逻辑、性能特性差异显著,落地场景需根据向量规模、实时性要求、精度要求、内存资源灵活选型。各主流算法的详细对比与适配场景如下:
LSH(局部敏感哈希):核心原理为局部敏感哈希映射,依靠特殊哈希函数实现相似向量聚类匹配。检索速度较快,但召回精度处于中等水平,索引内存占用中高。该算法无需训练、增量更新友好且天然支持分布式,更适配小规模向量库、增量更新频繁的业务场景。
IVF(倒排文件索引):核心依托聚类倒排分块检索逻辑,通过K-Means聚类划分向量空间,缩小检索遍历范围。检索速度极快、召回精度中高,同时内存占用极低,算法结构简单、构建效率高,非常适合大规模海量向量场景,多用于高吞吐的离线、准在线检索业务,也是FAISS框架的核心基础算法。唯一短板是检索效果依赖聚类质量,向量分布不均时易出现簇倾斜问题。
PQ/IVFPQ(乘积量化):核心基于向量分段量化压缩技术,通过拆分高维向量、子向量聚类量化的方式极致压缩向量体积。检索速度达到极快级别,内存占用极低,可支撑十亿级超大规模向量库与内存资源受限场景;缺点是量化压缩会带来轻微不可逆精度损耗,单独使用召回精度中等,工业界多结合IVF算法形成IVFPQ方案,兼顾检索性能、内存占用与检索精度。
HNSW(分层导航小世界):依托分层小世界图网络结构,构建多层稀疏拓扑索引,实现粗定位+精检索的分层检索逻辑。综合性能最优,检索速度极速、召回精度极高,几乎无精度损耗,且原生支持向量增量插入、更新、删除,实时性极佳。仅存在索引构建阶段内存占用中高的短板,是目前工业界主流算法,广泛应用于大模型RAG检索、精准匹配、线上实时核心业务、人脸比对等对精度和时延要求严苛的场景。
五、向量检索底层核心优化逻辑总结
所有向量检索底层算法的优化逻辑均可归纳为三大核心方向,也是技术迭代的底层逻辑:
1.空间裁剪:通过聚类、哈希、分层构图等方式,摒弃全库遍历,仅检索高概率候选区域,降低检索计算量,代表算法IVF、HNSW;
2.数据压缩:通过量化、降维等方式压缩向量存储体积,减少内存占用与距离计算开销,代表算法PQ;
3.结构优化:通过构建多层、拓扑网络结构,平衡粗定位与精检索效率,实现速度与精度的最优均衡,代表算法HNSW。
六、结语
向量检索的底层本质,是高维空间最近邻搜索的工程化最优解。KNN精确检索奠定了检索的理论基础,而各类ANN近似检索算法通过差异化的空间优化、数据压缩、结构设计,解决了高维海量向量检索的性能瓶颈。其中HNSW凭借综合性能优势成为当前主流方案,IVF+PQ组合则扛起超大规模低成本检索的场景需求。深入理解各类算法的底层原理与取舍逻辑,是向量数据库选型、RAG系统优化、检索业务落地的核心基础,也是AI非结构化数据处理的关键技术支撑。
点赞数:14
© 2021 - 现在 杭州极深数据有限公司 版权所有 (深数据® DEEPDATA® 极深®) 联系我们 
浙公网安备 33018302001059号  浙ICP备18026513号-1号