Are They the Same Picture? Adapting Concept Bottleneck Models for Human-AI Collaboration in Image Retrieval


Image retrieval plays a pivotal role in wildlife conservation, specifically in state-of-the-art platforms like ElephantBook for finding individual animals. Although deep learning techniques for image retrieval have advanced significantly, their imperfect real-world performance often necessitates including human expertise. Human-in-the-loop approaches typically rely on humans completing the task independently and then combining their opinions with an AI model in various ways, as these models offer very little interpretability or correctability. To allow humans to intervene in the AI model instead, thereby saving human time and effort, we adapt the Concept Bottleneck Model (CBM) and propose CHAIR. CHAIR (a) enables humans to correct intermediate concepts, which helps improve embeddings generated, and (b) allows for flexible levels of intervention that accommodate varying levels of human expertise for better retrieval. To show the efficacy of CHAIR, we demonstrate that our method performs better than similar models on image retrieval metrics without any external intervention. Furthermore, we also showcase how human intervention helps further improve retrieval performance, thereby achieving human-AI complementarity.

IJCAI'24 Human-Centered Artificial Intelligence Track (<5 % Acceptance Rate)
Click the Cite button above to cite this work.
Vaibhav Balloli
Vaibhav Balloli
Ph.D. student

My research interests include Multi Agent Systems, Reinforcement Learning and AI for Social Good.