Do we need equivariant models for molecule generation?
Not on the map. Why?
Abstract (arXiv)
Deep generative models are increasingly used for molecular discovery, with most recent approaches relying on equivariant graph neural networks (GNNs) under the assumption that explicit equivariance is essential for generating high-quality 3D molecules. However, these models are complex, difficult to train, and scale poorly. We investigate whether non-equivariant convolutional neural networks (CNNs) trained with rotation augmentations can learn equivariance and match the performance of equivariant models. We derive a loss decomposition that separates prediction error from equivariance error, and evaluate how model size, dataset size, and training duration affect performance across denoising, molecule generation, and property prediction. To our knowledge, this is the first study to analyze learned equivariance in generative tasks.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 19 of its 25 references that are in the map. First posted on arXiv 13 Jul 2025.
Builds on
19 of 25 references.
References · 19
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,167
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
- 20183D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Datacited by 354
- 2020GEOM: Energy-annotated molecular conformations for property prediction and molecular generationcited by 243
- 2023Geometric Latent Diffusion Models for 3D Molecule Generationcited by 186
- 2019Symmetry-adapted generation of 3d point sets for the targeted discovery of moleculescited by 172
- 2017Underdamped Langevin MCMC: A non-asymptotic analysiscited by 133
- 2023MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generationcited by 98
- 2022Equivariant Energy-Guided SDE for Inverse Molecular Designcited by 73
- 2019Neural Empirical Bayescited by 46
- 20233D molecule generation by denoising voxel gridscited by 40
7 more
- 2020Learning a Continuous Representation of 3D Molecular Structures with Deep Generative Modelscited by 24
- 2022Equivariance versus Augmentation for Spherical Imagescited by 24
- 2024Structure-based drug design by denoising voxel gridscited by 19
- 2024Emergent Equivariance in Deep Ensemblescited by 18
- 2023Revisiting Data Augmentation for Rotational Invariance in Convolutional Neural Networkscited by 16
- 2024NEBULA: Neural Empirical Bayes Under Latent Representations for Efficient and Controllable Design of Molecular Librariescited by 3
- 2023Soft Random Sampling: A Theoretical and Empirical Analysiscited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.
- 2025Training Dynamics of Learning 3D-Rotational Equivariancecited by 1
- 2026Balancing Symmetry and Efficiency in Graph Flow Matchingcited by 0
- 2026Balancing Symmetry and Efficiency in Graph Flow Matching20 Feb 2026
- 2025Training Dynamics of Learning 3D-Rotational Equivariance2 Dec 2025