Product quantization search method and device, terminal and storage medium

Abstract

The invention is suitable for the technical field of computers, and provides a product quantization search method and device, a terminal and a storage medium. The product quantization search method specifically comprises the following steps: inputting a reference high-dimensional feature of a reference object into a feature compression network to obtain a reference low-dimensional feature output by the feature compression network, wherein the loss function of the feature compression network is a function obtained based on a high-dimensional neighbor relation of a sample object and a low-dimensional neighbor relation of the sample object; determining a plurality of clustering centers by using the reference low-dimensional features; obtaining a target feature of the target object; and performing product quantization search by using the target feature and the plurality of clustering centers to obtain a reference object closest to the target object. According to the embodiment of the invention, the precision of product quantization search can be improved.

Description

technical field [0001] The present application belongs to the field of computer technology, and in particular relates to a product quantization search method, device, terminal and storage medium. Background technique [0002] The main goal of the approximate product quantization search algorithm is to retrieve multiple data feature vectors most similar to a given query object from a database containing a large number of data feature vectors under a determined similarity measurement criterion. Approximate product quantitative search is the basis of information retrieval, and it is widely used in various search engines and recommendation systems. How to quickly and accurately implement approximate product quantization search under the condition of limited hardware cost has always been a research hotspot in the field of information retrieval. [0003] Search algorithms based on Product Quantization (PQ) are a class of algorithms that are widely used at present. The PQ search ...

AI Technical Summary

Problems solved by technology

[0004] However, directly quantizing from high-dimensional floating-point data to low-dimensional integer data will cause more information loss, which will lead to a decrease in search accuracy.

Images