Page 122 - 《软件学报》2026年第3期
P. 122

王可 等: 面向批量更新的向量索引召回率优化                                                          1085


                 algorithms  and  lead  to  sparsification  issues  in  the  connections  among  similar  vectors,  collectively  causing  a  significant  decline  in  retrieval
                 recall.  To  address  these  issues,  this  study  proposes  an  adaptive  fine-grained  pruning  strategy  based  on  local  adjustments  to  the  graph
                 structure  and  constructs  a  comprehensive  optimization  scheme  that  integrates  an  identification  and  repair  mechanism.  First,  in  the
                 identification  phase,  the  regional  neighbor  distance  is  calculated  to  quantify  local  topological  density,  thereby  precisely  locating  the  dense
                 regions  requiring  intervention.  Second,  in  the  repair  phase,  for  hub  nodes  in  dense  regions,  a  dual  pruning  neighbor  selection  strategy  is
                 adopted:  native  and  modified  heuristic  pruning  rules  are  applied  synergistically,  and  the  results  of  both  rules  are  merged  to  enhance
                 neighbor  connection  diversity  while  maintaining  retrieval  accuracy,  effectively  alleviating  over-pruning  and  connection  sparsification  issues.
                 Experimental  results  on  multiple  public  datasets  show  that  the  proposed  method  demonstrates  good  adaptability  in  scenarios  with  frequent
                 data updates, achieving a 1%–4% improvement in recall while maintaining stable query latency and throughput.
                 Key words:  approximate nearest neighbor search (ANNS); vector retrieval; graph-based vector index

                    向量的近似最近邻搜索 (approximate nearest neighbor search, ANNS) [1−6] 致力于在大规模高维数据 (即高维向
                 量) 中迅速获取目标向量的最近向量, 是当代人工智能应用, 特别是大模型相关应用中的重要技术. 在当今主流
                                                                       [7]
                 AI 系统中, 多模态异构数据       (如文本、图像、音视频等非结构化数据) 首先经由向量嵌入 (vector embedding)                [8,9]
                 技术将其语义抽取表征为高维向量, 向量间的几何距离 (如欧氏距离、余弦相似度等) 表征了向量间的语义关系.
                 在这种模式下, 在多模态数据中检索语义相近的数据的任务即转化为了向量间的最近匹配. 由于高维空间中最近
                 邻匹配的计算复杂度较高, 近似最近邻 (ANN) 以匹配精度换取计算的高效性, 提升了算法在                          AI 应用中的可行性.
                 向量数据库    [10,11] 存储与管理向量数据, 集成    ANNS  等向量检索算法, 现已成为前沿大模型应用的重要数据基座.
                 例如, 生成式大语言模型 (generative large language model) [12,13] 、检索增强技术 (retrieval-augmented generation,
                 RAG) [14−17] 通过向量检索从外源数据获取相关语义来增加           LLM  的生成和推理能力, 有效地解决了大模型的幻觉              [18]
                 问题, 显著提升大模型生成内容的质量.
                    为了实现高效的       ANNS, 学术界与工业界已探索出多种索引方法, 主要分为基于树 (tree-based)               [19,20] 、基于图
                 (graph-based) [21–23] 、基于哈希 (hash-based) [24,25] 、基于聚类 (clustering-based) [26] 和基于量化 (quantization-based) [27,28]
                 等几类. 其中, 基于图的      ANNS  方法因其在查询效率、召回精度与可扩展性之间取得了卓越的平衡而备受青睐.
                 这类方法构建并利用向量间的近邻图进行快速的图导航. 作为该领域的代表性算法, 分层可导航小世界 (hierarchical
                 navigable small world, HNSW) [29] 通过引入层级化的图结构与可导航小世界网络, 实现了以较低的计算开销达到较
                 高的召回率, 已成为当前应用最为广泛的图索引方法之一.
                    然而, 尽管现有的      ANNS  算法在通用场景下性能优异, 但在向量数据频繁写入场景中, 其性能面临显著挑战.
                 这一挑战在许多现实应用中尤为突出, 例如在知识库构建或实时推荐系统中, 新数据常以批量形式集中写入, 其批
                 内向量因其相似的来源或主题, 在空间中呈现高度的局部聚集性. 在这种批量插入相似数据模式下, HNSW                                 索引
                 的召回率会发生显著下降, 经过本文实验测试, 在              GIST1M  数据集上, 最大降幅达到       9%. 经研究发现, 其召回率退
                 化的核心原因可归结为两点: 其一是           HNSW  选择邻居关系时所采用的启发式规则引发的过度剪枝, 即在处理过度
                 密集的局部区域时, 该规则会误判冗余连接, 从而错误地移除能提供关键搜索方向的邻居; 其二则是相似向量连接
                 稀疏化的问题, 即批量插入的相似数据节点难以建立足够数量的有效邻居连接, 从而在局部形成了导航能力严重
                 受损的稀疏子图, 导致索引无法支撑起有效的全局导航. 正是由于缺乏对索引结构性问题的有效感知, 现有索引方
                 法  [30−32] 在处理相似向量的批量插入时普遍暴露其性能局限, 最终导致召回率的显著下降. 鉴于此, 在不影响性能的
                 前提下, 如何设计一种能够有效应对上述结构性问题的索引优化方法, 保证向量检索在动态场景下的召回率, 已成
                 为一个关键问题.
                    为了解决过度剪枝和相似向量的连接稀疏化引发的召回率退化问题, 本文提出一种基于图结构局部调整的自
                 适应细粒度剪枝策略. 该策略通过多阶段的识别与修复框架, 在索引构建过程中对图的局部拓扑进行精准干预. 其
                 核心机制包含以下       3  个环节.
                    (1) 数据驱动的局部致密区域识别. 为精准应对过度剪枝问题, 该策略通过数据驱动的预分析过程, 自适应地
                 设定识别局部密度的阈值          β. 在此基础上, 根据节点的区域邻居平均距离与全局基线的差异来识别过度致密
                 区域.
   117   118   119   120   121   122   123   124   125   126   127