English

Few-Shot Graph Learning for Molecular Property Prediction

Machine Learning 2021-02-17 v1 Social and Information Networks

Abstract

The recent success of graph neural networks has significantly boosted molecular property prediction, advancing activities such as drug discovery. The existing deep neural network methods usually require large training dataset for each property, impairing their performances in cases (especially for new molecular properties) with a limited amount of experimental data, which are common in real situations. To this end, we propose Meta-MGNN, a novel model for few-shot molecular property prediction. Meta-MGNN applies molecular graph neural network to learn molecular representation and builds a meta-learning framework for model optimization. To exploit unlabeled molecular information and address task heterogeneity of different molecular properties, Meta-MGNN further incorporates molecular structure, attribute based self-supervised modules and self-attentive task weights into the former framework, strengthening the whole learning model. Extensive experiments on two public multi-property datasets demonstrate that Meta-MGNN outperforms a variety of state-of-the-art methods.

Keywords

Cite

@article{arxiv.2102.07916,
  title  = {Few-Shot Graph Learning for Molecular Property Prediction},
  author = {Zhichun Guo and Chuxu Zhang and Wenhao Yu and John Herr and Olaf Wiest and Meng Jiang and Nitesh V. Chawla},
  journal= {arXiv preprint arXiv:2102.07916},
  year   = {2021}
}

Comments

To appear in WWW 2021 (long paper); Code is available at https://github.com/zhichunguo/Meta-MGNN

R2 v1 2026-06-23T23:11:43.704Z