MedCPI: A Construct–Personalize–Integrate Framework for KG-enhanced Clinical Prediction

Hang Wang, Hang Dong, Lu Liu


Abstract
Electronic health records (EHRs) provide longitudinal evidence for clinical prediction, but EHR data are sparse, incomplete, and heterogeneous, which can limit robustness. Medical knowledge graphs (MKGs) have therefore been incorporated to support KG-enhanced clinical prediction by linking heterogeneous EHR codes to shared medical concepts via structured relations. However, existing KG-enhanced approaches remain limited in two aspects: (i) task-specific knowledge selection when extracting knowledge from a large multi-source MKG; and (ii) patient-level personalization and knowledge integration, where personalization is often weakly controlled and knowledge integration is not sufficiently aligned with longitudinal patient trajectories. To address these issues, we propose MedCPI, a unified Construct–Personalize–Integrate framework. MedCPI first performs task-guided schema induction and KG normalization to build a task-specific Concept MKG as a denoised knowledge pool, then constructs controlled patient-level PKGs via local expansion and short path search, and finally integrates PKG representations with time-aware EHR representations via cross-attention for prediction. Experiments on MIMIC-III and MIMIC-IV across four clinical prediction tasks show consistent improvements over strong EHR-only and KG-enhanced baselines. Ablations and additional analyses further validate the contribution of each stage and illustrate how MedCPI utilizes structured medical knowledge.
Anthology ID:
2026.findings-acl.1215
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
24266–24282
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1215/
DOI:
Bibkey:
Cite (ACL):
Hang Wang, Hang Dong, and Lu Liu. 2026. MedCPI: A Construct–Personalize–Integrate Framework for KG-enhanced Clinical Prediction. In Findings of the Association for Computational Linguistics: ACL 2026, pages 24266–24282, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
MedCPI: A Construct–Personalize–Integrate Framework for KG-enhanced Clinical Prediction (Wang et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1215.pdf
Checklist:
 2026.findings-acl.1215.checklist.pdf