TreeRef-BFN: Equivariance-Free De Novo Molecule Generation based on 2D Topology and Internal 3D Geometry
Not on the map. Why?
Abstract (arXiv)
De novo 3D molecular generation jointly models molecular size, topology, and geometry. Most methods pre-sample molecular size and generate Cartesian coordinates, limiting variable-size conditional tasks such as fragment completion and scaffold decoration while often relying on equivariant architectures. Internal-coordinate methods avoid rigid-body redundancy but typically require a known molecular graph or autoregressive construction, which may accumulate errors. We propose TreeRef, a tree-based molecular representation that assigns molecular topology and topology-dependent local 3D geometry to a naturally variable-size tree. RingRef nodes encode ring closures while preserving the tree structure, while Null nodes allow molecular size to emerge directly from node occupancy. Based on TreeRef, we develop TreeRef-BFN, a Bayesian Flow Network with a standard Transformer backbone that globally couples these locally defined variables and jointly generates discrete molecular variables and continuous local geometry. A single pretrained TreeRef-BFN supports unconditional generation and variable-size structure-conditioned 3D generation through masking alone, without retraining. Empirical studies demonstrate strong chemical validity, molecular stability, and diversity, accurate local geometric distributions, fast sampling, and competitive property-conditioned generation, establishing TreeRef-BFN as an efficient and flexible framework for 3D molecular generation.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 27 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
22 of 27 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 652
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generationcited by 516
- 2022Torsional Diffusion for Molecular Conformer Generationcited by 263
- 2020GEOM: Energy-annotated molecular conformations for property prediction and molecular generationcited by 243
- 2023Geometric Latent Diffusion Models for 3D Molecule Generationcited by 186
- 2023MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generationcited by 98
- 2025All-atom Diffusion Transformers: Unified generative modelling of molecules and materialscited by 75
10 more
- 2023Bayesian Flow Networkscited by 63
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 41
- 2023Geometry-Complete Diffusion for 3D Molecule Generation and Optimizationcited by 37
- 2023Learning Joint 2D & 3D Diffusion Models for Complete Molecule Generationcited by 27
- 2025FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generationcited by 21
- 2024Unified Generative Modeling of 3D Molecules via Bayesian Flow Networkscited by 18
- 2025NExT-Mol: 3D Diffusion Meets 1D Language Modeling for 3D Molecule Generationcited by 17
- 2024Any-Property-Conditional Molecule Generation with Self-Criticism using Spanning Treescited by 3
- 2026Geometric Flow Matching for Molecular Conformation Generation via Manifold Decompositioncited by 3
- 2026VecMol: Vector-Field Representations for 3D Molecule Generationcited by 1
Built on by
Not cited yet.