PreprintCited by 2Topic ModelingInformation Retrieval and Search BehaviorNatural Language Processing Techniques

HIRO: Hierarchical Information Retrieval Optimization

Open access
Goel, Krish, Chandak, Mahek·2024·arXiv

Abstract

Retrieval-Augmented Generation (RAG) has revolutionized natural language processing by dynamically integrating external knowledge into Large Language Models (LLMs), addressing their limitation of static training datasets. Recent implementations of RAG leverage hierarchical data structures, which organize documents at various levels of summarization and information density. This complexity, however, can cause LLMs to "choke" on information overload, necessitating more sophisticated querying mechanisms. In this context, we introduce Hierarchical Information Retrieval Optimization (HIRO), a novel querying approach that employs a Depth-First Search (DFS)-based recursive similarity score calculation and branch pruning. This method uniquely minimizes the context delivered to the LLM without informational loss, effectively managing the challenge of excessive data. HIRO's refined approach is validated by a 10.85% improvement in performance on the NarrativeQA dataset.

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Integrity signals

Registry

DataCite · 10.48550/arxiv.2406.09979

Cites

Statistics

Claims the paper makes

The claims and statistics of this paper haven’t been checked yet.

Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.

A few free checks a day · papers already verified open instantly · results are public

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • The hierarchical organization from episodes to facts to entities to communities extends existing hierarchical RAG strategies.

    “The resulting hierarchical organization-from episodes to facts to entities to communities-extends existing hierarchical RAG strategies [10][11].”

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

The self-check needs the paper’s full text.

Cites

No references could be resolved yet.

