3D Molecule Generation from Rigid Motifs via SE(3) Flows
Roman Poletukhin, Marcel Kollovieh, Eike Eberhard, Stephan Günnemann
原文摘要
Three-dimensional molecular structure generation is typically performed at the level of individual atoms, yet molecular graph generation techniques often consider fragments as their structural units. Building on the advances in frame-based protein structure generation, we extend these fragmentation ideas to 3D, treating general molecules as sets of rigid-body motifs. Utilising this representation, we employ SE(3)-equivariant generative modelling for de novo 3D molecule generation from rigid motifs. In our evaluations, we observe results competitive with state-of-the-art bond-implicit methods, surpassing them in atom stability on GEOM-Drugs, while yielding a 2x to 10x reduction in generation steps and 3.5x compression in molecular representations compared to standard atom-based methods. We release our code at https://github.com/rpoletukhin/MotiFlow.
cs.LG