[发明专利]向量检索方法、装置和设备

说明书

本说明书实施例提供一种向量检索方法、装置和设备。该方法包括：提取检索对象的特征向量；从图形处理器GPU中获取与特征向量相似的向量，作为第一部分相似向量；从中央处理器CPU中获取与特征向量相似的向量，作为第二部分相似向量；根据第一部分相似向量和第二部分相似向量，确定检索对象的检索结果。

技术领域

本说明书一个或多个实施例涉及计算机技术领域，尤其涉及一种向量检索方法、装置和设备。

背景技术

随着互联网技术的发展，人们在检索感兴趣的内容时，检索对象变得越来越复杂。例如，由搜索文本关键词，逐渐发展为搜索图片、搜索音频数据和视频数据等。随着搜索和查询对象变得更加复杂，搜索的难度指数级增加。

通常复杂的检索对象会使用高维向量来表示，因此检索过程中通常需要对比诸多高维向量之间的距离或相似度。当需要检索的数据量很大时，涉及的计算量巨大和耗费较长检索时间太长，因此需要提供一种提高检索性能的向量检索方案。

发明内容

本说明书实施例提供一种向量检索方法、装置和设备，可以利用CPU和GPU对数据对象进行检索，提高检索效率。

第一方面，本说明书实施例提供一种向量检索方法，包括：

提取检索对象的特征向量；从图形处理器GPU中获取与特征向量相似的向量，作为第一部分相似向量；从中央处理器CPU中获取与特征向量相似的向量，作为第二部分相似向量；根据第一部分相似向量和第二部分相似向量，确定检索对象的检索结果。

第二方面，本说明书实施例提供一种向量检索装置，包括：

向量提取模块，用于提取检索对象的特征向量；第一检索模块，用于从图形处理器GPU中获取与所述特征向量相似的向量，作为第一部分相似向量；第二检索模块，用于从中央处理器CPU中获取与所述特征向量相似的向量，作为第二部分相似向量；结果确定模块，用于根据所述第一部分相似向量和所述第二部分相似向量，确定所述检索对象的检索结果。

第三方面，本说明书实施例提供一种向量检索设备，包括：存储器和处理器；该存储器用于存储程序；该处理器用于读取存储器中存储的可执行程序代码以执行上述第一方面的向量检索方法。

根据本说明书实施例中的向量检索方法、装置和设备，预先对向量集合进行切分，利用CPU和GPU分别对切分得到的数据进行针对检索对象的特征向量的相似度检索，在利用GPU提高运算速度的同时，有效利用CPU的资源。

附图说明

为了更清楚地说明本说明书实施例的技术方案，下面将对本说明书实施例中所需要使用的附图作简单地介绍，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。

图1示出本说明书一个实施例的检索系统的架构示意图；

图2示出本说明书一个实施例的向量检索方法的流程示意图；

图3示出本说明书一个实施例的向量检索方法的流程示意图；

图4示出本说明书另一个实施例的向量检索方法的流程示意图；

图5示出本说明书一个实施例的向量检索装置的结构示意图；

图6示出能够实现本说明书实施例的计算设备的示例性硬件架构的结构图。

具体实施方式

下面将详细描述本说明书的各个方面的特征和示例性实施例，为了使本说明书的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本说明书进行进一步详细描述。应理解，此处所描述的具体实施例仅被配置为解释本说明书，并不被配置为限定本说明书。对于本领域技术人员来说，本说明书可以在不需要这些具体细节中的一些细节的情况下实施。下面对实施例的描述仅仅是为了通过示出本说明书的示例来提供对本说明书更好的理解。