Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

HGAN: Hierarchical Graph Alignment Network for Image-Text Retrieval

About

Image-text retrieval (ITR) is a challenging task in the field of multimodal information processing due to the semantic gap between different modalities. In recent years, researchers have made great progress in exploring the accurate alignment between image and text. However, existing works mainly focus on the fine-grained alignment between image regions and sentence fragments, which ignores the guiding significance of context background information. Actually, integrating the local fine-grained information and global context background information can provide more semantic clues for retrieval. In this paper, we propose a novel Hierarchical Graph Alignment Network (HGAN) for image-text retrieval. First, to capture the comprehensive multimodal features, we construct the feature graphs for the image and text modality respectively. Then, a multi-granularity shared space is established with a designed Multi-granularity Feature Aggregation and Rearrangement (MFAR) module, which enhances the semantic corresponding relations between the local and global information, and obtains more accurate feature representations for the image and text modalities. Finally, the ultimate image and text features are further refined through three-level similarity functions to achieve the hierarchical alignment. To justify the proposed model, we perform extensive experiments on MS-COCO and Flickr30K datasets. Experimental results show that the proposed HGAN outperforms the state-of-the-art methods on both datasets, which demonstrates the effectiveness and superiority of our model.

Jie Guo, Meiting Wang, Yan Zhou, Bin Song, Yuhao Chi, Wei Fan, Jianglong Chang• 2022

Related benchmarks

TaskDatasetResultRank
Image-to-Text RetrievalFlickr30K 1K (test)
R@180.3
439
Text-to-Image RetrievalFlickr30K 1K (test)
R@162.3
375
Text-to-Image RetrievalMS-COCO 5K (test)
R@145.4
223
Image-to-Sentence RetrievalMS-COCO 5K full (test)
Recall@160
35
Sentence-to-Image RetrievalMS-COCO 1K (5-folds) (test)
R@167.4
29
Image-to-Sentence RetrievalMS-COCO 1K (5-folds) (test)
Recall@181.1
29
Sentence RetrievalFlickr30K 1K (test)
R Sum518.3
26
Showing 7 of 7 rows

Other info

Follow for update