HiRef: Leveraging Hierarchical Ontology and Network Refinement for Robust Medication Recommendation
2508.10425v1
cs.AI, cs.LG
2025-08-16
Авторы:
Yan Ting Chok, Soyon Park, Seungheun Baek, Hajung Kim, Junhyun Lee, Jaewoo Kang
Резюме на русском
## Контекст
Medication recommendation является ключевым заданием для поддержки врачей в принятии оперативных решений на основе длительных пациентских медицинских записей. Однако реальные данные из ЭМС часто представляют собой разнообразные и неполные записи, трудно поддающиеся полному пониманию. Несмотря на то, что модели, обученные на ЭМС, определяются высоким количеством пациентов, всё же сталкиваются с проблемами, например, с редкостью встречаемых медицинских кодов или с непредсказуемыми условиями. Это приводит к снижению точности рекомендаций в нестандартных ситуациях. Основной мотивацией для разработки HiRef является необходимость развития методов, которые способны выполнять рекомендации в различных условиях, включая нестандартные ситуации.
## Метод
HiRef предлагается в качестве алгоритма, объединяющего два элемента: (i) классификация с использованием медицинской иерархии и (ii) усовершенствованные методы графа сущностей EHR. Он использует структуру верхнеуровневая иерархия в лечении, разработанную на основе медицинских наук, для более широкого усвоения знаний. Также гиперболическая семантическая структура применяется для поддержки **knowledge transfer** в иерархии. Обработка графа EHR-данных включает в себя расширенные регуляризационные схемы, чтобы уменьшить нагрузку от спутанных связей и поддерживать только клинически значимые. Этот подход позволяет HiRef достигать устойчивости в ситуациях с неизвестными кодами и смещенными данными.
## Результаты
HiRef проверен на двух ключевых датасетах: MIMIC-III и MIMIC-IV. Модель показывает значительное улучшение в сравнении с другими моделями в ситуациях с неизвестными кодами. Результаты включают высокую точность и устойчивость в различных клинических условиях. Также проводились анализы, подтверждающие эффективность встроенных механизмов очистки графов и глубокой иерархической семантики.
## Значимость
HiRef может быть применен в различных клинических задачах, таких как рекомендация лекарств и мониторинг медикаментозной терапии. Выделяются преимущества, такие как увеличение точности рекомендаций в условиях нестандартных данных и устойчивость к плохо представленным и новым кодам. Данный подход также может улучшить клинические решения и увеличить уровень доверия в рекомендациях, поддерживая большую роль в области медицины.
## Выводы
Данная работа показывает эффективность HiRef в решении проблем, связанных с рекомендацией лекарств в условиях неполных данных и неизвестных кодов. Несмотря на высокую точность, модель столкнулась с ограничениями в ситуациях с очень редкими кодами. Буду
Abstract
Medication recommendation is a crucial task for assisting physicians in
making timely decisions from longitudinal patient medical records. However,
real-world EHR data present significant challenges due to the presence of
rarely observed medical entities and incomplete records that may not fully
capture the clinical ground truth. While data-driven models trained on
longitudinal Electronic Health Records often achieve strong empirical
performance, they struggle to generalize under missing or novel conditions,
largely due to their reliance on observed co-occurrence patterns. To address
these issues, we propose Hierarchical Ontology and Network Refinement for
Robust Medication Recommendation (HiRef), a unified framework that combines two
complementary structures: (i) the hierarchical semantics encoded in curated
medical ontologies, and (ii) refined co-occurrence patterns derived from
real-world EHRs. We embed ontology entities in hyperbolic space, which
naturally captures tree-like relationships and enables knowledge transfer
through shared ancestors, thereby improving generalizability to unseen codes.
To further improve robustness, we introduce a prior-guided sparse
regularization scheme that refines the EHR co-occurrence graph by suppressing
spurious edges while preserving clinically meaningful associations. Our model
achieves strong performance on EHR benchmarks (MIMIC-III and MIMIC-IV) and
maintains high accuracy under simulated unseen-code settings. Extensive
experiments with comprehensive ablation studies demonstrate HiRef's resilience
to unseen medical codes, supported by in-depth analyses of the learned
sparsified graph structure and medical code embeddings.
Ссылки и действия
Дополнительные ресурсы: