mirrors
/
faiss
mirror of https://github.com/facebookresearch/faiss.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Exhaustive IVF probing in scalar quantizer tests (#4075) 

Summary:
Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4075

`TestScalarQuantizer.test_4variants_ivf` shouldn't mix IVF probing misses in with its evaluation. As is, it probes 4/64 centroids, so FP16 has only 73% recall. This fixes it to *still exercise residual encoding* and the resulting distributions, but exhaustively scan the index.

Reviewed By: junjieqi

Differential Revision: D66909687

fbshipit-source-id: f4ec32b59fe7885415489f82e6a17ce03eb10916
pull/4093/head
Tom Jackson 2024-12-15 14:44:53 -08:00 committed by Facebook GitHub Bot
parent 87a760e631
commit bfdda6ab5f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_index.py
View File

@ -319,10 +319,11 @@ class TestScalarQuantizer(unittest.TestCase):
nok = {} nok = {}
nprobe = 64 # Probe all centroids, only exercise residual quantizer.
index = faiss.IndexIVFFlat(quantizer, d, ncent, index = faiss.IndexIVFFlat(quantizer, d, ncent,
faiss.METRIC_L2) faiss.METRIC_L2)
index.cp.min_points_per_centroid = 5 # quiet warning index.cp.min_points_per_centroid = 5 # quiet warning
index.nprobe = 4 index.nprobe = nprobe
index.train(xt) index.train(xt)
index.add(xb) index.add(xb)
D, I = index.search(xq, 10) D, I = index.search(xq, 10)
@ -333,7 +334,7 @@ class TestScalarQuantizer(unittest.TestCase):
index = faiss.IndexIVFScalarQuantizer(quantizer, d, ncent, index = faiss.IndexIVFScalarQuantizer(quantizer, d, ncent,
qtype, faiss.METRIC_L2) qtype, faiss.METRIC_L2)
index.nprobe = 4 index.nprobe = nprobe
index.train(xt) index.train(xt)
index.add(xb) index.add(xb)
D, I = index.search(xq, 10) D, I = index.search(xq, 10)
@ -347,7 +348,7 @@ class TestScalarQuantizer(unittest.TestCase):
# jitter # jitter
self.assertGreaterEqual(nok['flat'], nok['QT_8bit']) self.assertGreaterEqual(nok['flat'], nok['QT_8bit'])
self.assertGreaterEqual(nok['QT_8bit'], nok['QT_4bit']) self.assertGreaterEqual(nok['QT_8bit'], nok['QT_4bit'])
self.assertGreaterEqual(nok['QT_8bit'], nok['QT_8bit_uniform']) # flaky: self.assertGreaterEqual(nok['QT_8bit'], nok['QT_8bit_uniform'])
self.assertGreaterEqual(nok['QT_4bit'], nok['QT_4bit_uniform']) self.assertGreaterEqual(nok['QT_4bit'], nok['QT_4bit_uniform'])
self.assertGreaterEqual(nok['QT_fp16'], nok['QT_8bit']) self.assertGreaterEqual(nok['QT_fp16'], nok['QT_8bit'])
self.assertGreaterEqual(nok['QT_bf16'], nok['QT_8bit']) self.assertGreaterEqual(nok['QT_bf16'], nok['QT_8bit'])
@ -376,7 +377,7 @@ class TestScalarQuantizer(unittest.TestCase):
self.assertGreaterEqual(nok['QT_8bit'], nq * 0.9) self.assertGreaterEqual(nok['QT_8bit'], nq * 0.9)
self.assertGreaterEqual(nok['QT_8bit'], nok['QT_4bit']) self.assertGreaterEqual(nok['QT_8bit'], nok['QT_4bit'])
self.assertGreaterEqual(nok['QT_8bit'], nok['QT_8bit_uniform']) # flaky: self.assertGreaterEqual(nok['QT_8bit'], nok['QT_8bit_uniform'])
self.assertGreaterEqual(nok['QT_4bit'], nok['QT_4bit_uniform']) self.assertGreaterEqual(nok['QT_4bit'], nok['QT_4bit_uniform'])
self.assertGreaterEqual(nok['QT_fp16'], nok['QT_8bit']) self.assertGreaterEqual(nok['QT_fp16'], nok['QT_8bit'])
self.assertGreaterEqual(nok['QT_bf16'], nq * 0.9) self.assertGreaterEqual(nok['QT_bf16'], nq * 0.9)