Scaling Biomedical Knowledge Graph Retrieval for Interpretable Reasoning: Applications to Clinical Diagnosis Prediction

Biomedical knowledge graphs (KGs) organize molecular mechanisms, biological pathways, and clinical concepts into structured representations that support diagnostic reasoning. As these graphs grow in scale and connectivity, scalable and interpretable multi-hop retrieval over deep biomedical graph structures has become a major computational bottleneck. We present LogosKG, a hardware-optimized retrieval system that enables efficient k-hop traversal over very large biomedical KGs using symbolic graph formulations and hardware-efficient execution. By integrating degree-aware partitioning, cross-partition routing, and on-demand caching, LogosKG scales to billion-edge graphs while preserving retrieval fidelity. Experiments demonstrate substantial efficiency improvements over CPU- and GPU-based baselines. Using diagnosis-oriented retrieval workloads as a downstream case study, we show that scalable access to deep, high-hop biomedical graph structures enables interpretable diagnostic evidence propagation. A clinician-aligned LLM-as-judge evaluation further indicates that high-hop KG retrieval improves reasoning quality in terms of accuracy, comprehensibility, and succinctness, underscoring the value of deep graph retrieval for diagnostic reasoning.

Authors

Institutions

Publication Details

Journal
medRxiv
Published
2026-01-13
DOI
https://doi.org/10.64898/2026.01.12.26343957
Primary Topic
Advanced Graph Neural Networks
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Scaling Biomedical Knowledge Graph Retrieval for Interpretable Reasoning: Applications to Clinical Diagnosis Prediction

Dmitriy Dligach, Miller Tl, Cheng He, Yifu Wu et al.
medRxiv
Advanced Graph Neural Networks
article

Scaling Biomedical Knowledge Graph Retrieval for Interpretable Reasoning: Applications to Clinical Diagnosis Prediction

Dmitriy Dligach, Miller Tl, Cheng He, Yifu Wu, Maya Kruse, Saksham Khatwani, Yanjun Gao, Majid Afshar
article en

Abstract

Biomedical knowledge graphs (KGs) organize molecular mechanisms, biological pathways, and clinical concepts into structured representations that support diagnostic reasoning. As these graphs grow in scale and connectivity, scalable and interpretable multi-hop retrieval over deep biomedical graph structures has become a major computational bottleneck. We present LogosKG, a hardware-optimized retrieval system that enables efficient k-hop traversal over very large biomedical KGs using symbolic graph formulations and hardware-efficient execution. By integrating degree-aware partitioning, cross-partition routing, and on-demand caching, LogosKG scales to billion-edge graphs while preserving retrieval fidelity. Experiments demonstrate substantial efficiency improvements over CPU- and GPU-based baselines. Using diagnosis-oriented retrieval workloads as a downstream case study, we show that scalable access to deep, high-hop biomedical graph structures enables interpretable diagnostic evidence propagation. A clinician-aligned LLM-as-judge evaluation further indicates that high-hop KG retrieval improves reasoning quality in terms of accuracy, comprehensibility, and succinctness, underscoring the value of deep graph retrieval for diagnostic reasoning.

medRxiv
Boston Children's Hospital (US), University of Wisconsin–Madison (US), Harvard University (US), University of Colorado Boulder (US), Loyola University Chicago (US), University of Colorado Anschutz Medical Campus (US)
U.S. National Library of Medicine
Openalex Percentile: Top 89%
Advanced Graph Neural Networks
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Scaling Biomedical Knowledge Graph Retrieval for Interpretable Reasoning: Applications to Clinical Diagnosis Prediction — Dmitriy Dligach, Miller Tl, et al. · medRxiv (2026) | TGRS Research Map | TGRS