Cross-Modal and Uni-Modal Soft-Label Alignment for Image-Text Retrieval
About
Current image-text retrieval methods have demonstrated impressive performance in recent years. However, they still face two problems: the inter-modal matching missing problem and the intra-modal semantic loss problem. These problems can significantly affect the accuracy of image-text retrieval. To address these challenges, we propose a novel method called Cross-modal and Uni-modal Soft-label Alignment (CUSA). Our method leverages the power of uni-modal pre-trained models to provide soft-label supervision signals for the image-text retrieval model. Additionally, we introduce two alignment techniques, Cross-modal Soft-label Alignment (CSA) and Uni-modal Soft-label Alignment (USA), to overcome false negatives and enhance similarity recognition between uni-modal samples. Our method is designed to be plug-and-play, meaning it can be easily applied to existing image-text retrieval models without changing their original architectures. Extensive experiments on various image-text retrieval models and datasets, we demonstrate that our method can consistently improve the performance of image-text retrieval and achieve new state-of-the-art results. Furthermore, our method can also boost the uni-modal retrieval performance of image-text retrieval models, enabling it to achieve universal retrieval. The code and supplementary files can be found at https://github.com/lerogo/aaai24_itr_cusa.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image-to-Text Retrieval | Flickr30K 1K (test) | R@177.4 | 439 | |
| Text-to-Image Retrieval | Flickr30K 1K (test) | R@190.8 | 375 | |
| Text-to-Image Retrieval | MSCOCO 5K (test) | R@167.9 | 286 | |
| Image-to-Text Retrieval | MSCOCO 5K (test) | R@152.4 | 46 | |
| Image-to-Text Retrieval | ECCV Caption (test) | R@180.9 | 7 | |
| Text-to-Image Retrieval | ECCV Caption (test) | R@188.2 | 7 |