JTVAE
A junction tree variational autoencoder for generating valid molecular graphs with desired chemical properties.
What is JTVAE?
JT-VAE is a deep learning model for generating molecular graphs with desired chemical properties. It uses a junction tree decomposition to represent molecules as trees of chemically valid substructures, combined with a variational autoencoder to learn latent representations. This enables property-guided molecule generation for applications like drug discovery and materials design.
Researchers and practitioners in computational chemistry, drug discovery, and materials science who need to generate novel molecules with specific properties using machine learning.
It generates chemically valid molecules by enforcing syntactic constraints through junction tree decomposition, offers property-guided optimization, and provides an accelerated implementation for efficient training and inference.
Overview
Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)
Use Cases
Best For
- Generating novel drug-like molecules with optimized properties
- Exploring chemical space for materials design
- Molecular optimization using Bayesian optimization
- Learning latent representations of molecular structures
- Pre-training molecular language models on large chemical datasets
- Research in graph-based generative models for chemistry
Not Ideal For
- Production environments needing real-time molecule generation with low latency
- Teams without expertise in computational chemistry and deep learning
- Projects requiring Python 3.x and modern machine learning frameworks like TensorFlow 2
- Applications demanding cross-platform support beyond Linux
Pros & Cons
Pros
Uses junction tree decomposition to enforce syntactic validity, ensuring all generated molecules are chemically plausible based on substructure trees.
Integrates Bayesian optimization and joint training with property predictors, enabling targeted generation of molecules with specific desired traits like drug-likeness.
Includes optimized code in `fast_jtnn/` and `fast_molvae/` directories, significantly speeding up training and inference compared to the original version.
Based on a peer-reviewed ICML paper, providing a reproducible methodology with clear experimental scripts in directories like `bo/` and `molopt/`.
Cons
Requires Python 2.7, which is no longer supported, complicating integration with modern libraries and tools, as noted in the README.
Necessitates installation of RDKit and specific PyTorch versions, making initial configuration error-prone, especially for users unfamiliar with conda.
The README explicitly recommends using a newer repository (hgraph2graph), indicating this version may be less maintained and potentially buggy for current use.
Frequently Asked Questions
Related Projects
DiffDockImplementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking
Molecular TransformerMolecular Transformer is a neural machine translation model adapted for chemistry that predicts chemical reaction outcomes and retrosynthetic pathways. It translates between molecular representations (SMILES strings) to forecast how molecules react or how target molecules can be synthesized, accelerating discovery in organic chemistry and drug development. ## Key Features - **Retrosynthesis Prediction** — Predicts reactant molecules needed to synthesize a target product molecule. - **Uncertainty Calibration** — Provides confidence estimates for predictions, helping chemists assess reliability. - **SMILES Tokenization** — Uses custom tokenization of SMILES strings to treat molecules as sequences for transformer models. - **Data Augmentation** — Doubles training data by generating random equivalent SMILES representations via RDKit. - **Pre-trained Models** — Includes models trained on public datasets (USPTO_MIT, USPTO_STEREO) with mixed or separated reactant/reagent formats. ## Philosophy Molecular Transformer aims to make AI-assisted chemical reaction prediction accessible to organic chemists, with the goal of integrating these models into daily laboratory workflows to accelerate molecular discovery.
REINVENTREINVENT is a reinforcement learning framework specifically designed for de novo drug design, enabling the generation of novel molecular structures with optimized properties. It addresses the challenge of discovering new chemical entities by combining generative models with property prediction to explore chemical space efficiently. ## Key Features - **Reinforcement Learning Pipeline** — Uses RL to optimize molecular structures toward desired chemical properties and biological activities - **De Novo Molecular Generation** — Creates entirely new molecular entities rather than modifying existing compounds - **Property Optimization** — Incorporates scoring functions to guide generation toward molecules with specific target properties - **Template-Based Execution** — Provides configurable JSON templates for different running modes and experiments - **TensorBoard Integration** — Enables real-time monitoring and visualization of training logs and progress ## Philosophy REINVENT applies reinforcement learning principles to drug discovery, treating molecular generation as an optimization problem where the agent learns to propose molecules that maximize desired chemical and biological properties.
TargetDiffThe official implementation of 3D Equivariant Diffusion for Target-Aware Molecule Generation and Affinity Prediction (ICLR 2023)
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.