DKINet: Medication Recommendation via Domain Knowledge Informed Deep Learning

Sicen Liu,Xiaolong Wang,Xianbing Zhao,Hao Chen
2024-05-08
Abstract:Medication recommendation is a fundamental yet crucial branch of healthcare that presents opportunities to assist physicians in making more accurate medication prescriptions for patients with complex health conditions. Previous studies have primarily focused on learning patient representation from electronic health records (EHR). While considering the clinical manifestations of the patient is important, incorporating domain-specific prior knowledge is equally significant in diagnosing the patient's health conditions. However, effectively integrating domain knowledge with the patient's clinical manifestations can be challenging, particularly when dealing with complex clinical manifestations. Therefore, in this paper, we first identify comprehensive domain-specific prior knowledge, namely the Unified Medical Language System (UMLS), which is a comprehensive repository of biomedical vocabularies and standards, for knowledge extraction. Subsequently, we propose a knowledge injection module that addresses the effective integration of domain knowledge with complex clinical manifestations, enabling an effective characterization of the health conditions of the patient. Furthermore, considering the significant impact of a patient's medication history on their current medication, we introduce a historical medication-aware patient representation module to capture the longitudinal influence of historical medication information on the representation of current patients. Extensive experiments on three publicly benchmark datasets verify the superiority of our proposed method, which outperformed other methods by a significant margin. The code is available at: <a class="link-external link-https" href="https://github.com/sherry6247/DKINet" rel="external noopener nofollow">this https URL</a>.
Artificial Intelligence,Information Retrieval,Machine Learning
What problem does this paper attempt to address?
The paper aims to address the issue of drug recommendation in medical recommendation systems, particularly how to assist doctors in prescribing medications more accurately for patients when faced with complex disease conditions. To achieve this goal, the study proposes a new framework called DKINet (Domain Knowledge Informed Network). The paper points out that previous research mainly focused on learning patient representation patterns from electronic health records (EHR). While considering patients' clinical manifestations is important, it is equally crucial to incorporate domain-specific prior knowledge into the diagnostic process. However, in practical applications, effectively integrating these domain knowledge with patients' clinical manifestations remains a challenge, especially when dealing with complex clinical symptoms. Therefore, this paper proposes a method to identify comprehensive domain-specific prior knowledge and combine it with patients' clinical manifestations to improve the effectiveness of drug recommendations. Specifically, the main contributions of DKINet include: 1. Proposing a new framework to combine domain knowledge with observable EHR information for drug recommendation. To the best of the authors' knowledge, this is the first attempt to use the Unified Medical Language System (UMLS) for drug recommendation. 2. Designing a graph aggregation module to extract domain knowledge; designing a knowledge-injected patient representation module to integrate domain knowledge based on clinical manifestations; and introducing a historical drug-aware patient representation module to collect historical drug information. 3. Conducting extensive experimental validation on three public benchmark datasets, with results showing that DKINet significantly outperforms other existing methods. Through these methods, DKINet can better understand patients' health conditions, thereby improving the accuracy of drug recommendations.