OntoMedRec: Logically-Pretrained Model-Agnostic Ontology Encoders for Medication Recommendation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tan, Weicong, Wang, Weiqing, Zhou, Xin, Buntine, Wray, Bingham, Gordon, Yin, Hongzhi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929244029845504
author Tan, Weicong
Wang, Weiqing
Zhou, Xin
Buntine, Wray
Bingham, Gordon
Yin, Hongzhi
author_facet Tan, Weicong
Wang, Weiqing
Zhou, Xin
Buntine, Wray
Bingham, Gordon
Yin, Hongzhi
contents Most existing medication recommendation models learn representations for medical concepts based on electronic health records (EHRs) and make recommendations with learnt representations. However, most medications appear in the dataset for limited times, resulting in insufficient learning of their representations. Medical ontologies are the hierarchical classification systems for medical terms where similar terms are in the same class on a certain level. In this paper, we propose OntoMedRec, the logically-pretrained and model-agnostic medical Ontology Encoders for Medication Recommendation that addresses data sparsity problem with medical ontologies. We conduct comprehensive experiments on benchmark datasets to evaluate the effectiveness of OntoMedRec, and the result shows the integration of OntoMedRec improves the performance of various models in both the entire EHR datasets and the admissions with few-shot medications. We provide the GitHub repository for the source code on https://anonymous.4open.science/r/OntoMedRec-D123
format Preprint
id arxiv_https___arxiv_org_abs_2401_15814
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle OntoMedRec: Logically-Pretrained Model-Agnostic Ontology Encoders for Medication Recommendation
Tan, Weicong
Wang, Weiqing
Zhou, Xin
Buntine, Wray
Bingham, Gordon
Yin, Hongzhi
Machine Learning
Most existing medication recommendation models learn representations for medical concepts based on electronic health records (EHRs) and make recommendations with learnt representations. However, most medications appear in the dataset for limited times, resulting in insufficient learning of their representations. Medical ontologies are the hierarchical classification systems for medical terms where similar terms are in the same class on a certain level. In this paper, we propose OntoMedRec, the logically-pretrained and model-agnostic medical Ontology Encoders for Medication Recommendation that addresses data sparsity problem with medical ontologies. We conduct comprehensive experiments on benchmark datasets to evaluate the effectiveness of OntoMedRec, and the result shows the integration of OntoMedRec improves the performance of various models in both the entire EHR datasets and the admissions with few-shot medications. We provide the GitHub repository for the source code on https://anonymous.4open.science/r/OntoMedRec-D123
title OntoMedRec: Logically-Pretrained Model-Agnostic Ontology Encoders for Medication Recommendation
topic Machine Learning
url https://arxiv.org/abs/2401.15814