LLM-Augmented Retrieval: Enhancing Retrieval Models Through Language Models and Doc-Level Embedding

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Mingrui, Cao, Sheng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914744978374656
author Wu, Mingrui
Cao, Sheng
author_facet Wu, Mingrui
Cao, Sheng
contents Recently embedding-based retrieval or dense retrieval have shown state of the art results, compared with traditional sparse or bag-of-words based approaches. This paper introduces a model-agnostic doc-level embedding framework through large language model (LLM) augmentation. In addition, it also improves some important components in the retrieval model training process, such as negative sampling, loss function, etc. By implementing this LLM-augmented retrieval framework, we have been able to significantly improve the effectiveness of widely-used retriever models such as Bi-encoders (Contriever, DRAGON) and late-interaction models (ColBERTv2), thereby achieving state-of-the-art results on LoTTE datasets and BEIR datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2404_05825
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LLM-Augmented Retrieval: Enhancing Retrieval Models Through Language Models and Doc-Level Embedding
Wu, Mingrui
Cao, Sheng
Information Retrieval
Artificial Intelligence
Recently embedding-based retrieval or dense retrieval have shown state of the art results, compared with traditional sparse or bag-of-words based approaches. This paper introduces a model-agnostic doc-level embedding framework through large language model (LLM) augmentation. In addition, it also improves some important components in the retrieval model training process, such as negative sampling, loss function, etc. By implementing this LLM-augmented retrieval framework, we have been able to significantly improve the effectiveness of widely-used retriever models such as Bi-encoders (Contriever, DRAGON) and late-interaction models (ColBERTv2), thereby achieving state-of-the-art results on LoTTE datasets and BEIR datasets.
title LLM-Augmented Retrieval: Enhancing Retrieval Models Through Language Models and Doc-Level Embedding
topic Information Retrieval
Artificial Intelligence
url https://arxiv.org/abs/2404.05825