Revisiting Human-in-the-Loop Object Retrieval with Pre-Trained Vision Transformers

TL;DR

This paper explores improving human-in-the-loop object retrieval by leveraging pre-trained vision transformers and active learning to efficiently identify diverse object instances in complex images.

Contribution

It revisits the task using ViT representations, addressing key design choices and comparing strategies for local versus global feature extraction in multi-object datasets.

Findings

Pre-trained ViT representations enhance retrieval performance.

Local descriptors improve detection of small, cluttered objects.

Active learning strategies effectively reduce annotation effort.

Abstract

Building on existing approaches, we revisit Human-in-the-Loop Object Retrieval, a task that consists of iteratively retrieving images containing objects of a class-of-interest, specified by a user-provided query. Starting from a large unlabeled image collection, the aim is to rapidly identify diverse instances of an object category relying solely on the initial query and the user's Relevance Feedback, with no prior labels. The retrieval process is formulated as a binary classification task, where the system continuously learns to distinguish between relevant and non-relevant images to the query, through iterative user interaction. This interaction is guided by an Active Learning loop: at each iteration, the system selects informative samples for user annotation, thereby refining the retrieval performance. This task is particularly challenging in multi-object datasets, where the object of interest may occupy only a small region of the image within a complex, cluttered scene. Unlike object-centered settings where global descriptors often suffice, multi-object images require more adapted, localized descriptors. In this work, we formulate and revisit the Human-in-the-Loop Object Retrieval task by leveraging pre-trained ViT representations, and addressing key design questions, including which object instances to consider in an image, what form the annotations should take, how Active Selection should be applied, and which representation strategies best capture the object's features. We compare several representation strategies across multi-object datasets highlighting trade-offs between capturing the global context and focusing on fine-grained local object details. Our results offer practical insights for the design of effective interactive retrieval pipelines based on Active Learning for object class retrieval.