AweGNN: Auto-parametrized weighted element-specific graph neural networks for molecules

Timothy Szocinski, Duc Duy Nguyen, Guo Wei Wei

Research output: Contribution to journalArticlepeer-review

5 Scopus citations

Abstract

While automated feature extraction has had tremendous success in many deep learning algorithms for image analysis and natural language processing, it does not work well for data involving complex internal structures, such as molecules. Data representations via advanced mathematics, including algebraic topology, differential geometry, and graph theory, have demonstrated superiority in a variety of biomolecular applications, however, their performance is often dependent on manual parametrization. This work introduces the auto-parametrized weighted element-specific graph neural network, dubbed AweGNN, to overcome the obstacle of this tedious parametrization process while also being a suitable technique for automated feature extraction on these internally complex biomolecular data sets. The AweGNN is a neural network model based on geometric-graph features of element-pair interactions, with its graph parameters being updated throughout the training, which results in what we call a network-enabled automatic representation (NEAR). To enhance the predictions with small data sets, we construct multi-task (MT) AweGNN models in addition to single-task (ST) AweGNN models. The proposed methods are applied to various benchmark data sets, including four data sets for quantitative toxicity analysis and another data set for solvation prediction. Extensive numerical tests show that AweGNN models can achieve state-of-the-art performance in molecular property predictions.

Original languageEnglish
Article number104460
JournalComputers in Biology and Medicine
Volume134
DOIs
StatePublished - Jul 2021

Bibliographical note

Publisher Copyright:
© 2021 Elsevier Ltd

Keywords

  • Automated feature extraction
  • Deep neural network
  • Mathematical representation
  • Solvation
  • Toxicity

ASJC Scopus subject areas

  • Health Informatics
  • Computer Science Applications

Fingerprint

Dive into the research topics of 'AweGNN: Auto-parametrized weighted element-specific graph neural networks for molecules'. Together they form a unique fingerprint.

Cite this