Beyond Traditional Diagnostics: Transforming Patient-Side Information into Predictive Insights with Knowledge Graphs and Prototypes
About
Predicting diseases solely from patient-side information, such as demographics and self-reported symptoms, has attracted significant research attention due to its potential to enhance patient awareness, facilitate early healthcare engagement, and improve healthcare system efficiency. However, existing approaches encounter critical challenges, including imbalanced disease distributions and a lack of interpretability, resulting in biased or unreliable predictions. To address these issues, we propose the Knowledge graph-enhanced, Prototype-aware, and Interpretable (KPI) framework. KPI systematically integrates structured and trusted medical knowledge into a unified disease knowledge graph, constructs clinically meaningful disease prototypes, and employs contrastive learning to enhance predictive accuracy, which is particularly important for long-tailed diseases. Additionally, KPI utilizes large language models (LLMs) to generate patient-specific, medically relevant explanations, thereby improving interpretability and reliability. Extensive experiments on real-world datasets demonstrate that KPI outperforms state-of-the-art methods in predictive accuracy and provides clinically valid explanations that closely align with patient narratives, highlighting its practical value for patient-centered healthcare delivery.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Disease prediction | Haodf Lung | Hit Rate @ 192.38 | 16 | |
| Disease prediction | Haodf Pneumonia | Hit@192.69 | 16 | |
| Disease prediction | Haodf Depression | Hit Rate @ 10.8119 | 16 | |
| Disease prediction | Haodf Coronary Heart Disease | Hit Rate @ 167.55 | 16 | |
| Disease prediction | Haodf Diabetes | Hit Rate @ 179.69 | 16 | |
| Disease prediction | Haodf Common Cold | Hit@188.08 | 16 | |
| Medical Disease Diagnosis | Haodf Lung (test) | Hit Rate @ 192.38 | 8 | |
| Medical Disease Diagnosis | Haodf Pneu (test) | Hit@10.9269 | 8 | |
| Medical Disease Diagnosis | Haodf (Depr.) (test) | Hit@10.8119 | 8 | |
| Medical Disease Diagnosis | Haodf CHD (test) | Hit Rate@167.55 | 8 |