| Title: | KEEP: Integrating Medical Ontologies with Clinical Data for Robust Code Embeddings. |
| Journal: | Proceedings of machine learning research |
| Published: | 1 Jun 2025 |
| Pubmed: | https://pubmed.ncbi.nlm.nih.gov/42438584/ |
Publication 20666
WARNING: the interactive features of this website use CSS3, which your browser does not support. To use the full features of this website, please update your browser.
Abstract
Machine learning in healthcare requires effective representation of structured medical codes, but current methods face a trade-off: knowledge graph-based approaches capture formal relationships but miss real-world patterns, while data-driven methods learn empirical associations but often overlook structured knowledge in medical terminologies. We present KEEP (Knowledge-preserving and Empirically refined Embedding Process), an efficient framework that bridges this gap by combining knowledge graph embeddings with adaptive learning from clinical data. KEEP first generates embeddings from knowledge graphs, then employs regularized training on patient records to adaptively integrate empirical patterns while preserving ontological relationships. Importantly, KEEP produces final embeddings without task-specific axillary or end-to-end training enabling KEEP to support multiple downstream applications and model architectures. Evaluations on structured EHR from UK Biobank and MIMIC-IV demonstrate that KEEP outperforms both traditional and Language Model-based approaches in capturing semantic relationships and predicting clinical outcomes. Moreover, KEEP's minimal computational requirements make it particularly suitable for resource-constrained environments.</p>
6 Authors
- Ahmed Elhussein
- Paul Meddeb
- Abigail Newbury
- Jeanne Mirone
- Martin Stoll
- Gamze Gürsoy
Enabling scientific discoveries that improve human health