[发明专利]向量索引构建方法、向量检索方法和与方法对应的系统

说明书

本申请提出一种向量索引构建方法、向量检索方法和与方法对应的系统，涉及向量搜索技术领域，其中，该方法包括：获取待构建对象的向量；根据所述待构建对象的向量构建多个第一聚类空间，并分别确定每个第一聚类空间的聚类中心；根据所述多个第一聚类空间内的多个向量和所述多个第一聚类空间的聚类中心的距离，分别以所述每个第一聚类空间的聚类中心为中心扩大对应的第一聚类空间，得到多个第二聚类空间；根据所述多个第二聚类空间和对应的中心构建向量索引。该方案可以应用在图像检索、人脸对比、导航推荐等领域。

技术领域

本申请涉及向量搜索技术领域，尤其涉及一种向量索引构建方法、向量检索方法和与方法对应的系统。

背景技术

目前，随着深度学习的使用，在图像检索、人脸对比、导航推荐等应用场景的系统的架构变成了如DSSM等“双塔”架构，即将用户和物品映射到同一向量空间，然后采用相似度计算的方式进行向量索引召回距离最近的若干个向量。

现有技术中，一般采用聚类方法建立向量索引，然而采用聚类方法建立索引时某一个向量仅能归属于一个聚类中心，使向量检索时聚类空间边缘点误差较大，进而导致向量检索的准确度较低。

发明内容

本申请提供一种向量索引构建方法、向量检索方法和与方法对应的系统，能够提高向量检索的准确度。

一方面，本申请提供一种向量索引构建方法，包括：获取待构建对象的向量；根据所述待构建对象的向量构建多个第一聚类空间，并分别确定每个第一聚类空间的聚类中心；根据所述多个第一聚类空间内的多个向量和所述多个第一聚类空间的聚类中心的距离，分别以所述每个第一聚类空间的聚类中心为中心扩大对应的第一聚类空间，得到多个第二聚类空间；根据所述多个第二聚类空间和对应的中心构建向量索引。

另一方面，本申请提供一种向量检索方法，包括：获取待检索对象的向量；从上述的向量索引中确定与所述待检索对象的向量对应的目标聚类中心；根据所述目标聚类中心对应的第二聚类空间确定多个待匹配向量；以及通过多次并行处理方法根据预设规约从所述多个待匹配向量之中确定所述待检索对象的向量对应的至少一个目标向量。

又一方面，本申请提供一种向量索引构建系统，包括：

构建获取模块，用于获取待构建对象的向量；

聚类模块，与所述构建获取模块相连，用于根据所述待构建对象的向量构建多个第一聚类空间，并分别确定每个第一聚类空间的聚类中心；

扩大模块，与所述聚类模块相连，用根据所述多个第一聚类空间内的多个向量和所述多个第一聚类空间的聚类中心的距离，分别以所述每个第一聚类空间的聚类中心为中心扩大对应的第一聚类空间，得到多个第二聚类空间；以及

索引构建模块，与所述扩大模块相连，用于根据所述多个第二聚类空间和对应的中心构建向量索引。

又一方面，本申请提供一种向量检索系统，包括：

检索获取模块，用于获取待检索对象的向量；

中心确定模块，与所述检索获取模块相连，用于从如权利要求12-18任一所述的向量索引中确定与所述待检索对象的向量对应的目标聚类中心；

匹配获取模块，与所述中心确定模块相连，用于根据所述目标聚类中心对应的第二聚类空间确定多个待匹配向量；以及

目标获取模块，与所述匹配获取模块相连，用于通过多次并行处理方法根据预设规约从所述多个待匹配向量之中确定所述待检索对象的向量对应的至少一个目标向量。

又一方面，本申请提供一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，处理器执行计算机程序时，实现上述向量索引构建方法或向量检索方法。