QAEA-DR: A Unified Text Augmentation Framework for Dense Retrieval

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Tan, Hongming, Zhan, Shaoxiong, Lin, Hai, Zheng, Hai-Tao, Chan, Wai Kin
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910852675796992
author Tan, Hongming
Zhan, Shaoxiong
Lin, Hai
Zheng, Hai-Tao
Chan, Wai Kin
author_facet Tan, Hongming
Zhan, Shaoxiong
Lin, Hai
Zheng, Hai-Tao
Chan, Wai Kin
contents In dense retrieval, embedding long texts into dense vectors can result in information loss, leading to inaccurate query-text matching. Additionally, low-quality texts with excessive noise or sparse key information are unlikely to align well with relevant queries. Recent studies mainly focus on improving the sentence embedding model or retrieval process. In this work, we introduce a novel text augmentation framework for dense retrieval. This framework transforms raw documents into information-dense text formats, which supplement the original texts to effectively address the aforementioned issues without modifying embedding or retrieval methodologies. Two text representations are generated via large language models (LLMs) zero-shot prompting: question-answer pairs and element-driven events. We term this approach QAEA-DR: unifying question-answer generation and event extraction in a text augmentation framework for dense retrieval. To further enhance the quality of generated texts, a scoring-based evaluation and regeneration mechanism is introduced in LLM prompting. Our QAEA-DR model has a positive impact on dense retrieval, supported by both theoretical analysis and empirical experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2407_20207
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle QAEA-DR: A Unified Text Augmentation Framework for Dense Retrieval
Tan, Hongming
Zhan, Shaoxiong
Lin, Hai
Zheng, Hai-Tao
Chan, Wai Kin
Computation and Language
Artificial Intelligence
Information Retrieval
In dense retrieval, embedding long texts into dense vectors can result in information loss, leading to inaccurate query-text matching. Additionally, low-quality texts with excessive noise or sparse key information are unlikely to align well with relevant queries. Recent studies mainly focus on improving the sentence embedding model or retrieval process. In this work, we introduce a novel text augmentation framework for dense retrieval. This framework transforms raw documents into information-dense text formats, which supplement the original texts to effectively address the aforementioned issues without modifying embedding or retrieval methodologies. Two text representations are generated via large language models (LLMs) zero-shot prompting: question-answer pairs and element-driven events. We term this approach QAEA-DR: unifying question-answer generation and event extraction in a text augmentation framework for dense retrieval. To further enhance the quality of generated texts, a scoring-based evaluation and regeneration mechanism is introduced in LLM prompting. Our QAEA-DR model has a positive impact on dense retrieval, supported by both theoretical analysis and empirical experiments.
title QAEA-DR: A Unified Text Augmentation Framework for Dense Retrieval
topic Computation and Language
Artificial Intelligence
Information Retrieval
url https://arxiv.org/abs/2407.20207