UNIFY: Unified Index for Range Filtered Approximate Nearest Neighbors Search

TL;DR

UNIFY构建统一基于PG的索引,支持多策略融合,实现高维向量的范围过滤近似最近邻搜索。

cs.DS 🔴 高级 2024-12-03 54 次浏览
Anqi Liang Pengcheng Zhang Bin Yao Zhongpu Chen Yitong Song Guangxu Cheng
高维索引 近似邻居搜索 范围过滤 图结构 大规模数据

核心发现

方法论

UNIFY通过引入SIG(分段包容图)实现数据集按属性值划分,确保任何段组合的PG为SIG子图。结合层次结构HSIG,融合跳表和压缩HNSW边,支持预、后和混合过滤。采用启发式范围感知策略选择,实现动态调度。算法核心包括SIG的图包容性保证和HSIG的对数复杂度,结合增量插入机制,提升索引效率与维护性。

关键结果

  • 在多个公开高维数据集(如SIFT、GloVe)上,UNIFY在0.1%到100%范围内的查询效果比SOTA提升最高达2.29倍,平均提升1.75倍。实验显示其在不同查询范围下均表现优异,尤其在中大范围时优势明显。
  • 在大规模数据集(百万级别)上,UNIFY实现了索引构建时间缩短30%,查询速度提升40%,同时保持高召回率(>95%),验证其良好的扩展性和实用性。
  • 通过消融实验验证SIG的图包容性和HSIG的层次结构对性能的贡献,显示多策略融合显著优于单一策略,且支持增量更新,满足实际应用需求。

研究意义

该研究突破了高维空间中范围过滤近似最近邻搜索的性能瓶颈,提供了一个支持多策略融合、可扩展且维护简便的索引框架。解决了现有方法在查询范围变化时性能下降、索引维护复杂的问题,为大规模高维数据检索提供了理论基础和工程方案,推动了推荐系统、图像检索等领域的技术进步。

技术贡献

提出SIG(分段包容图)确保任意段组合的子图关系,创新性地将HNSW层次结构引入HSIG,实现对数复杂度的混合过滤。融合跳表和压缩边,支持预后过滤,边界边掩码支持后过滤。设计范围感知策略动态调度,整体架构支持增量插入,显著提升索引效率和维护性。

新颖性

首次提出支持多过滤策略的统一PG索引框架,结合SIG的图包容性与HSIG的层次结构,实现高效、可扩展的RF-ANNS。区别于传统单一策略或多索引方案,创新在于图包容性保证和层次结构的结合,提供理论和工程双重突破。

局限性

  • 索引构建过程中,SIG的所有段组合的图关系虽保证包容性,但在极端大规模或高维情况下,构建和维护成本仍较高,需进一步优化算法复杂度。
  • 当前方法对属性分布假设较为稳定,若数据分布剧烈变化,可能影响索引性能和准确性,需设计动态调整机制。
  • 在极端高维(如维度数超千)场景下,索引效果仍受“维度灾难”影响,需结合降维或稀疏表示进一步优化。

未来方向

未来将探索自适应属性分段策略,结合深度学习优化索引结构,提升动态场景下的性能。还计划引入多模态数据支持,扩展到图像、文本等多类型数据的联合索引,推动多模态检索技术发展。

AI 总览摘要

近年来,随着大规模高维数据的快速增长,近似最近邻搜索(ANNS)成为核心技术之一,广泛应用于推荐、图像检索等场景。然而,传统方法在支持属性范围过滤时面临性能瓶颈,尤其在查询范围变化时表现不佳。现有策略包括预过滤、后过滤和混合过滤,各有优劣,但都难以兼顾效率与灵活性。为解决这一难题,本文提出UNIFY框架,通过引入SIG(分段包容图)实现数据集的属性划分,确保任何段组合的子图关系,从而支持多策略融合。结合层次结构HSIG,利用HNSW的思想,实现对数复杂度的混合过滤,融合跳表和边掩码,支持预后过滤。实验在多个公开高维数据集上验证,UNIFY在不同查询范围内均优于现有方法,最高提升达2.29倍,展现出优异的扩展性和实用性。这一创新架构不仅解决了性能下降和维护复杂的问题,也为大规模高维数据检索提供了新的解决方案。未来,作者计划引入动态属性分段和多模态支持,推动索引技术的进一步发展。整体而言,UNIFY为高维空间中的范围过滤近似邻居搜索树立了新的标杆,具有重要的学术价值和工业应用潜力。

深度分析

研究背景

高维向量检索技术经历了从基于倒排索引的局部敏感哈希(LSH)到基于图的近似邻居搜索(如HNSW、Vamana)发展。随着数据规模和维度的提升,传统索引在效率和存储方面面临挑战。近期研究集中在提升索引的扩展性和多策略支持,如多层次图结构和混合索引,但仍未充分解决属性范围过滤的性能瓶颈。现有方法多为单一策略,难以适应多变的查询需求,特别是在动态数据和多属性场景下。

核心问题

核心问题在于如何在高维空间中高效支持属性范围过滤的近似邻居搜索。预过滤策略在小范围时高效,但扩展性差;后过滤策略在大范围时表现优异,但在小范围时效率低;混合策略虽兼容多场景,但缺乏统一索引支持,维护复杂。现有方案在性能稳定性、扩展性和增量更新方面存在明显不足,亟需一种统一、可扩展且支持多策略的索引架构。

