arXiv · 2506.10532
Equivariant Neural Diffusion for Molecule Generation
Abstract
We introduce Equivariant Neural Diffusion (END), a novel diffusion model for molecule generation in 3D that is equivariant to Euclidean transformations. Compared to current state-of-the-art equivariant diffusion models, the key innovation in END lies in its learnable forward process for enhanced generative modelling. Rather than pre-specified, the forward process is parameterized through a time- and data-dependent transformation that is equivariant to rigid transformations. Through a series of experiments on standard molecule generation benchmarks, we demonstrate the competitive performance of END compared to several strong baselines for both unconditional and conditional generation.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
François Cornet, Grigory Bartosh, Mikkel N. Schmidt, Christian A. Naesseth. 2025-06-12. Equivariant Neural Diffusion for Molecule Generation. https://arxiv.org/abs/2506.10532
Cite the original work for its findings. Save a collection to share your selection of sources.