Generative Design

Generative Molecular Design

De novo design of novel molecules with desired properties using generative models — the core ML capability for lead generation and scaffold hopping in drug discovery.

When to Use This Skill

• Generate molecules with target properties (QED, LogP, SA, docking score)
• Explore chemical space around a hit/lead (analogue generation, scaffold hopping)
• Design molecules conditioned on a protein pocket (SBDD)
• Optimize multi-property objectives (Pareto front: potency + selectivity + ADMET)
• Benchmark or compare generative models (MOSES / GuacaMol suites)
• Build a RL-based focused library generator (REINVENT 4)
• Design linkers or grow fragments (fragment-based generative design)

Generation Paradigms

Evaluation Metrics (Know These)

Quick Start — SELFIES + GPT sampling

import selfies as sf
from rdkit import Chem

# Encode/decode SELFIES (guaranteed valid)
smiles = "CC(=O)Oc1ccccc1C(=O)O"  # aspirin
selfies_str = sf.encoder(smiles)
decoded_smiles = sf.decoder(selfies_str)
mol = Chem.MolFromSmiles(decoded_smiles)  # always valid

# Get SELFIES alphabet for tokenization
alphabet = sf.get_semantic_robust_alphabet()

# Decode a random generated SELFIES token sequence (always valid):
generated_tokens = ["[C]", "[Branch1]", "[C]", "[=O]", "[N]", "[H]"]
generated_smiles = sf.decoder("".join(generated_tokens))

Quick Start — REINVENT 4 scoring component

# Install: pip install reinvent
# REINVENT 4 uses TOML config for staged learning

import toml

config = {
    "run_type": "reinforcement_learning",
    "device": "cuda",
    "tb_logdir": "tb_logs",
    "json_out_config": "run_config.json",
    "parameters": {
        "use_checkpoint": False,
        "prior_file": "path/to/prior.prior",
        "agent_file": "path/to/prior.prior",
        "batch_size": 128,
        "n_steps": 1000,
    },
    "scoring": {
        "type": "custom_product",
        "parallel": False,
        "components": [
            {"type": "qed", "name": "QED", "weight": 1.0},
            {"type": "sa_score", "name": "SA", "weight": 1.0,
             "transform": {"type": "reverse_sigmoid", "low": 1.0, "high": 6.0, "k": 0.5}},
        ],
        "diversity_filter": {
            "type": "IdenticalMurckoScaffold",
            "minscore": 0.4,
            "bucket_size": 25,
        }
    }
}
with open("rl_config.toml", "w") as f:
    toml.dump(config, f)
# Run: reinvent -l rl_run.log rl_config.toml

Router — What to Read

Software Stack

Key Pitfalls

• SMILES LMs can generate 10-50% invalid → use SELFIES; or add validity filter post-hoc
• Reward hacking in RL: model learns degenerate structures that maximize score — add diversity filter + SA penalty
• FCD is not computed from structure: requires ChemNet embeddings (guacamol includes this)
• Novelty ≠ synthesizability: always check SA score ≤ 4, run retrosynthesis (ASKCOS/AiZynthFinder)
• 3D diffusion needs pocket quality: must use properly prepared protein (see homology-modeling → structure-prep)
• Mode collapse in VAE: monitor KL weight β; schedule β-VAE warmup

Related Skills

• torchdrug — GCPN, GraphAF, GraphDF, JT-VAE implementation
• rdkit — validity checks, property scoring oracles (QED, SA, fingerprints)
• docking — docking oracle for RL scoring (Vina/Gnina scoring function)
• pharmacophore — pharmacophore constraints for conditional generation
• homology-modeling → structure-prep — pocket preparation for SBDD
• scientific-skills:zinc-database — training/reference sets (ZINC20)
• mmpa (upcoming) — matched molecular pair analysis on generated series