Find your Needle: Small Object Image Retrieval via Multi-Object Attention Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Green, Michael, Levy, Matan, Tzachor, Issar, Samuel, Dvir, Darshan, Nir, Ben-Ari, Rami
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918329260703744
author Green, Michael
Levy, Matan
Tzachor, Issar
Samuel, Dvir
Darshan, Nir
Ben-Ari, Rami
author_facet Green, Michael
Levy, Matan
Tzachor, Issar
Samuel, Dvir
Darshan, Nir
Ben-Ari, Rami
contents We address the challenge of Small Object Image Retrieval (SoIR), where the goal is to retrieve images containing a specific small object, in a cluttered scene. The key challenge in this setting is constructing a single image descriptor, for scalable and efficient search, that effectively represents all objects in the image. In this paper, we first analyze the limitations of existing methods on this challenging task and then introduce new benchmarks to support SoIR evaluation. Next, we introduce Multi-object Attention Optimization (MaO), a novel retrieval framework which incorporates a dedicated multi-object pre-training phase. This is followed by a refinement process that leverages attention-based feature extraction with object masks, integrating them into a single unified image descriptor. Our MaO approach significantly outperforms existing retrieval methods and strong baselines, achieving notable improvements in both zero-shot and lightweight multi-object fine-tuning. We hope this work will lay the groundwork and inspire further research to enhance retrieval performance for this highly practical task. Code and Data are available on our project page: $\href{https://pihash2k.github.io/findyourneedle.github.io}{https://pihash2k.github.io/findyourneedle.github.io}$.
format Preprint
id arxiv_https___arxiv_org_abs_2503_07038
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Find your Needle: Small Object Image Retrieval via Multi-Object Attention Optimization
Green, Michael
Levy, Matan
Tzachor, Issar
Samuel, Dvir
Darshan, Nir
Ben-Ari, Rami
Computer Vision and Pattern Recognition
We address the challenge of Small Object Image Retrieval (SoIR), where the goal is to retrieve images containing a specific small object, in a cluttered scene. The key challenge in this setting is constructing a single image descriptor, for scalable and efficient search, that effectively represents all objects in the image. In this paper, we first analyze the limitations of existing methods on this challenging task and then introduce new benchmarks to support SoIR evaluation. Next, we introduce Multi-object Attention Optimization (MaO), a novel retrieval framework which incorporates a dedicated multi-object pre-training phase. This is followed by a refinement process that leverages attention-based feature extraction with object masks, integrating them into a single unified image descriptor. Our MaO approach significantly outperforms existing retrieval methods and strong baselines, achieving notable improvements in both zero-shot and lightweight multi-object fine-tuning. We hope this work will lay the groundwork and inspire further research to enhance retrieval performance for this highly practical task. Code and Data are available on our project page: $\href{https://pihash2k.github.io/findyourneedle.github.io}{https://pihash2k.github.io/findyourneedle.github.io}$.
title Find your Needle: Small Object Image Retrieval via Multi-Object Attention Optimization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.07038