├── LICENSE ├── MRL.py ├── README.md ├── __init__.py ├── adanns ├── README.md ├── ablations │ ├── centroid_recall.ipynb │ ├── class_matches_per_cluster.ipynb │ ├── cluster-distribution.ipynb │ ├── plots.ipynb │ └── relative_contrast.ipynb ├── adanns-ivf-optimized.ipynb ├── adanns-ivf-unoptimized.ipynb ├── compute_metrics.ipynb ├── diskann │ ├── README.md │ └── adanns-diskann.ipynb ├── dpr-nq │ ├── README.md │ └── adanns-nq.ipynb ├── generate_nn │ ├── hnsw_exactl2.ipynb │ ├── ivf-experiments.ipynb │ └── ivfpq_opq_kmeans.ipynb └── utils.py ├── generate_embeddings ├── pytorch_inference.py └── run-inference.sh ├── images ├── accuracy-compute.png ├── adanns-opq.png ├── adanns-teaser.png ├── diskann-table.png ├── diskann-top1.png ├── encoders.png ├── flowchart.png ├── opq-1k.png └── opq-nq.png ├── requirements.txt └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/LICENSE -------------------------------------------------------------------------------- /MRL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/MRL.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adanns/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/README.md -------------------------------------------------------------------------------- /adanns/ablations/centroid_recall.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/ablations/centroid_recall.ipynb -------------------------------------------------------------------------------- /adanns/ablations/class_matches_per_cluster.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/ablations/class_matches_per_cluster.ipynb -------------------------------------------------------------------------------- /adanns/ablations/cluster-distribution.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/ablations/cluster-distribution.ipynb -------------------------------------------------------------------------------- /adanns/ablations/plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/ablations/plots.ipynb -------------------------------------------------------------------------------- /adanns/ablations/relative_contrast.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/ablations/relative_contrast.ipynb -------------------------------------------------------------------------------- /adanns/adanns-ivf-optimized.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/adanns-ivf-optimized.ipynb -------------------------------------------------------------------------------- /adanns/adanns-ivf-unoptimized.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/adanns-ivf-unoptimized.ipynb -------------------------------------------------------------------------------- /adanns/compute_metrics.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/compute_metrics.ipynb -------------------------------------------------------------------------------- /adanns/diskann/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/diskann/README.md -------------------------------------------------------------------------------- /adanns/diskann/adanns-diskann.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/diskann/adanns-diskann.ipynb -------------------------------------------------------------------------------- /adanns/dpr-nq/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/dpr-nq/README.md -------------------------------------------------------------------------------- /adanns/dpr-nq/adanns-nq.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/dpr-nq/adanns-nq.ipynb -------------------------------------------------------------------------------- /adanns/generate_nn/hnsw_exactl2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/generate_nn/hnsw_exactl2.ipynb -------------------------------------------------------------------------------- /adanns/generate_nn/ivf-experiments.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/generate_nn/ivf-experiments.ipynb -------------------------------------------------------------------------------- /adanns/generate_nn/ivfpq_opq_kmeans.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/generate_nn/ivfpq_opq_kmeans.ipynb -------------------------------------------------------------------------------- /adanns/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/adanns/utils.py -------------------------------------------------------------------------------- /generate_embeddings/pytorch_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/generate_embeddings/pytorch_inference.py -------------------------------------------------------------------------------- /generate_embeddings/run-inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/generate_embeddings/run-inference.sh -------------------------------------------------------------------------------- /images/accuracy-compute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/accuracy-compute.png -------------------------------------------------------------------------------- /images/adanns-opq.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/adanns-opq.png -------------------------------------------------------------------------------- /images/adanns-teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/adanns-teaser.png -------------------------------------------------------------------------------- /images/diskann-table.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/diskann-table.png -------------------------------------------------------------------------------- /images/diskann-top1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/diskann-top1.png -------------------------------------------------------------------------------- /images/encoders.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/encoders.png -------------------------------------------------------------------------------- /images/flowchart.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/flowchart.png -------------------------------------------------------------------------------- /images/opq-1k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/opq-1k.png -------------------------------------------------------------------------------- /images/opq-nq.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/images/opq-nq.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RAIVNLab/AdANNS/HEAD/utils.py --------------------------------------------------------------------------------