HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Similarity Reasoning and Filtration for Image-Text Matching

Haiwen Diao Ying Zhang Lin Ma Huchuan Lu

Similarity Reasoning and Filtration for Image-Text Matching

Abstract

Image-text matching plays a critical role in bridging the vision and language, and great progress has been made by exploiting the global alignment between image and sentence, or local alignments between regions and words. However, how to make the most of these alignments to infer more accurate matching scores is still underexplored. In this paper, we propose a novel Similarity Graph Reasoning and Attention Filtration (SGRAF) network for image-text matching. Specifically, the vector-based similarity representations are firstly learned to characterize the local and global alignments in a more comprehensive manner, and then the Similarity Graph Reasoning (SGR) module relying on one graph convolutional neural network is introduced to infer relation-aware similarities with both the local and global alignments. The Similarity Attention Filtration (SAF) module is further developed to integrate these alignments effectively by selectively attending on the significant and representative alignments and meanwhile casting aside the interferences of non-meaningful alignments. We demonstrate the superiority of the proposed method with achieving state-of-the-art performances on the Flickr30K and MSCOCO datasets, and the good interpretability of SGR and SAF modules with extensive qualitative experiments and analyses.

Code Repositories

Paranioar/SGRAF
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
cross-modal-retrieval-on-coco-2014SGRAF
Image-to-text R@1: 57.8
Image-to-text R@10: 91.6
Image-to-text R@5: 84.9
Text-to-image R@1: 41.9
Text-to-image R@10: 81.3
Text-to-image R@5: 70.7
cross-modal-retrieval-on-flickr30kSGRAF
Image-to-text R@1: 77.8
Image-to-text R@10: 97.4
Image-to-text R@5: 94.1
Text-to-image R@1: 58.5
Text-to-image R@10: 88.8
Text-to-image R@5: 83.0
image-retrieval-on-flickr30k-1k-testSGRAF
R@1: 58.5
R@10: 88.8
R@5: 83.0

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Similarity Reasoning and Filtration for Image-Text Matching | Papers | HyperAI