mlsanityData to 7 Oct 2026

KohakuRAG: A simple RAG framework with hierarchical document indexing

arXiv 2603.07612 ↗First posted 8 Mar 2026cs.CLcited by 1

On the map: Retrieval-augmented generation → RAG pipeline design and evaluation

Abstract (arXiv)

Retrieval-augmented generation (RAG) systems that answer questions from document collections face compounding difficulties when high-precision citations are required: flat chunking strategies sacrifice document structure, single-query formulations miss relevant passages through vocabulary mismatch, and single-pass inference produces stochastic answers that vary in both content and citation selection. We present KohakuRAG, a hierarchical RAG framework that preserves document structure through a four-level tree representation (document rightarrow section rightarrow paragraph rightarrow sentence) with bottom-up embedding aggregation, improves retrieval coverage through an LLM-powered query planner with cross-query reranking, and stabilizes answers through ensemble inference with abstention-aware voting. We evaluate on the WattBot 2025 Challenge, a benchmark requiring systems to answer technical questions from 32 documents with ±0.1% numeric tolerance and exact source attribution. KohakuRAG achieves first place on both public and private leaderboards (final score 0.861), as the only team to maintain the top position across both evaluation partitions. Ablation studies reveal that prompt ordering (+80% relative), retry mechanisms (+69%), and ensemble voting with blank filtering (+1.2pp) each contribute substantially, while hierarchical dense retrieval alone matches hybrid sparse-dense approaches (BM25 adds only +3.1pp). We release KohakuRAG as open-source software at https://github.com/KohakuBlueleaf/KohakuRAG.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in RAG pipeline design and evaluation, one of 15 directions in Retrieval-augmented generation. Builds on 39 of its 49 references that are in the map. First posted on arXiv 8 Mar 2026.

Builds on

39 of 49 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024A Survey on Retrieval-Augmented Text Generation for Large Language Models17 Apr
  2. 2024Retrieval-Augmented Generation for AI-Generated Content: A Survey29 Feb
  3. 2023Retrieval-Augmented Generation for Large Language Models: A Survey18 Dec

Other references · 36

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
  5. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,049
  6. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  7. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  8. 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,151
  9. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  10. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
  11. 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
  12. 2017Billion-scale similarity search with GPUscited by 2,477
24 more
  1. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  2. 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,800
  3. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,258
  4. 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
  5. 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
  6. 2023A Survey on Multimodal Large Language Modelscited by 898
  7. 2016Efficient and robust approximate nearest neighbor search using Hierarchical Navigable Small World graphscited by 853
  8. 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 846
  9. 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 705
  10. 2019Passage Re-ranking with BERTcited by 629
  11. 2023C-Pack: Packed Resources For General Chinese Embeddingscited by 556
  12. 2021ColBERTv2: Effective and Efficient Retrieval via Lightweight Late Interactioncited by 464
  13. 2024A Survey on Hallucination in Large Vision-Language Modelscited by 352
  14. 2023Improving Text Embeddings with Large Language Modelscited by 295
  15. 2021SPLADE: Sparse Lexical and Expansion Model for First Stage Rankingcited by 283
  16. 2025Agentic Retrieval-Augmented Generation: A Survey on Agentic RAGcited by 242
  17. 2023Query Rewriting for Retrieval-Augmented Large Language Modelscited by 211
  18. 2024ColPali: Efficient Document Retrieval with Vision Language Modelscited by 190
  19. 2023Query2doc: Query Expansion with Large Language Modelscited by 146
  20. 2024M3DocRAG: Multi-modal Retrieval is What You Need for Multi-page Multi-document Understandingcited by 110
  21. 2025Harnessing Multiple Large Language Models: A Survey on LLM Ensemblecited by 86
  22. 2024Late Chunking: Contextual Chunk Embeddings Using Long-Context Embedding Modelscited by 70
  23. 2023Query Expansion by Prompting Large Language Modelscited by 56
  24. 2024Sparse Meets Dense: A Hybrid Approach to Enhance Scientific Document Retrievalcited by 16

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026CHOP: Chunkwise Context-Preserving Framework for RAG on Multi Documents
  2. 2026SF-RAG: Structure-Fidelity Retrieval-Augmented Generation for Academic Question Answering
  3. 2025mmRAG: A Modular Benchmark for Retrieval-Augmented Generation over Text, Tables, and Knowledge Graphs
  4. 2025HiFi-RAG: Hierarchical Content Filtering and Two-Pass Generation for Open-Domain RAG
  5. 2024Stochastic RAG: End-to-End Retrieval-Augmented Generation through Expected Utility Maximization
  6. 2026Beyond the Reranker: Do RAG Retrieval Enhancements Help Once a Strong Reranker Is Present?
  7. 2026LURE-RAG: Lightweight Utility-driven Reranking for Efficient RAG
  8. 2025BookRAG: A Hierarchical Structure-aware Index-based Approach for Retrieval-Augmented Generation on Complex Documents