Timezone: »

 
Poster
Modular Flows: Differential Molecular Generation
Yogesh Verma · Samuel Kaski · Markus Heinonen · Vikas Garg

Thu Dec 01 02:00 PM -- 04:00 PM (PST) @ Hall J #909
in Poster Session 6 »

Generating new molecules is fundamental to advancing critical applications such as drug discovery and material synthesis. Flows can generate molecules effectively by inverting the encoding process, however, existing flow models either require artifactual dequantization or specific node/edge orderings, lack desiderata such as permutation invariance, or induce discrepancy between encoding and decoding steps that necessitates post hoc validity correction. Inspired by graph PDEs, we circumvent these issues with novel continuous normalizing E(3)-equivariant flows, based on a system of coupled node ODEs, that repeatedly reconcile locally toward globally aligned densities. Our models can be cast as message passing temporal networks, and result in superlative density estimation and molecular generation. In particular, our generated samples achieve state of the art on both the standard QM9 and ZINC250K benchmarks.

Keywords: graph neural networks · normalizing flow · neural ode · molecule generation
[ Paper]  [ Poster]  [ OpenReview] 

Author Information

Yogesh Verma (Aalto University)
Samuel Kaski (Aalto University and University of Manchester)
Markus Heinonen (Aalto University)
Vikas Garg (Aalto University/YaiYai Ltd)

More from the Same Authors