跳到正文
原文
arXiv cs.CV· Dongfang Zhao·· 3 小时前AI 评分39

CertEx:通过认证排除实现大规模精确视觉检索

CertEx: Exact Visual Retrieval at Scale by Certified Exclusion

AI 导读

CertEx 提出基于"认证排除"的 GPU 精确 kNN 方法,在 16 个最多 100M 向量的数据集上对每条查询都返回精确结果。它用半精度存储每个向量的下界投影,并把舍入证书折进同一个 tensor-core 点积,从而只对少数保留向量计算精确距离。其吞吐量达单张 H100 上最快 cuVS 搜索的 7.1× 到 18.0×,在每条查询都必须精确时优于量化与图索引。

来源:arXiv cs.CV · arxiv.org