Method and apparatus for training an image-text mutual retrieval model, image-text mutual retrieval method, and device

Abstract

Disclosed in the present disclosure are an image-text mutual retrieval model training method and apparatus, an image-text mutual retrieval method, and a device, applied to the technical field of retrieval. The image-text mutual retrieval model training method includes: acquiring training data pairs; inputting the training data pair in an initial model, and extracting text coding features of text training data and image coding features of image training data by using a text coding module and an image coding module in the initial model, respectively; calculating a training loss based on the text coding features and the image coding features, and performing parameter adjustment on the initial model based on the training loss; and in response to the training loss meeting a convergence condition, determining the initial model after the parameter adjustment as an image-text mutual retrieval model.