| 参数 m | 子空间个数,d 需能被 m 整除 |
|---|---|
| nbits | 每段码本位数,常取 8(256 个质心) |
| 存储 | 每条向量约 m 字节,替代 4·d 字节 |
FAISS IVFPQ 示例
FAISS IVFPQ 示例
import faiss, numpy as np
d = 768
quantizer = faiss.IndexFlatL2(d)
index = faiss.IndexIVFPQ(quantizer, d, 1024, 96, 8) # nlist=1024, m=96, nbits=8
index.train(np.random.rand(200000, d).astype("float32"))
index.add(np.random.rand(2000000, d).astype("float32"))
index.nprobe = 32PQ 与 IVF 常组合使用:先聚类粗排、段内量化近似距离,再做重排补偿精度。