编程知识 cdmana.com

facebook大数据搜索库faiss使用——选择Index

选择Index并不明显,有几个问题可以帮助选择Index。

是否需要精确结果

使用Flat
IndexFlat2是唯一能保证精确结果的Index。它为其他Index提供了对比标准。它不会压缩向量,不支持带标签添加,只能顺序添加。所以,如果你需要add_with_ids,使用IDMap,Flat

内存是否有限制

请注意Faiss所有的Index都是存储在RAM里的,如果不需要精确结果,而同时RAM是有限的,在该限制下,我们在精度-速度中间进行衡量选择。可以考虑下面的问题:

内存充足无限制:使用HNSWx

如果你有很多RAM或者搜索库很小,HNSW是最好的选择。它很快、很精确。4<=x<=64是每个向量的链接数,这个值越高越精确,但是使用的RAM越多。而速度-精度衡量则通过efSearch参数调节实现,每个向量占用内存数为(d4+x2*4)字节。
HNSW不支持序列添加(not add_with_ids),同样的,如果需要,则使用IDMap。HNSW不需要训练,也不支持从index中删除向量。

内存不关注,使用"...,Flat"

"..."指的是需要进行的数据集聚类操作。聚类之后,“Flat”就把这些向量归于桶中,所以并没有亚索,所占的存储空间大小等于原数据集。速度-精度衡量通过nprobe参数控制。

内存比较有限,使用“PCARx,...,SQ8”

如果存储整个向量非常占内存资源,可以通过两种方法降低占用量:

  • PCA降维;
  • 每个向量元素进行可计量量化至1字节;

所以,存储总量是每个向量的x字节。

内存非常有限,使用“OPQx_y, ..., PQx”

PQx使用PQ算法将每个向量进行压缩输出x字节的编码,x一般来说小于等于64,对于大的编码,SQ通常更精确、快速。OPQ是将向量进行线性转换,使得其更容易压缩。y表示的是一个维度,需满足:

  • y是x的因子;
  • y<=d,d是输入向量的维度;
  • y <= 4*x

数据集有多大

这个问题就是对上文"..."部分的回答。数据集被聚成多个桶,在搜索时只有一部分桶被遍历(nprobe个桶)。这个聚类操作只作用于数据集向量的具有代表性的样例之上,通常是数据集的一个样本集。我们针对样本集给出其最优尺寸选择:

如果低于1M向量:"...,IVFx,..."

其中x是4sqrt(N)16sqrt(N),N是数据集的大小。这只是使用kmeans将向量聚起来,你需要30x256x向量训练(更多越好)。

如果1M-10M:"...,IMI2x10,..."

其中x是符号,不是数字。
IMI也对训练集进行k-means聚类,聚成210中心点,但是在向量的前半部分和后半部分是独立的。增加聚类数到2*(210)。你需要大约64*(2^10)向量训练。

如果10M-100M:"...,IMI2x12,..."<千万级-亿级>

同上面的,10用12替换。

如果100M-1B:"...,IMI2X14,..."<亿级-十亿级>

同上面的,10用14替换。

 

 

https://www.cnblogs.com/imagezy/p/8329229.html

 

 

版权声明
本文为[osc_bk3qeylz]所创,转载请带上原文链接,感谢
https://my.oschina.net/u/4403337/blog/4839571

Scroll to Top