LLM-DDI: Leveraging Large Language Models for Drug-Drug Interaction Prediction on Biomedical Knowledge Graph

Abstract

Drug-drug interaction (DDI) refers to the interaction relationships between drugs. Discovering new DDIs is crucial for advancing drug development and enhancing clinical treatments. Given the significant progress achieved through graph neural networks (GNNs), network-based models have become a prevalent approach for tackling this challenge. However, current network-based approaches are incapable of seamlessly integrating a wide range of information. Motivated by this discovery, we propose a novel model, namely LLM-DDI, which aims to comprehensively tackle DDI prediction tasks by integrating various information of molecules in the BKG. LLM-DDI initially incorporates the generative pre-trained transformer (GPT) model to generate embeddings for each molecule within the biomedical knowledge graph (BKG). These embeddings encompass diverse types of information pertaining to each molecule. Subsequently, LLM-DDI utilizes a message-passing GNN framework to enhance the learning of molecular representations with the embeddings derived from GPT as input. LLM-DDI governs the propagation of information within the BKG by semantic relationships. These semantic relationships determine how information flows and is exchanged between different entities in the BKG. Finally, LLM-DDI leverages the learned drug representations to predict potential DDIs. Experiments show the effectiveness of LLM-DDI, as it achieves the best performance on two real-world datasets, providing valuable guidance for drug development and clinical treatment.