finished

Author: TabibitoQZP

README.md

@@ -1,5 +1,9 @@
 # zk-IVF-PQ
 
+## Technical Note
+
+Implementation details are provided in our [technical note](Technical_Details_of_V3DB.pdf).
+
 ## Build
 
 Build and install the Python extension:

Technical_Details_of_V3DB.pdf

@@ -6,31 +6,31 @@ set -euo pipefail
 
 # Experiment 1 / Classic ANN (Table "Experiment 1 datasets and IVF-PQ layouts")
 # SIFT1M high-acc: (n_list, n_probe, n) = (8192, 64, 256)
-# python -m tests.acc_bench \
-#   --data-name sift \
-#   --name gpu5_exp1_sift_high_acc_multik \
-#   --num-runs 1 \
-#   --M 8 \
-#   --K 256 \
-#   --top-k 100 \
-#   --report-ks 1,10,50,100,500 \
-#   --scale-n 65536 \
-#   --n-list 8192 \
-#   --n-probe 64 \
-#   --cluster-bound 256 \
-#   --layout none
-#
-# sleep 60
+python -m tests.acc_bench \
+  --data-name sift \
+  --name exp1_sift_high_acc_multik \
+  --num-runs 1 \
+  --M 8 \
+  --K 256 \
+  --top-k 100 \
+  --report-ks 10,50 \
+  --scale-n 65536 \
+  --n-list 8192 \
+  --n-probe 64 \
+  --cluster-bound 256 \
+  --layout none
+
+sleep 60
 
 # GIST1M high-acc: (n_list, n_probe, n) = (8192, 64, 256)
-CUDA_VISIBLE_DEVICES=1,2,3,4,5,6,7 python -m tests.acc_bench \
+python -m tests.acc_bench \
   --data-name gist \
-  --name gpu7_exp1_gist_high_acc_multik \
+  --name exp1_gist_high_acc_multik \
   --num-runs 1 \
   --M 8 \
   --K 256 \
   --top-k 100 \
-  --report-ks 1,10,50,100,500 \
+  --report-ks 10,50 \
   --scale-n 65536 \
   --n-list 8192 \
   --n-probe 64 \
@@ -40,31 +40,31 @@ CUDA_VISIBLE_DEVICES=1,2,3,4,5,6,7 python -m tests.acc_bench \
 sleep 60
 
 # SIFT1M zk-opt: (n_list, n_probe, n) = (1024, 8, 2048)
-# python -m tests.acc_bench \
-#   --data-name sift \
-#   --name gpu5_exp1_sift_zk_opt_multik \
-#   --num-runs 1 \
-#   --M 8 \
-#   --K 256 \
-#   --top-k 100 \
-#   --report-ks 1,10,50,100,500 \
-#   --scale-n 65536 \
-#   --n-list 1024 \
-#   --n-probe 8 \
-#   --cluster-bound 2048 \
-#   --layout none
-#
-# sleep 60
+python -m tests.acc_bench \
+  --data-name sift \
+  --name exp1_sift_zk_opt_multik \
+  --num-runs 1 \
+  --M 8 \
+  --K 256 \
+  --top-k 100 \
+  --report-ks 10,50 \
+  --scale-n 65536 \
+  --n-list 1024 \
+  --n-probe 8 \
+  --cluster-bound 2048 \
+  --layout none
+
+sleep 60
 
 # GIST1M zk-opt: (n_list, n_probe, n) = (512, 4, 4096)
-CUDA_VISIBLE_DEVICES=1,2,3,4,5,6,7 python -m tests.acc_bench \
+python -m tests.acc_bench \
   --data-name gist \
-  --name gpu7_exp1_gist_zk_opt_multik \
+  --name exp1_gist_zk_opt_multik \
   --num-runs 1 \
   --M 8 \
   --K 256 \
   --top-k 100 \
-  --report-ks 1,10,50,100,500 \
+  --report-ks 10,50 \
   --scale-n 65536 \
   --n-list 512 \
   --n-probe 4 \