Multitask Learning On Graph Neural Networks Applied To Molecular Property Predictions
arXiv:1910.13124
Abstract
Prediction of molecular properties, including physico-chemical properties, is a challenging task in chemistry. Herein we present a new state-of-the-art multitask prediction method based on existing graph neural network models. We have used different architectures for our models and the results clearly demonstrate that multitask learning can improve model performance. Additionally, a significant reduction of variance in the models has been observed. Most importantly, datasets with a small amount of data points reach better results without the need of augmentation.
References in corpus (7)
- Semi-Supervised Classification with Graph Convolutional Networks
- An Overview of Multi-Task Learning in Deep Neural Networks
- Fast Graph Representation Learning with PyTorch Geometric
- Massively Multitask Networks for Drug Discovery
- Multi-task Neural Networks for QSAR Predictions
- Machine Learning for Scent: Learning Generalizable Perceptual Representations of Small Molecules
- Graph Attribute Aggregation Network with Progressive Margin Folding