KSHFS: Research on Drug-Drug Interaction Prediction Based on Knowledge Subgraph and High-Order Feature-Aware Structure
摘要
Effective drug-drug interaction (DDI) prediction can prevent adverse reactions and side effects caused by taking multiple drugs at the same time. However, most methods that obtain drug information through large-scale biomedical knowledge graphs (KGs), ignore the problem of high noise and complexity, and have certain limitations in obtaining rich neighborhood information for each entity in the KG. Therefore, this paper proposes an end-to-end method called Knowledge Subgraph and High-order Feature-aware Structure (KSHFS) to address DDI prediction. In KSHFS, this paper first designs a subgraph extraction module to reduce the noise caused by the KG, remove irrelevant information, and effectively utilize the entity information in external knowledge graphs to assist DDI prediction. Then, a high-order feature-aware module is designed to aggregate entity information propagated from high-order neighbors, learn high-order structural embeddings for each entity, and effectively capture potential semantic neighborhood features of drug pairs. Finally, in binary DDI prediction, a self-attention mechanism is used for feature fusion to predict drug interaction events. The experimental results demonstrate that the KSHFS model outperforms the baseline models in binary and multi-relation DDI prediction based on various evaluation metrics, including AUC, AUPR, and F1.