Information retrieval using an augmented query produced by graph convolution

Abstract

An information retrieval technique uses one or more machine-trained models to generate one or more metadata embeddings. The technique then combines a query embedding with the metadata embedding(s). In some cases, the technique performs this operation using a graph convolution operation. This yields an augmented embedding. The technique then uses the augmented embedding to retrieve at least one item. The augmented embedding lies in the same vector space as target-item embeddings associated with candidate target items. Otherwise, the vector spaces associated with the query embedding and metadata embedding(s) can be different. In some implementations, the technique use dense retrieval, which enables the technique to deliver output results in real time.