KEEP: Integrating Medical Ontologies with Clinical Data for Robust Code Embeddings

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Elhussein, Ahmed, Meddeb, Paul, Newbury, Abigail, Mirone, Jeanne, Stoll, Martin, Gursoy, Gamze
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911194129891328
author Elhussein, Ahmed
Meddeb, Paul
Newbury, Abigail
Mirone, Jeanne
Stoll, Martin
Gursoy, Gamze
author_facet Elhussein, Ahmed
Meddeb, Paul
Newbury, Abigail
Mirone, Jeanne
Stoll, Martin
Gursoy, Gamze
contents 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 auxiliary 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.
format Preprint
id arxiv_https___arxiv_org_abs_2510_05049
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle KEEP: Integrating Medical Ontologies with Clinical Data for Robust Code Embeddings
Elhussein, Ahmed
Meddeb, Paul
Newbury, Abigail
Mirone, Jeanne
Stoll, Martin
Gursoy, Gamze
Machine Learning
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 auxiliary 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.
title KEEP: Integrating Medical Ontologies with Clinical Data for Robust Code Embeddings
topic Machine Learning
url https://arxiv.org/abs/2510.05049