Tiered Graph Autoencoders with PyTorch Geometric for Molecular Graphs

by   Daniel T Chang, et al.

Tiered latent representations and latent spaces for molecular graphs provide a simple but effective way to explicitly represent and utilize groups (e.g., functional groups), which consist of the atom (node) tier, the group tier and the molecule (graph) tier. They can be learned using the tiered graph autoencoder architecture. In this paper we discuss adapting tiered graph autoencoders for use with PyTorch Geometric, for both the deterministic tiered graph autoencoder model and the probabilistic tiered variational graph autoencoder model. We also discuss molecular structure information sources that can be accessed to extract training data for molecular graphs. To support transfer learning, a critical consideration is that the information must utilize standard unique molecule and constituent atom identifiers. As a result of using tiered graph autoencoders for deep learning, each molecular graph possesses tiered latent representations. At each tier, the latent representation consists of: node features, edge indices, edge features, membership matrix, and node embeddings. This enables the utilization and exploration of tiered molecular latent spaces, either individually (the node tier, the group tier, or the graph tier) or jointly, as well as navigation across the tiers.


page 1

page 2

page 3

page 4


Deep Learning for Molecular Graphs with Tiered Graph Autoencoders and Graph Classification

Tiered graph autoencoders provide the architecture and mechanisms for le...

Decoding Molecular Graph Embeddings with Reinforcement Learning

We present RL-VAE, a graph-to-graph variational autoencoder that uses re...

Operator Autoencoders: Learning Physical Operations on Encoded Molecular Graphs

Molecular dynamics simulations produce data with complex nonlinear dynam...

Uncovering the Folding Landscape of RNA Secondary Structure with Deep Graph Embeddings

Biomolecular graph analysis has recently gained much attention in the em...

A Graph VAE and Graph Transformer Approach to Generating Molecular Graphs

We propose a combination of a variational autoencoder and a transformer ...

Geometric Graph Representations and Geometric Graph Convolutions for Deep Learning on Three-Dimensional (3D) Graphs

The geometry of three-dimensional (3D) graphs, consisting of nodes and e...

Variational Autoencoding Molecular Graphs with Denoising Diffusion Probabilistic Model

In data-driven drug discovery, designing molecular descriptors is a very...

Please sign up or login with your details

Forgot password? Click here to reset