Knowledge-Enhanced Dual Graph Neural Network for Robust Medicine Recommendation

Medicine recommendation assists physicians in automatically providing medicine combinations, which is critical in health care. Existing efforts focus on making medicine recommendations based on the patient’s electronic health record(EHR). However, they ignore external medicine knowledge and are vulnerable to the missing EHR. In this paper, a knowledge-enhanced dual graph neural network (KDGN) is proposed to recommend medicine sets. KDGN combines diagnosis-level and procedure-level attention mechanisms to encode multiple types of medical codes. In order to mine medicine from medical knowledge, KDGN further designs a dual-graph neural network, which constructs a medicine co-occurrence graph and molecular connection graph, and retrieves potential therapeutic drugs. Furthermore, during the training phase, we introduce the automatic correction loss based on maximum likelihood estimation to mitigate the impact of missing EHR and enhance the robustness of KDGN. We evaluate the proposed model on the public MIMIC-III dataset, and experimental results show that KDGN outperforms the state-of-the-art model in 4 out of 5 evaluation metrics. Our dataset and code are available at: https://github.com/Benjamin-cell/KDGN.