Papers › Nearest Neighbor Normalization Improves Multimodal Retrieval

Nearest Neighbor Normalization Improves Multimodal Retrieval

31 Oct 2024arXiv:2410.24114archive 2025-07-28

Neil Chowdhury, Franklin Wang, Sumedh Shenoy, Douwe Kiela, Sarah Schwettmann, Tristan Thrush

Multimodal models leverage large-scale pre-training to achieve strong but still imperfect performance on tasks such as image captioning, visual question answering, and cross-modal retrieval. In this paper, we present a simple and efficient method for correcting errors in trained contrastive image-text retrieval models with no additional training, called Nearest Neighbor Normalization (NNN). We show an improvement on retrieval metrics in both text retrieval and image retrieval for all of the contrastive models that we tested (CLIP, BLIP, ALBEF, SigLIP, BEiT) and for both of the datasets that we used (MS-COCO and Flickr30k). NNN requires a reference database, but does not require any training on this database, and can even increase the retrieval accuracy of a model after finetuning.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

multimodal-interpretability/nnn officialmentioned in papermentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Cross-Modal RetrievalImage CaptioningImage RetrievalImage-text RetrievalQuestion AnsweringRetrievalText RetrievalVisual Question Answering

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

ALBEFBLIP

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections