Our new X account is live! Follow @wizwand_team for updates
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
460
Image-to-Text RetrievalFlickr30K 1K (test)
R@174.1
439
Text-to-Image RetrievalFlickr30k (test)
Recall@153.9
423
Image-to-Text RetrievalFlickr30K
R@174.1
379
Text-to-Image RetrievalFlickr30K 1K (test)
R@153.9
375
Image-to-Text RetrievalFlickr30k (test)
R@174.1
370
Image-to-Text RetrievalMS-COCO 5K (test)
R@169.9
299
Text-to-Image RetrievalMS-COCO 5K (test)
R@155.9
223
Image RetrievalMS-COCO 5K (test)
R@139.7
217
Image RetrievalFlickr30k (test)
R@153.9
195
Showing 10 of 50 rows

Other info

Code

Follow for update