RAG over Tables: Hierarchical Memory Index, Multi-Stage Retrieval, and Benchmarking

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zou, Jiaru, Fu, Dongqi, Chen, Sirui, He, Xinrui, Li, Zihao, Zhu, Yada, Han, Jiawei, He, Jingrui
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911191506354176
author Zou, Jiaru
Fu, Dongqi
Chen, Sirui
He, Xinrui
Li, Zihao
Zhu, Yada
Han, Jiawei
He, Jingrui
author_facet Zou, Jiaru
Fu, Dongqi
Chen, Sirui
He, Xinrui
Li, Zihao
Zhu, Yada
Han, Jiawei
He, Jingrui
contents Retrieval-Augmented Generation (RAG) enhances Large Language Models (LLMs) by integrating them with an external knowledge base to improve the answer relevance and accuracy. In real-world scenarios, beyond pure text, a substantial amount of knowledge is stored in tables, and user questions often require retrieving answers that are distributed across multiple tables. Retrieving knowledge from a table corpora (i.e., various individual tables) for a question remains nascent, at least, for (i) how to understand intra- and inter-table knowledge effectively, (ii) how to filter unnecessary tables and how to retrieve the most relevant tables efficiently, (iii) how to prompt LLMs to infer over the retrieval, (iv) how to evaluate the corresponding performance in a realistic setting. Facing the above challenges, in this paper, we first propose a table-corpora-aware RAG framework, named T-RAG, which consists of the hierarchical memory index, multi-stage retrieval, and graph-aware prompting for effective and efficient table knowledge retrieval and inference. Further, we first develop a multi-table question answering benchmark named MultiTableQA, which spans 3 different task types, 57,193 tables, and 23,758 questions in total, and the sources are all from real-world scenarios. Based on MultiTableQA, we did the holistic comparison over table retrieval methods, RAG methods, and table-to-graph representation learning methods, where T-RAG shows the leading accuracy, recall, and running time performance. Also, under T-RAG, we evaluate the inference ability upgrade of different LLMs. Code and Data are available at https://github.com/jiaruzouu/T-RAG
format Preprint
id arxiv_https___arxiv_org_abs_2504_01346
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RAG over Tables: Hierarchical Memory Index, Multi-Stage Retrieval, and Benchmarking
Zou, Jiaru
Fu, Dongqi
Chen, Sirui
He, Xinrui
Li, Zihao
Zhu, Yada
Han, Jiawei
He, Jingrui
Computation and Language
Information Retrieval
Machine Learning
Retrieval-Augmented Generation (RAG) enhances Large Language Models (LLMs) by integrating them with an external knowledge base to improve the answer relevance and accuracy. In real-world scenarios, beyond pure text, a substantial amount of knowledge is stored in tables, and user questions often require retrieving answers that are distributed across multiple tables. Retrieving knowledge from a table corpora (i.e., various individual tables) for a question remains nascent, at least, for (i) how to understand intra- and inter-table knowledge effectively, (ii) how to filter unnecessary tables and how to retrieve the most relevant tables efficiently, (iii) how to prompt LLMs to infer over the retrieval, (iv) how to evaluate the corresponding performance in a realistic setting. Facing the above challenges, in this paper, we first propose a table-corpora-aware RAG framework, named T-RAG, which consists of the hierarchical memory index, multi-stage retrieval, and graph-aware prompting for effective and efficient table knowledge retrieval and inference. Further, we first develop a multi-table question answering benchmark named MultiTableQA, which spans 3 different task types, 57,193 tables, and 23,758 questions in total, and the sources are all from real-world scenarios. Based on MultiTableQA, we did the holistic comparison over table retrieval methods, RAG methods, and table-to-graph representation learning methods, where T-RAG shows the leading accuracy, recall, and running time performance. Also, under T-RAG, we evaluate the inference ability upgrade of different LLMs. Code and Data are available at https://github.com/jiaruzouu/T-RAG
title RAG over Tables: Hierarchical Memory Index, Multi-Stage Retrieval, and Benchmarking
topic Computation and Language
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2504.01346