Tree Variational Autoencoders
Laura Manduchi, Moritz Vandenhirtz, Alain Ryser, Julia E. Vogt
Abstract
Traditional antibiotic development remains slow and costly, with virtually no new classes targeting resistant gramnegative bacteria (GNB), highlighting the need for innovative approaches in generating new gram-negative antibacterial (GNAB) compounds to combat GNB. This study introduces a computational framework utilizing Junction Tree Variational Autoencoders (JT-VAE) and computational screening to design novel GNAB compounds. Generated compounds underwent Tanimoto similarity analysis and Lipinski’s Rule of Five (LRo5) screening to assess structural novelty and oral bioavailability, followed by agglomerative hierarchical clustering with cophenetic and silhouette score evaluation. Training on curated GNAB compounds from ChEMBL, the GVAE model, specifically the Junction Tree Variational Autoencoder (JT-VAE) model, demonstrated superior performance, producing molecules with 100% validity. Subsequent filtering retained 2,141 compounds (21.41%) within optimal Tanimoto similarity thresholds that balance novelty with known antibacterial substructures while meeting LRo5 with violations. Property distributions aligned with fragment-based design principles, such as the Rule of Three. Clustering analysis revealed nuanced performance differences among benchmark models: JT-VAE achieved a cophenetic correlation coefficient (CCC) of 0.934 and silhouette score of 0.76, while unconstrained GVAE demonstrated superior clustering performance (silhouette score: 0.86, CCC: 0.98) despite occasionally generating disconnected molecular fragments. JT-VAE’s fragment-based approach successfully addresses limitations of atom-by-atom generation, particularly in handling aromatic systems essential for antibacterial activity. By combining hierarchical graph generation with rigorous cheminformatics filtering, this study provides a reproducible blueprint for accelerating GNAB compound discovery, advancing toward timely, data- driven solutions for the escalating antimicrobial resistance crisis.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a293f662-2ec3-4c68-b97d-9d134be0a083Cited by top-tier papers5
- TreeVI: Reparameterizable Tree-structured Variational Inference for Instance-level Correlation CapturingJunxi Xiao, Qinliang SuNeurIPS 2024 · 2 citations
- HoT-VI: Reparameterizable Variational Inference for Capturing Instance-Level High-Order CorrelationsJunxi Xiao, Qinliang Su, Zexin YuanNeurIPS 2025
- From Logits to Hierarchies: Hierarchical Clustering made SimpleEmanuele Palumbo, Moritz Vandenhirtz, Alain Ryser, Imant Daunhawer et al.ICML 2025
- PhyloVAE: Unsupervised Learning of Phylogenetic Trees via Variational AutoencodersTianyu Xie, Harry Richman, Jiansi Gao, Frederick A. Matsen IV et al.ICLR 2025
- HDTree: Generative Modeling of Cellular Hierarchies for Robust Lineage InferenceZelin Zang, WenZhe Li, Yongjie Xu, Chang Yu et al.ICML 2026
Builds on17
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- NVAE: A Deep Hierarchical Variational AutoencoderArash Vahdat, Jan KautzNeurIPS 2020 · 1,141 citations
- Contrastive ClusteringYunfan Li, Peng Hu, Jerry Zitao Liu, Dezhong Peng et al.AAAI 2021 · 798 citations
- On Computing Probabilistic Explanations for Decision TreesMarcelo Arenas, Pablo Barceló, Miguel A. Romero Orth, Bernardo SubercaseauxNeurIPS 2022 · 57 citations
Related papers
- Path-Aware and Structure-Preserving Generation of Synthetically Accessible MoleculesJuhwan Noh, Dae-Woong Jeong, Kiyoung Kim, Sehui Han et al.ICML 2022 · 11 citations
- CogMol: Target-Specific and Selective Drug Design for COVID-19 Using Deep Generative ModelsVijil Chenthamarakshan, Payel Das, Samuel C. Hoffman, Hendrik Strobelt et al.NeurIPS 2020 · 82 citations
- FragFM: Hierarchical Framework for Efficient Molecule Generation via Fragment-Level Discrete Flow MatchingJoongwon Lee, Seonghwan Kim, Seokhyun Moon, Hyunwoo Kim et al.ICLR 2026 · 6 citations
- 3DLinker: An E(3) Equivariant Variational Autoencoder for Molecular Linker DesignYinan Huang, Xingang Peng, Jianzhu Ma, Muhan ZhangICML 2022 · 65 citations
- Graph VQ-Transformer (GVT): Fast and Accurate Molecular Generation via High-Fidelity Discrete LatentsHaozhuo Zheng, Cheng Wang, Yang LiuAAAI 2026
