Interaction-aware Drug Package Recommendation via Policy Gradient
Zhi Zheng,Chao Wang,Tong Xu,Dazhong Shen,Penggang Qin,Xiangyu Zhao,Baoxing Huai,Xian Wu,Enhong Chen
DOI: https://doi.org/10.1145/3511020
IF: 4.657
2022-02-14
ACM Transactions on Information Systems
Abstract:Recent years have witnessed the rapid accumulation of massive electronic medical records (EMRs), which highly support intelligent medical services such as drug recommendation. However, although there are multiple interaction types between drugs, e.g., synergism and antagonism, which can influence the effect of a drug package significantly, prior arts generally neglect the interaction between drugs or consider only a single type of interaction. Moreover, most existing studies generally formulate the problem of package recommendation as getting a personalized scoring function for users, despite the limits of discriminative models to achieve satisfactory performance in practical applications. To this end, in this paper, we propose a novel end-to-end D rug P ackage G eneration (DPG) framework, which develops a new generative model for drug package recommendation with considering the interaction effect within drugs that affected by patient conditions. Specifically, we propose to formulate the drug package generation as a sequence generation process. Along this line, we first initialize the drug interaction graph based on medical records and domain knowledge. Then, we design a novel message passing neural network to capture the drug interaction, as well as a drug package generator based on a recurrent neural network. In detail, a mask layer is utilized to capture the impact of patient condition, and the deep reinforcement learning technique is leveraged to reduce the dependence on the drug order. Finally, extensive experiments on a real-world dataset from a first-rate hospital demonstrate the effectiveness of our DPG framework compared with several competitive baseline methods.
computer science, information systems