Enhancing Knowledge Retrieval with In-Context Learning and Semantic Search through Generative AI

TL;DR

This paper introduces GTR and GTR-T, innovative retrieval models that combine generative AI with vector databases, significantly improving accuracy and efficiency in knowledge retrieval from diverse data sources without fine-tuning.

Contribution

The paper presents novel generative retrieval models that operate effectively on structured and unstructured data, achieving state-of-the-art performance without the need for domain-specific fine-tuning.

Findings

GTR achieved over 90% accuracy and 87% truthful outputs.

GTR outperformed existing models with a Rouge-L F1 score of 0.98 on MSMARCO.

GTR-T demonstrated high efficiency with 0.82 execution accuracy on Spider.

Abstract

Retrieving and extracting knowledge from extensive research documents and large databases presents significant challenges for researchers, students, and professionals in today's information-rich era. Existing retrieval systems, which rely on general-purpose Large Language Models (LLMs), often fail to provide accurate responses to domain-specific inquiries. Additionally, the high cost of pretraining or fine-tuning LLMs for specific domains limits their widespread adoption. To address these limitations, we propose a novel methodology that combines the generative capabilities of LLMs with the fast and accurate retrieval capabilities of vector databases. This advanced retrieval system can efficiently handle both tabular and non-tabular data, understand natural language user queries, and retrieve relevant information without fine-tuning. The developed model, Generative Text Retrieval (GTR), is adaptable to both unstructured and structured data with minor refinement. GTR was evaluated on both manually annotated and public datasets, achieving over 90% accuracy and delivering truthful outputs in 87% of cases. Our model achieved state-of-the-art performance with a Rouge-L F1 score of 0.98 on the MSMARCO dataset. The refined model, Generative Tabular Text Retrieval (GTR-T), demonstrated its efficiency in large database querying, achieving an Execution Accuracy (EX) of 0.82 and an Exact-Set-Match (EM) accuracy of 0.60 on the Spider dataset, using an open-source LLM. These efforts leverage Generative AI and In-Context Learning to enhance human-text interaction and make advanced AI capabilities more accessible. By integrating robust retrieval systems with powerful LLMs, our approach aims to democratize access to sophisticated AI tools, improving the efficiency, accuracy, and scalability of AI-driven information retrieval and database querying.