Abstract
Drug repositioning utilizes existing drugs for new therapeutic applications, driven by the rapid increase in disease and drug-related data. However, organizing knowledge in this field and integrating the complex and scattered data from multiple systems into a cohesive knowledge network have become urgent problems to address. In this paper, we propose a drug repositioning model based on knowledge graph embedding. The model employs multivariate relational data to embed entities and relationships in a low-dimensional vector space. It also innovatively introduces the attention mechanism into translation and bilinear models, forming new models such as Attranse, Attdismult, and Attrescal. This model's feature extraction does not rely on a single approach, instead, it integrates multiple models and combines their screening results to enhance drug screening quality. The model's effectiveness was validated using COVID-19 data, yielding results consistent with 7 clinically approved drugs for COVID-19 treatment, indicating high accuracy in identifying new drug indications. The successful application of this model to COVID-19 suggests its potential for broader use in emerging infectious diseases and complex conditions, providing valuable insights for future drug development.