HADA: A graph-based amalgamation framework in image-text retrieval
Nguyen, Manh-DuyORCID: 0000-0001-6878-7039, Nguyen, Thanh BinhORCID: 0000-0001-5249-9702 and Gurrin, CathalORCID: 0000-0003-2903-3968
(2023)
HADA: A graph-based amalgamation framework in image-text retrieval.
In: European Conference on Information Retrieval (ECIR 2023), 2-6 Apr 2023, Dublin, Ireland.
ISBN 978-3-031-28243-0
Many models have been proposed for vision and language tasks, especially the image-text retrieval task. State-of-the-art (SOTA) models in this challenge contain hundreds of millions of parameters. They also were pretrained on large external datasets that have been proven to significantly improve overall performance. However, it is not easy to propose a new model with a novel architecture and intensively train it on a massive dataset with many GPUs to surpass many SOTA models already available to use on the Internet. In this paper, we propose a compact graph-based framework named HADA, which can combine pretrained models to produce a better result rather than starting from scratch. Firstly, we created a graph structure in which the nodes were the features extracted from the pretrained models and the edges connecting them. The graph structure was employed to capture and fuse the information from every pretrained model. Then a graph neural network was applied to update the connection between the nodes to get the representative embedding vector for an image and text. Finally, we employed cosine similarity to match images with their relevant texts and vice versa to ensure a low inference time. Our experiments show that, although HADA contained a tiny number of trainable parameters, it could increase baseline performance by more than 3.6% in terms of evaluation metrics on the Flickr30k dataset. Additionally, the proposed model did not train on any external dataset and only required a single GPU to train due to the small number of parameters required. The source code is available at https://github.com/m2man/HADA.
Metadata
Item Type:
Conference or Workshop Item (Paper)
Event Type:
Conference
Refereed:
Yes
Uncontrolled Keywords:
Image-text retrieval; Graph neural network; Fusion model
Science Foundation Ireland grant number SFI/12/RC/2289, Science Foundation Ireland grant number SFI/13/RC/2106, Science Foundation Ireland grant number 18/CRT/6223
ID Code:
28883
Deposited On:
08 Aug 2023 10:00 by
Manh Duy Nguyen
. Last Modified 08 Aug 2023 10:29