Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literature

Pouria Mortezaagha, Arya Rahgozar
3/23/2026
cs.AIcs.IR

No official code found. Best implementation: pouriamrt/gralc-rag (1 star, MIT, last commit 2026-03-23).

Abstract

Retrieval-Augmented Generation (RAG) systems for biomedical literature are typically evaluated using ranking metrics like Mean Reciprocal Rank (MRR), which measure how well the system identifies the single most relevant chunk. We argue that for full-text scientific documents, this paradigm is incomplete: it rewards retrieval precision while ignoring retrieval breadth -- the ability to surface evidence from across a document's structural sections. We propose GraLC-RAG, a framework that unifies late chunking with graph-aware structural intelligence, introducing structure-aware chunk boundary detection, UMLS knowledge graph infusion, and graph-guided hybrid retrieval. We evaluate six strategies on 2,359 IMRaD-filtered PubMed Central articles using 2,033 cross-section questions and two metric families: standard ranking metrics (MRR, Recall@k) and structural coverage metrics (SecCov@k, CS Recall). Our results expose a sharp divergence: content-similarity methods achieve the highest MRR (0.517) but always retrieve from a single section, while structure-aware methods retrieve from up to 15.6x more sections. Generation experiments show that KG-infused retrieval narrows the answer-quality gap to delta-F1 = 0.009 while maintaining 4.6x section diversity. These findings demonstrate that standard metrics systematically undervalue structural retrieval and that closing the multi-section synthesis gap is a key open problem for biomedical RAG.

View on arXivView PDF

Code Implementations(1)

MIT

GraLC-RAG: Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literature. Integrates document structure graphs, UMLS knowledge graph infusion, and graph-guided retrieval into the late chunking paradigm.

10Shell, PythonMar 22, 20267 months agoMIT

Similar Papers

Cite this paper

@article{mortezaagha2026graphaware,
  title  = {Graph-Aware Late Chunking for Retrieval-Augmented Generation in Biomedical Literature},
  author = {Pouria Mortezaagha and Arya Rahgozar},
  year   = {2026},
  eprint = {2603.22633},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.22633v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.22633.svg)](https://codeofpaper.com/paper/2603.22633)

Discussion