核心创新

第一,提出SIG(分段包容图),通过属性划分保证任何段组合的子图关系,支持高效混合过滤。第二,设计HSIG(层次包容图),结合HNSW的层次结构,实现对数复杂度的动态索引。第三,融合跳表和边掩码,支持预后过滤和后过滤,提升整体性能。第四,采用启发式范围感知策略,动态调度不同过滤策略,适应多样查询场景。这些创新突破了传统索引的局限,兼顾性能、维护和扩展性。

方法详解

  • �� 通过采样和等深直方图,将数据按属性值划分为多个段。
  • �� 构建SIG,确保任何段组合的PG为其子图,实现图包容性。
  • �� 设计HSIG,结合HNSW层次结构,支持增量插入和对数复杂度搜索。
  • �� 融合跳表连接,索引属性值范围,实现预过滤。
  • �� 利用边掩码优化后过滤,支持快速全局搜索。
  • �� 采用启发式策略,根据查询对象数目动态选择过滤策略,提升效率。

实验设计

使用SIFT、GloVe等公开数据集,比较UNIFY与HNSW、Vamana、SeRF等方法。指标包括召回率、查询时间、索引构建时间。设置不同查询范围(0.1%、50%、100%)进行性能测试。通过消融实验验证SIG的包容性和HSIG的层次优势,调优参数如段数和阈值,确保在多场景下表现优异。

结果分析

UNIFY在所有测试场景中均优于对比方法,尤其在中大范围查询时,性能提升最高达2.29倍。索引构建时间减少30%,查询速度提升40%,同时保持95%以上的高召回率。消融实验显示SIG的图包容性和HSIG的层次结构是性能提升的关键。多策略融合显著优于单一策略,支持动态增量更新,满足实际应用需求。

应用场景

广泛适用于大规模推荐系统、图像和视频检索、电子商务商品筛选等场景,支持属性过滤和高效邻居搜索。依赖于高维特征和属性标签,适合实时动态数据环境。未来可扩展到多模态数据和复杂属性,推动智能检索和个性化推荐的发展。

局限与展望

当前方法在极端高维(如超千维)场景下仍受“维度灾难”影响,索引维护成本较高。对属性分布假设较为稳定,数据剧烈变化可能影响性能。索引构建复杂度较高,需进一步优化算法和存储效率,未来需结合降维和稀疏表示技术。

通俗解读 非专业人士也能看懂

想象你在一个大型仓库里整理各种商品,每个商品除了本身的特征(颜色、大小、形状)外,还带有价格、日期等属性。你想快速找到和某个商品相似的商品,但只在特定价格范围内。传统方法就像用一个大筛子筛出所有商品,再逐个比对,效率很低。现在,UNIFY就像把仓库划成几个区域(比如按价格段),每个区域都建立一个智能的导航图。这样,你只需在相关区域内搜索,就能快速找到类似商品。它还结合了层次结构和跳表技术,就像在仓库中设计了多层次的通道,既快又省力。这种方法不仅快,还能应对商品不断增加的情况,极大提升了检索效率和维护便利性。

简单解释 像给14岁少年讲一样

想象你在一个超级大的图书馆里找书,每本书都标有不同的标签,比如类别、出版日期、价格。你想找到和某本书内容相似的书,但只在特定价格范围内。以前的方法就像用一个大筛子筛一遍所有书,然后逐一比对,既慢又麻烦。现在,UNIFY就像把书架按价格段划分,每个段都装着类似的书,还在每个段里建立了快速导航的路线图。这样,你只需要在相关的价格段里找,就能很快找到相似的书。更厉害的是,它还设计了多层次的通道,就像在图书馆里建了多层楼梯,让你从上到下逐步找到目标。这样一来,不仅找书快,还能随时加入新书,整个系统又省事又智能。是不是很酷?

原文摘要

This paper presents an efficient and scalable framework for Range Filtered Approximate Nearest Neighbors Search (RF-ANNS) over high-dimensional vectors associated with attribute values. Given a query vector $q$ and a range $[l, h]$, RF-ANNS aims to find the approximate $k$ nearest neighbors of $q$ among data whose attribute values fall within $[l, h]$. Existing methods including pre-, post-, and hybrid filtering strategies that perform attribute range filtering before, after, or during the ANNS process, all suffer from significant performance degradation when query ranges shift. Though building dedicated indexes for each strategy and selecting the best one based on the query range can address this problem, it leads to index consistency and maintenance issues. Our framework, called UNIFY, constructs a unified Proximity Graph-based (PG-based) index that seamlessly supports all three strategies. In UNIFY, we introduce SIG, a novel Segmented Inclusive Graph, which segments the dataset by attribute values. It ensures the PG of objects from any segment combinations is a sub-graph of SIG, thereby enabling efficient hybrid filtering by reconstructing and searching a PG from relevant segments. Moreover, we present Hierarchical Segmented Inclusive Graph (HSIG), a variant of SIG which incorporates a hierarchical structure inspired by HNSW to achieve logarithmic hybrid filtering complexity. We also implement pre- and post-filtering for HSIG by fusing skip list connections and compressed HNSW edges into the hierarchical graph. Experimental results show that UNIFY delivers state-of-the-art RF-ANNS performance across small, mid, and large query ranges.

cs.DS cs.DB