Research works printed but not linked: 20

  1. Ekin Akyürek, Nathaniel Iluz, Jip van den Berg, and Oren Etzioni. 2022. Towards Tracing Factual Knowledge in Language Models Back to the Training Data. arXiv preprint arXiv:2205.11482 (2022). · arXiv
  2. Tom Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah, Jared D Kaplan, Prafulla Dhariwal, Arvind Neelakantan, Pranav Shyam, Girish Sastry, Amanda Askell, et al. 2020. Language models are few-shot learners. In Advances in neural information processing systems, Vol. 33. 1877-1901.
  3. Sébastien Bubeck, Andrey Chuklin, Rémi Viezzer, Audrow Graves, Demis Kar- alekas, Brian Chan, and Dougal Maclaurin. 2023. Sparks of Artificial General Intelligence: Early Experiments with GPT-4. arXiv preprint arXiv:2303.12712 (2023). · arXiv
  4. Xiaoxin He, Manzil Zaheer, Tuan Ngo, Dian Guo, Jing Han, Ryan Trivedi, Bhaskar Mitra, Mohit Iyyer, and Wen tau Yih. 2024. G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering. arXiv preprint arXiv:2402.07630 (2024). · arXiv
  5. V. R. Kanagavalli and G. Maheeja. 2016. A Study on the Usage of Data Structures in Information Retrieval. arXiv preprint arXiv:1602.07799 (2016). · arXiv
  6. Tomáš Kočiský, Jonathan Schwarz, Phil Blunsom, Chris Dyer, Karl Moritz Her- mann, Gábor Melis, and Edward Grefenstette. 2018. The narrativeqa reading comprehension challenge. Transactions of the Association for Computational Linguistics 6 (2018), 317-328.
  7. Mosh Levy, Alon Jacoby, and Yoav Goldberg. 2024. Same Task, More Tokens: the Impact of Input Length on the Reasoning Performance of Large Language Models. arXiv preprint arXiv:2402.14848 (2024). · arXiv
  8. Patrick Lewis, Ethan Perez, Aleksandara Piktus, Fabio Petroni, Vladimir Karpukhin, Naman Goyal, Heinrich Küttler, Mike Lewis, Wen tau Yih, Tim Rocktäschel, Sebastian Riedel, and Douwe Kiela. 2020. Retrieval-augmented generation for knowledge-intensive nlp tasks. In Advances in Neural Information Processing Systems, Vol. 33. 9459-9474.
  9. Nelson F. Liu, Tao Chen, Chenghao Geigle, Chenxi Song, Tie-Yan Meng, Virgina R. de Sa, and C. Lee Giles. 2024. Lost in the middle: How language models use long contexts. Transactions of the Association for Computational Linguistics 12 (2024), 157-173.
  10. Shervin Minaee, Aishwarya Mani, and Charu C. Aggarwal. 2024. Large Language Models: A Survey. arXiv preprint arXiv:2402.06196 (2024). · arXiv
  11. Richard Yuanzhe Pang, Samyam Srinivasan, Tushar Khot, Daniel Khashabi, John Mitchell, Chenhao Tan, Snigdha Galhotra, Avi Holtzman, Bhargavi Majumder, Elvis Saravia, et al. 2021. QuALITY: Question Answering with Long Input Texts, Yes! arXiv preprint arXiv:2112.08608 (2021). · arXiv
  12. Fabio Petroni, Tim Rocktäschel, Sebastian Riedel, Patrick Lewis, Anton Bakhtin, Yuxiang Wu, and Alexander Miller. 2019. Language Models as Knowledge Bases? arXiv preprint arXiv:1909.01066 (2019). · arXiv
  13. Ori Ram, Yuhuai Shiv, Tushar Khot, Felix Hill, Chris Alberti, Gabriel Ilharco, Fabio Petroni, David Saxton, and John Mitchell. 2023. In-context retrieval-augmented language models. Transactions of the Association for Computational Linguistics 11 (2023), 1316-1331.
  14. Nils Reimers and Iryna Gurevych. 2019. Sentence-BERT: Sentence Embeddings Using Siamese BERT-Networks. arXiv preprint arXiv:1908.10084 (2019). · arXiv
  15. Parth Sarthi, Xiang Ji, Hao-Yuan Huang, Chitta Baral, and Jimmy Lin. 2024. RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval. arXiv preprint arXiv:2401.18059 (2024). · arXiv
  16. Kurt Shuster, Spencer Xu, Pankaj Anand, Amrit Ghosh, Zachary Long, Yizhe Ji, Sangwoo Kim, Pratyai Singh Batra, Philip Yu, Seungwhan Lee, and Moojan Son. 2021. Retrieval Augmentation Reduces Hallucination in Conversation. arXiv preprint arXiv:2104.07567 (2021). · arXiv
  17. Jasper Snoek, Hugo Larochelle, and Ryan P. Adams. 2012. Practical Bayesian Optimization of Machine Learning Algorithms. (2012). arXiv:1206.2944 [stat.ML] https://arxiv.org/abs/1206.2944 · arXiv
  18. Simeng Sun, Zhiquan Lu, Hai Wang, Haoyang Er, Noam Avrahami, and Isil Rehbein. 2021. Do Long-Range Language Models Actually Use Long-Range Context? arXiv preprint arXiv:2109.09115 (2021). · arXiv
  19. Alon Talmor, Jonathan Herzig, Nicholas Lourie, and Jonathan Berant. 2020. oLMpics-on what language model pre-training captures. Transactions of the Association for Computational Linguistics 8 (2020), 743-758.
  20. Jiawei Wang, Hao Chen, Sung Lee, Hanzhang Zhang, Eric P. Xing, and Alexander Smola. 2024. Detect-Order-Construct: A Tree Construction Based Approach for Hierarchical Document Structure Analysis. arXiv preprint arXiv:2401.11874 (2024). · arXiv

Cited by

Sorted by citations

Authors
Goel, K., & Chandak, M.
Venue
arXiv
Publisher
arXiv
Published
2024
ISSN
2331-8422 (ISSN-L)
OpenAlex
W4399759255
Cited by
0
References
0
Type
posted content
Language
en
Open access
Locations
  • arXiv (Cornell University) (submittedVersion, cc-by)
  • arXiv (Cornell University) (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Information Systems › Information Retrieval and Search Behavior
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
Keywords
retrieval-augmented generation, depth-first search, branch pruning, context reduction, information overload, large language models

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review