Needle: A Generative AI-Powered Multi-modal Database for Answering Complex Natural Language Queries

TL;DR

Needle introduces a generative AI-powered multi-modal database that enhances complex natural language query answering in image retrieval by generating synthetic samples, outperforming existing methods and supporting easy integration of foundation models.

Contribution

The paper presents Needle, a novel generative-based approach for multi-modal data retrieval that leverages foundation models to generate synthetic samples, improving complex query answering.

Findings

Outperforms state-of-the-art text-to-image retrieval methods

Utilizes generative models for capturing query complexity

Supports easy integration of various foundation models

Abstract

Multi-modal datasets, like those involving images, often miss the detailed descriptions that properly capture the rich information encoded in each item. This makes answering complex natural language queries a major challenge in this domain. In particular, unlike the traditional nearest neighbor search, where the tuples and the query are represented as points in a single metric space, these settings involve queries and tuples embedded in fundamentally different spaces, making the traditional query answering methods inapplicable. Existing literature addresses this challenge for image datasets through vector representations jointly trained on natural language and images. This technique, however, underperforms for complex queries due to various reasons. This paper takes a step towards addressing this challenge by introducing a Generative-based Monte Carlo method that utilizes foundation models to generate synthetic samples that capture the complexity of the natural language query and represent it in the same metric space as the multi-modal data. Following this method, we propose Needle, a database for image data retrieval. Instead of relying on contrastive learning or metadata-searching approaches, our system is based on synthetic data generation to capture the complexities of natural language queries. Our system is open-source and ready for deployment, designed to be easily adopted by researchers and developers. The comprehensive experiments on various benchmark datasets verify that this system significantly outperforms state-of-the-art text-to-image retrieval methods in the literature. Any foundation model and embedder can be easily integrated into Needle to improve the performance, piggybacking on the advancements in these technologies.