Poster
Equivariant Neural Diffusion for Molecule Generation
François Cornet · Grigory Bartosh · Mikkel Schmidt · Christian Andersson Naesseth
East Exhibit Hall A-C #2403
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 that END improves on several strong baselines for both unconditional and conditional generation.
Live content is unavailable. Log in and register to view live content