Learning Graph-Level Representation for Drug Discovery
Predicating macroscopic influences of drugs on human body, like efficacy and toxicity, is a central problem of small-molecule based drug discovery. Molecules can be represented as an undirected graph, and we can utilize graph convolution networks to predication molecular properties. However, graph convolutional networks and other graph neural networks all focus on learning node-level representation rather than graph-level representation. Previous works simply sum all feature vectors for all nodes in the graph to obtain the graph feature vector for drug predication. In this paper, we introduce a dummy super node that is connected with all nodes in the graph by a directed edge as the representation of the graph and modify the graph operation to help the dummy super node learn graph-level feature. Thus, we can handle graph-level classification and regression in the same way as node-level classification and regression. In addition, we apply focal loss to address class imbalance in drug datasets. The experiments on MoleculeNet show that our method can effectively improve the performance of molecular properties predication.
Code (2)
Tasks
AllDrug DiscoveryGeneral ClassificationregressionMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Learn molecular representations from large-scale unlabeled molecules for drug discovery
How to produce expressive molecular representations is a fundamental challenge in AI-driven drug discovery. Graph neural network (GNN) has emerged as a powerful technique for modeling molecular data. However, previous su…
Drug DiscoveryGraph Neural NetworkAn effective self-supervised framework for learning expressive molecular global representations to drug discovery
How to produce expressive molecular representations is a fundamental challenge in artificial intelligence-driven drug discovery. Graph neural network (GNN) has emerged as a powerful technique for modeling molecular data.…
Drug DiscoveryGraph Neural NetworkHiGraphDTI: Hierarchical Graph Representation Learning for Drug-Target Interaction Prediction
The discovery of drug-target interactions (DTIs) plays a crucial role in pharmaceutical development. The deep learning model achieves more accurate results in DTI prediction due to its ability to extract robust and expre…
Graph Representation Learningmolecular representationRepresentation LearningCardiGraphormer: Unveiling the Power of Self-Supervised Learning in Revolutionizing Drug Discovery
In the expansive realm of drug discovery, with approximately 15,000 known drugs and only around 4,200 approved, the combinatorial nature of the chemical space presents a formidable challenge. While Artificial Intelligenc…
Drug DiscoverySelf-Supervised LearningA Knowledge Graph-Enhanced Tensor Factorisation Model for Discovering Drug Targets
The drug discovery and development process is a long and expensive one, costing over 1 billion USD on average per drug and taking 10-15 years. To reduce the high levels of attrition throughout the process, there has been…
BIG-bench Machine LearningDrug DiscoveryGraph Representation LearningKnowledge Graph Embeddings+1