Bi-Level Graph Neural Networks for Drug-Drug Interaction Prediction

Yunsheng Bai,Ken Gu,Yizhou Sun,Wei Wang
DOI: https://doi.org/10.48550/arXiv.2006.14002
2020-06-11
Abstract:We introduce Bi-GNN for modeling biological link prediction tasks such as drug-drug interaction (DDI) and protein-protein interaction (PPI). Taking drug-drug interaction as an example, existing methods using machine learning either only utilize the link structure between drugs without using the graph representation of each drug molecule, or only leverage the individual drug compound structures without using graph structure for the higher-level DDI graph. The key idea of our method is to fundamentally view the data as a bi-level graph, where the highest level graph represents the interaction between biological entities (interaction graph), and each biological entity itself is further expanded to its intrinsic graph representation (representation graphs), where the graph is either flat like a drug compound or hierarchical like a protein with amino acid level graph, secondary structure, tertiary structure, etc. Our model not only allows the usage of information from both the high-level interaction graph and the low-level representation graphs, but also offers a baseline for future research opportunities to address the bi-level nature of the data.
Computational Engineering, Finance, and Science,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to effectively use the graph representation of drug molecules and the graph structure of interactions between drugs to improve the prediction accuracy in the drug - drug interaction (DDI) prediction task. Existing methods either only use the link structure between drugs without using the graph representation of each drug molecule, or only utilize the structure of a single drug compound without considering the higher - level DDI graph structure. BI - GNN (Bi - Level Graph Neural Networks) aims to overcome these limitations by constructing a two - level graph neural network framework, which can simultaneously utilize the intrinsic graph representation of drug molecules and the high - level graph structure of interactions between drugs at different levels, thereby more comprehensively capturing the complex interaction relationships between drugs. Specifically, BI - GNN regards data as a two - level graph, where the highest level represents the interactions between biological entities (interaction graph), and each biological entity itself is further expanded into its intrinsic graph representation (representation graph). For example, in the case of drug - drug interactions, each drug can be regarded as a node, its molecular structure as the low - level graph representation, and the interactions between drugs form the high - level interaction graph. In this way, BI - GNN can not only utilize the information of both high and low levels, but also provides a basis for future research to deal with the two - level nature of biological interaction networks.