Hypercube-Based Retrieval-Augmented Generation for Scientific Question-Answering

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shi, Jimeng, Zhou, Sizhe, Jin, Bowen, Hu, Wei, Tian, Runchu, Wang, Shaowen, Narasimhan, Giri, Han, Jiawei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909720243077120
author Shi, Jimeng
Zhou, Sizhe
Jin, Bowen
Hu, Wei
Tian, Runchu
Wang, Shaowen
Narasimhan, Giri
Han, Jiawei
author_facet Shi, Jimeng
Zhou, Sizhe
Jin, Bowen
Hu, Wei
Tian, Runchu
Wang, Shaowen
Narasimhan, Giri
Han, Jiawei
contents Large language models (LLMs) often need to incorporate external knowledge to solve theme-specific problems. Retrieval-augmented generation (RAG) has shown its high promise, empowering LLMs to generate more qualified responses with retrieved external data and knowledge. However, most RAG methods retrieve relevant documents based on either sparse or dense retrieval methods or their combinations, which overlooks the essential, multi-dimensional, and structured semantic information present in documents. This structured information plays a critical role in finding concise yet highly relevant information for domain knowledge-intensive tasks, such as scientific question-answering (QA). In this work, we introduce a multi-dimensional (cube) structure, Hypercube, which can index and allocate documents in a pre-defined multi-dimensional space. Built on the hypercube, we further propose Hypercube-RAG, a novel RAG framework for precise and efficient retrieval. Given a query, Hypercube-RAG first decomposes it based on its entities, phrases, and topics along with pre-defined hypercube dimensions, and then retrieves relevant documents from cubes by aligning these decomposed components with corresponding dimensions. Experiments on three datasets across different domains demonstrate that our method improves response accuracy by 3.7% and retrieval accuracy by 5.3% over the strongest RAG baseline. It also boosts retrieval efficiency (speed) by one or two magnitudes faster than graph-based RAG. Notably, our Hypercube-RAG inherently offers explainability by revealing those underlying dimensions used for retrieval. The code and data are available at https://github.com/JimengShi/Hypercube-RAG.
format Preprint
id arxiv_https___arxiv_org_abs_2505_19288
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Hypercube-Based Retrieval-Augmented Generation for Scientific Question-Answering
Shi, Jimeng
Zhou, Sizhe
Jin, Bowen
Hu, Wei
Tian, Runchu
Wang, Shaowen
Narasimhan, Giri
Han, Jiawei
Machine Learning
Large language models (LLMs) often need to incorporate external knowledge to solve theme-specific problems. Retrieval-augmented generation (RAG) has shown its high promise, empowering LLMs to generate more qualified responses with retrieved external data and knowledge. However, most RAG methods retrieve relevant documents based on either sparse or dense retrieval methods or their combinations, which overlooks the essential, multi-dimensional, and structured semantic information present in documents. This structured information plays a critical role in finding concise yet highly relevant information for domain knowledge-intensive tasks, such as scientific question-answering (QA). In this work, we introduce a multi-dimensional (cube) structure, Hypercube, which can index and allocate documents in a pre-defined multi-dimensional space. Built on the hypercube, we further propose Hypercube-RAG, a novel RAG framework for precise and efficient retrieval. Given a query, Hypercube-RAG first decomposes it based on its entities, phrases, and topics along with pre-defined hypercube dimensions, and then retrieves relevant documents from cubes by aligning these decomposed components with corresponding dimensions. Experiments on three datasets across different domains demonstrate that our method improves response accuracy by 3.7% and retrieval accuracy by 5.3% over the strongest RAG baseline. It also boosts retrieval efficiency (speed) by one or two magnitudes faster than graph-based RAG. Notably, our Hypercube-RAG inherently offers explainability by revealing those underlying dimensions used for retrieval. The code and data are available at https://github.com/JimengShi/Hypercube-RAG.
title Hypercube-Based Retrieval-Augmented Generation for Scientific Question-Answering
topic Machine Learning
url https://arxiv.org/abs/2505.19288