3D Graph Conditional Distributions via Semi-Equivariant Continuous Normalizing Flows

Abstract

A general method for learning the conditional distribution $p(G | \hat{G})$ of two 3D graphs is proposed. The method is designed to be invariant to rigid body transformations and to permutations of the vertices of either graph. The core of the method is a continuous normalizing flow and semi-equivariance conditions are established to ensure the aforementioned invariance conditions. The utility of the technique is demonstrated as a conditional generative model for the molecular setting.

Cite

Text

Rozenberg et al. "3D Graph Conditional Distributions via Semi-Equivariant Continuous Normalizing Flows." ICLR 2023 Workshops: ML4Materials, 2023.

Markdown

[Rozenberg et al. "3D Graph Conditional Distributions via Semi-Equivariant Continuous Normalizing Flows." ICLR 2023 Workshops: ML4Materials, 2023.](https://mlanthology.org/iclrw/2023/rozenberg2023iclrw-3d/)

BibTeX

@inproceedings{rozenberg2023iclrw-3d,
  title     = {{3D Graph Conditional Distributions via Semi-Equivariant Continuous Normalizing Flows}},
  author    = {Rozenberg, Eyal and Rivlin, Ehud and Freedman, Daniel},
  booktitle = {ICLR 2023 Workshops: ML4Materials},
  year      = {2023},
  url       = {https://mlanthology.org/iclrw/2023/rozenberg2023iclrw-3d/}
}