A medical RAG system built on ontologies, papers, and patience
It forces medical LLMs to retrieve evidence across patient records, papers, and clinical ontologies.

What it does
The system grounds clinical LLM outputs in a three-tier knowledge graph, linking private patient data (via MIMIC) to medium-level medical literature and bottom-level ontologies like UMLS inside Neo4j. Answers trace back to actual evidence instead of model weights.
The interesting bit
The “trinity” hierarchy is the core mechanic: patient records, medical papers, and dictionary concepts are explicitly connected, forcing retrieval to hop across graph layers before generating a response. The authors also provide a Docker demo that searches PubMed live to dodge the licensing wall around local textbook storage.
Key highlights
- Hierarchical evidence retrieval across patient records, medical literature, and UMLS ontologies via Neo4j.
- Built on the CAMEL multi-agent framework for pipeline construction.
- Ships with a Docker demo using live PubMed search to work around medical data licensing restrictions.
- Accepted at ACL 2025; paper and arXiv preprint available.
Caveats
- Reproducing the full stack requires buying copyrighted medical textbooks, applying for UMLS access, and licensing MIMIC IV data.
- The simplified example dataset only covers the top patient-record layer; medium and bottom-level alternatives are still a work in progress.
- The documented path requires OpenAI and Neo4j credentials, with no guidance on swapping either backend shown.
Verdict
A solid reference for medical AI researchers and clinicians who can clear the data-access hurdles. If you are looking for a drop-in RAG toy, the licensing and infrastructure overhead will send you elsewhere.
Frequently asked
- What is ImprintLab/Medical-Graph-RAG?
- It forces medical LLMs to retrieve evidence across patient records, papers, and clinical ontologies.
- Is Medical-Graph-RAG open source?
- Yes — ImprintLab/Medical-Graph-RAG is open source, released under the MIT license.
- What language is Medical-Graph-RAG written in?
- ImprintLab/Medical-Graph-RAG is primarily written in Python.
- How popular is Medical-Graph-RAG?
- ImprintLab/Medical-Graph-RAG has 814 stars on GitHub.
- Where can I find Medical-Graph-RAG?
- ImprintLab/Medical-Graph-RAG is on GitHub at https://github.com/ImprintLab/Medical-Graph-RAG.