向量化召回算法ANN、IVF、HNSW、IVF-PQ

ANN(Approximate Nearest Neighbor) 意为近似近邻搜索,即通过模型预计算,建立索引或图以便在线检索时通过剪枝或图检索从而加速检索过程,通常为非穷尽检索。目前常用的ANN 检索方式有 IVF, HNSW, BTree, IVF-PQ。

IVF

IVF(lnverted File)通过在模型预计算时将所有候选进行KMeans聚类,即检索距离最近的m个聚类蔟,遍历这m个聚类蔟下所有候选,计算得出距离最近的topk候选。扩大IVF检索覆盖范围或增加聚类数都可以提高检索精度,这也正是IVF-PQ以及BTree分别采用的思路。

HNSW

HNSW(Hierarchical Navigable Small World) 类似于跳表的思想,在构建阶段,在搜索时,从最高层的入点出发,逐层贪心搜索距离query vec最近的点,当一层达到收敛后,进入下一层,直至到达底层,在底层L0中,包含候选集中所有的点,即全量数据图网络。在L0层中开始深度优先搜索距离query vec最近的候选点集,每次将距离query vec最近的点加入候选点集中,若候选点集已满,则弹出最远候选点,直至候选点集收敛。

微软、FB、阿里的相关论文(DistANN,NSG,SSG)对HNSW的性能做了大量的研究与改进,经验上来说,召回的topk比较小的情况下(例如召回只有几百的情况下),HNSW的召回率较高,因此HNSW更适合推荐场景;但如果召回的topk比较大的情况下(例如召回达到2000左右的规模),IVF更适合一些,因此IVF和后面的PQ-IVF也更适合广告场景

图片来自 https://www.pinecone.io/learn/series/faiss/hnsw/

BTree

BTree和IVF相比主要是通过使用GPU来大幅度增加聚类中心点来优化。IVF跑在CPU上如果聚类中心点过多,速度会比较慢。

IVF-PQ

IVF-PG中的PQ意思是Product Quantization,常用的量化方式有Fix16、Int8和PQ

Fix16

通过全局的scale,把float32量化到fix16,从而节省一半的存储

int8

int8就不能像Fix16直接使用全局的scale,精度丢失的太厉害,因此常有的有两种思路:

QAT (Quantization-aware Training / 量化感知训练)

简单来说就是把量化带来的误差引入到模型训练中,使得模型能够学习到全局唯一的scale(对称量化

PTQ(Post Training Quantization /训练后量化)

逐条向量中的每一条根据最大元素放缩,每一条向量都有自己的scale和offset(非对称量化

Product Quantization

In short, PQ is the process of:

  • Taking a big, high-dimensional vector,
    Splitting it into equally sized chunks --- our subvectors,
  • Assigning each of these subvectors to its nearest centroid (also called reproduction/reconstruction values),
  • Replacing these centroid values with unique IDs --- each ID represents a centroid

在PQ中,每个subvector的量化器还是由kmeans得到

图片来自https://www.pinecone.io/learn/series/faiss/product-quantization/

IVF-PQ

IVF_PQ先将所有的候选向量通过kmeans划分到多个聚类簇中,然后将每个候选向量与其中心点做差得到残差向量,得到残差向量矩阵,再对此残差向量矩阵进行PQ量化。如果不用残差的话,原始向量可能会有特别大的分布差异/不平衡

相关推荐
古城小栈1 小时前
Rust变量设计核心:默认不可变与mut显式可变的深层逻辑
算法·rust
电商API&Tina1 小时前
跨境电商 API 对接指南:亚马逊 + 速卖通接口调用全流程
大数据·服务器·数据库·python·算法·json·图搜索算法
LYFlied2 小时前
【每日算法】LeetCode 1143. 最长公共子序列
前端·算法·leetcode·职场和发展·动态规划
长安er3 小时前
LeetCode 20/155/394/739/84/42/单调栈核心原理与经典题型全解析
数据结构·算法·leetcode·动态规划·
MarkHD3 小时前
智能体在车联网中的应用:第28天 深度强化学习实战:从原理到实现——掌握近端策略优化(PPO)算法
算法
能源系统预测和优化研究3 小时前
【原创代码改进】考虑共享储能接入的工业园区多类型负荷需求响应经济运行研究
大数据·算法
yoke菜籽4 小时前
LeetCode——三指针
算法·leetcode·职场和发展
小高不明4 小时前
前缀和一维/二维-复习篇
开发语言·算法
bin91534 小时前
当AI优化搜索引擎算法:Go初级开发者的创意突围实战指南
人工智能·算法·搜索引擎·工具·ai工具
曹牧6 小时前
Java:Math.abs()‌
java·开发语言·算法