Trajectory-Aware Clinical Risk Prediction via Severity-Grounded Knowledge Graphs and Retrieval-Augmented Generation
Kyunghoon Jeon, Youmin Ko, Woohwan Jung, Hyunjoon Kim
Abstract
While Electronic Health Records (EHRs) offer a wealth of clinical data, effectively augmenting a patient's records with heterogeneous external knowledge to predict the patient's clinical risk remains a significant challenge. Existing methods fail to capture disease severity, treatment responses, and nuanced clinical progression, due to data sparsity and the underutilization of unstructured clinical notes. To address these challenges, we propose TRACER (a trajectory-aware and clinically grounded prediction framework) that (1) constructs a medical knowledge graph enriched with severity information from medical literature, (2) retrieves clinically relevant, severity-weighted paths of a patient's progression from the knowledge graph, (3) extracts clinically relevant events from unstructured clinical notes, and (4) augments patient context with similar peer cases. Experiments on the MIMIC-III and MIMIC-IV datasets demonstrate large gains over state-of-the-art baselines, with up to (28.5%) increase in Macro F1 score for the mortality prediction task, and (19.7%) increase for the readmission prediction task. Beyond performance, TRACER yields transparent, path?level explanations, enabling clinicians to inspect the literature-backed logic and specific textual evidence driving each risk estimate.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext be6c09f5-7c3d-453d-9f9d-a4fee19a28b6Builds on16
- SimCSE: Simple Contrastive Learning of Sentence EmbeddingsTianyu Gao, Xingcheng Yao, Danqi ChenEMNLP 2021 · 2,496 citations
- RECOMP: Improving Retrieval-Augmented LMs with Context Compression and Selective AugmentationFangyuan Xu, Weijia Shi, Eunsol ChoiICLR 2024 · 260 citations
- ConCare: Personalized Clinical Feature Embedding via Capturing the Healthcare ContextLiantao Ma, Chaohe Zhang, Yasha Wang, Wenjie Ruan et al.AAAI 2020 · 190 citations
- HiTANet: Hierarchical Time-Aware Attention Networks for Risk Prediction on Electronic Health RecordsJunyu Luo, Muchao Ye, Cao Xiao, Fenglong MaKDD 2020 · 187 citations
- AdaCare: Explainable Clinical Health Status Representation Learning via Scale-Adaptive Feature Extraction and RecalibrationLiantao Ma, Junyi Gao, Yasha Wang, Chaohe Zhang et al.AAAI 2020 · 156 citations
Related papers
- GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge GraphsPengcheng Jiang, Cao Xiao, Adam Cross, Jimeng SunICLR 2024 · 77 citations
- Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community RetrievalPengcheng Jiang, Cao Xiao, Minhao Jiang, Parminder Bhatia et al.ICLR 2025
- PathwayLLM: Explainable Clinical Trajectory Modeling with Structured Pathways for Sepsis PredictionZhengqiu Yu, Yueping Ding, Xiangrong LiuICML 2026
- Learning What to Ignore: Mitigating Negative Transfer in Medical Knowledge Fusion via Clinical Task-Adaptive SelectionXinyan Deng, Shoubin Dong, Xiaorou ZhengACL 2026
- MedPath: Augmenting Health Risk Prediction via Medical Knowledge PathsMuchao Ye, Suhan Cui, Yaqing Wang, Junyu Luo et al.WWW 2021 · 77 citations
