QCG-Rerank: Chunks Graph Rerank with Query Expansion in Retrieval-Augmented LLMs for Tourism Domain

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wei, Qikai, Yang, Mingzhi, Han, Chunlong, Wei, Jingfu, Zhang, Minghao, Shi, Feifei, Ning, Huansheng
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917836166791168
author Wei, Qikai
Yang, Mingzhi
Han, Chunlong
Wei, Jingfu
Zhang, Minghao
Shi, Feifei
Ning, Huansheng
author_facet Wei, Qikai
Yang, Mingzhi
Han, Chunlong
Wei, Jingfu
Zhang, Minghao
Shi, Feifei
Ning, Huansheng
contents Retrieval-Augmented Generation (RAG) mitigates the issue of hallucination in Large Language Models (LLMs) by integrating information retrieval techniques. However, in the tourism domain, since the query is usually brief and the content in the database is diverse, existing RAG may contain a significant amount of irrelevant or contradictory information contents after retrieval. To address this challenge, we propose the QCG-Rerank model. This model first performs an initial retrieval to obtain candidate chunks and then enhances semantics by extracting critical information to expand the original query. Next, we utilize the expanded query and candidate chunks to calculate similarity scores as the initial transition probability and construct the chunks graph. Subsequently, We iteratively compute the transition probabilities based on an initial estimate until convergence. The chunks with the highest score are selected and input into the LLMs to generate responses. We evaluate the model on Cultour, IIRC, StrategyQA, HotpotQA, SQuAD, and MuSiQue datasets. The experimental results demonstrate the effectiveness and superiority of the QCG-Rerank method.
format Preprint
id arxiv_https___arxiv_org_abs_2411_08724
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle QCG-Rerank: Chunks Graph Rerank with Query Expansion in Retrieval-Augmented LLMs for Tourism Domain
Wei, Qikai
Yang, Mingzhi
Han, Chunlong
Wei, Jingfu
Zhang, Minghao
Shi, Feifei
Ning, Huansheng
Computation and Language
Artificial Intelligence
Retrieval-Augmented Generation (RAG) mitigates the issue of hallucination in Large Language Models (LLMs) by integrating information retrieval techniques. However, in the tourism domain, since the query is usually brief and the content in the database is diverse, existing RAG may contain a significant amount of irrelevant or contradictory information contents after retrieval. To address this challenge, we propose the QCG-Rerank model. This model first performs an initial retrieval to obtain candidate chunks and then enhances semantics by extracting critical information to expand the original query. Next, we utilize the expanded query and candidate chunks to calculate similarity scores as the initial transition probability and construct the chunks graph. Subsequently, We iteratively compute the transition probabilities based on an initial estimate until convergence. The chunks with the highest score are selected and input into the LLMs to generate responses. We evaluate the model on Cultour, IIRC, StrategyQA, HotpotQA, SQuAD, and MuSiQue datasets. The experimental results demonstrate the effectiveness and superiority of the QCG-Rerank method.
title QCG-Rerank: Chunks Graph Rerank with Query Expansion in Retrieval-Augmented LLMs for Tourism Domain
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2411.08724