DeepAI AI Chat
Log In Sign Up

Improving Molecule Properties Through 2-Stage VAE

by   Chenghui Zhou, et al.
Carnegie Mellon University

Variational autoencoder (VAE) is a popular method for drug discovery and there had been a great deal of architectures and pipelines proposed to improve its performance. But the VAE model itself suffers from deficiencies such as poor manifold recovery when data lie on low-dimensional manifold embedded in higher dimensional ambient space and they manifest themselves in each applications differently. The consequences of it in drug discovery is somewhat under-explored. In this paper, we study how to improve the similarity of the data generated via VAE and the training dataset by improving manifold recovery via a 2-stage VAE where the second stage VAE is trained on the latent space of the first one. We experimentally evaluated our approach using the ChEMBL dataset as well as a polymer datasets. In both dataset, the 2-stage VAE method is able to improve the property statistics significantly from a pre-existing method.


page 1

page 2

page 3

page 4


Disentangle VAE for Molecular Generation

Automatic molecule generation plays an important role on drug discovery ...

Unsupposable Test-data Generation for Machine-learned Software

As for software development by machine learning, a trained model is eval...

Learning Manifold Dimensions with Conditional Variational Autoencoders

Although the variational autoencoder (VAE) and its conditional extension...

Reproducible, incremental representation learning with Rosetta VAE

Variational autoencoders are among the most popular methods for distilli...

Sparse Representation Learning with Modified q-VAE towards Minimal Realization of World Model

Extraction of low-dimensional latent space from high-dimensional observa...

A Generative Approach Towards Improved Robotic Detection of Marine Litter

This paper presents an approach to address data scarcity problems in und...

Variational autoencoders in the presence of low-dimensional data: landscape and implicit bias

Variational Autoencoders (VAEs) are one of the most commonly used genera...