Relation Matters in Sampling: A Scalable Multi-Relational Graph Neural Network for Drug-Drug Interaction Prediction
arXiv:2105.13975
Abstract
Sampling is an established technique to scale graph neural networks to large graphs. Current approaches however assume the graphs to be homogeneous in terms of relations and ignore relation types, critically important in biomedical graphs. Multi-relational graphs contain various types of relations that usually come with variable frequency and have different importance for the problem at hand. We propose an approach to modeling the importance of relation types for neighborhood sampling in graph neural networks and show that we can learn the right balance: relation-type probabilities that reflect both frequency and importance. Our experiments on drug-drug interaction prediction show that state-of-the-art graph neural networks profit from relation-dependent sampling in terms of both accuracy and efficiency.
References in corpus (9)
- Inductive Representation Learning on Large Graphs
- Link Prediction in Complex Networks: A Survey
- Modeling polypharmacy side effects with graph convolutional networks
- Fast Graph Representation Learning with PyTorch Geometric
- Adaptive Sampling Towards Fast Graph Representation Learning
- Learning Discrete Structures for Graph Neural Networks
- MR-GNN: Multi-Resolution and Dual Graph Neural Network for Predicting Structured Entity Interactions
- Drug Similarity Integration Through Attentive Multi-view Graph Auto-Encoders
- GENN: Predicting Correlated Drug-drug Interactions with Graph Energy Neural Networks