The invention relates to a double-attention-enhanced
drug prediction model MTDADP under a multi-
task learning framework, and belongs to the technical field of medical
artificial intelligence. According to the method,
disease diagnosis detail information can be fully utilized, association between the
disease diagnosis detail information and
medical history can be deeply mined, and
safer and more accurate personalized
drug prescription recommendation can be realized. The method comprises the following steps: firstly, performing embedded representation on historical treatment records, current symptoms and test details of a patient by the MTDADP; the
state representation of the patient is enhanced through double mechanisms of access level attention and
feature dimension attention; then constructing a patient
memory bank and carrying out two-channel retrieval, and comprehensively utilizing
drug distribution information of similar and dissimilar patients; then, the safety of drug representation is enhanced through a graph convolutional network by using an EHR graph and a DDI graph; and finally, integrating multi-source information to carry out joint prediction, introducing
disease diagnosis prediction as an auxiliary task, and carrying out optimization through a multi-objective
loss function. Experimental results on a disclosed PIC
data set show that the MTDADP is remarkably improved in Jaccard, F1 and PR-AUC indexes compared with an existing advanced
baseline model, and the effectiveness of the method in the aspect of improving the drug prediction accuracy is verified.