Balance Act: Mitigating Hubness in Cross-Modal Retrieval with Query and Gallery Banks

Yimu Wang,Xiangru Jian,Bo Xue
2023-10-18
Abstract:In this work, we present a post-processing solution to address the hubness problem in cross-modal retrieval, a phenomenon where a small number of gallery data points are frequently retrieved, resulting in a decline in retrieval performance. We first theoretically demonstrate the necessity of incorporating both the gallery and query data for addressing hubness as hubs always exhibit high similarity with gallery and query data. Second, building on our theoretical results, we propose a novel framework, Dual Bank Normalization (DBNorm). While previous work has attempted to alleviate hubness by only utilizing the query samples, DBNorm leverages two banks constructed from the query and gallery samples to reduce the occurrence of hubs during inference. Next, to complement DBNorm, we introduce two novel methods, dual inverted softmax and dual dynamic inverted softmax, for normalizing similarity based on the two banks. Specifically, our proposed methods reduce the similarity between hubs and queries while improving the similarity between non-hubs and queries. Finally, we present extensive experimental results on diverse language-grounded benchmarks, including text-image, text-video, and text-audio, demonstrating the superior performance of our approaches compared to previous methods in addressing hubness and boosting retrieval performance. Our code is available at <a class="link-external link-https" href="https://github.com/yimuwangcs/Better_Cross_Modal_Retrieval" rel="external noopener nofollow">this https URL</a>.
Machine Learning
What problem does this paper attempt to address?
The paper attempts to address the "hubness" problem in cross-modal retrieval. Specifically, the hubness phenomenon refers to the issue where a few data points are frequently retrieved, leading to a decline in retrieval performance. The paper first theoretically demonstrates the necessity of combining gallery and query data when dealing with hubness, as these central points often exhibit high similarity with both gallery and query data. Based on this theoretical result, the paper proposes a new framework—Dual Bank Normalization (DBNORM), which reduces the occurrence of central points by constructing two banks from query and gallery samples, respectively. Additionally, the paper introduces two new methods—Dual Inverted Softmax and Dynamic Dual Inverted Softmax, for normalizing similarity based on the two banks. These methods aim to reduce the similarity between central points and queries while increasing the similarity between non-central points and queries. Through extensive experiments on multiple language-guided benchmarks (including text-image, text-video, and text-audio retrieval), the paper demonstrates the superiority of the proposed methods in mitigating hubness and enhancing retrieval performance.