Back to Search
Start Over
Scale-Aware Graph-Based Machine Learning for Accurate Molecular Property Prediction
- Source :
- IEEE BigData
- Publication Year :
- 2020
- Publisher :
- IEEE, 2020.
-
Abstract
- With great growth in the volume of chemical databases, machine learning receives significant attention from various scientific communities for efficient high-throughput screening of molecular properties and drug discovery on the millions of chemical compounds. In particular, graph neural networks (GNNs) have been widely studied in chemistry-related fields because a molecule is natively represented as a mathematical graph. In GNNs for the graph-level analysis, a global operation called readout is applied after node embedding to generate a graph-level embedding that represents characteristics of the whole graph. However, commonly used readouts frequently distort scale information of the graph and consequently degrade the prediction accuracy of GNNs. This problem becomes more serious in molecular machine learning because molecules have many important scale information (e.g., molecular weight and total energy). In this paper, we investigate this scale distortion problem in GNNs caused by the readouts for the first time and propose an efficient solution with a new attention-based readout. In the experiments, the proposed readout outperformed commonly used readouts on various GNN architectures.
- Subjects :
- business.industry
Computer science
Scale (chemistry)
Node (networking)
Big data
010501 environmental sciences
Machine learning
computer.software_genre
01 natural sciences
Molecular machine
Distortion problem
0103 physical sciences
Embedding
Graph (abstract data type)
Artificial intelligence
010306 general physics
business
computer
Chemical database
0105 earth and related environmental sciences
Subjects
Details
- Database :
- OpenAIRE
- Journal :
- 2020 IEEE International Conference on Big Data (Big Data)
- Accession number :
- edsair.doi...........62f4804abaecba30d020c58c4b694bac