Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

IMRAM: Iterative Matching with Recurrent Attention Memory for Cross-Modal Image-Text Retrieval

About

Enabling bi-directional retrieval of images and texts is important for understanding the correspondence between vision and language. Existing methods leverage the attention mechanism to explore such correspondence in a fine-grained manner. However, most of them consider all semantics equally and thus align them uniformly, regardless of their diverse complexities. In fact, semantics are diverse (i.e. involving different kinds of semantic concepts), and humans usually follow a latent structure to combine them into understandable languages. It may be difficult to optimally capture such sophisticated correspondences in existing methods. In this paper, to address such a deficiency, we propose an Iterative Matching with Recurrent Attention Memory (IMRAM) method, in which correspondences between images and texts are captured with multiple steps of alignments. Specifically, we introduce an iterative matching scheme to explore such fine-grained correspondence progressively. A memory distillation unit is used to refine alignment knowledge from early steps to later ones. Experiment results on three benchmark datasets, i.e. Flickr8K, Flickr30K, and MS COCO, show that our IMRAM achieves state-of-the-art performance, well demonstrating its effectiveness. Experiments on a practical business advertisement dataset, named \Ads{}, further validates the applicability of our method in practical scenarios.

Hui Chen, Guiguang Ding, Xudong Liu, Zijia Lin, Ji Liu, Jungong Han• 2020

Related benchmarks

TaskDatasetResultRank
Text-to-Image RetrievalFlickr30K
R@153.9
531
Image-to-Text RetrievalFlickr30K 1K (test)
R@174.1
491
Text-to-Image RetrievalFlickr30k (test)
Recall@153.9
445
Text-to-Image RetrievalFlickr30K 1K (test)
R@153.9
432
Image-to-Text RetrievalFlickr30K
R@174.1
429
Image-to-Text RetrievalFlickr30k (test)
R@174.1
392
Image-to-Text RetrievalMS-COCO 5K (test)
R@169.9
320
Text-to-Image RetrievalMS-COCO 5K (test)
R@155.9
244
Image RetrievalMS-COCO 5K (test)
R@139.7
217
Image RetrievalFlickr30k (test)
R@153.9
210
Showing 10 of 51 rows

Other info

Code

Follow for update