arXiv Machine Learning

Transport-Coupled Bayesian Flows for Molecular Graph Generation

Transport-Coupled Bayesian Flows for Molecular Graph Generation (TopBF) addresses a key mismatch in existing diffusion models for molecular graph generation by eliminating the need for hard discretization during sampling. The framework generates graphs directly in continuous parameter distributions, learns graph topology via a Quasi-Wasserstein optimal‑transport coupling with geodesic costs, and enables property‑conditioned generation without retraining. Experiments on QM9 and ZINC250k show that TopBF achieves higher structural fidelity and more efficient generation compared to prior methods.

arXiv Machine Learning
Sep 7

Embedded Graph Flows for Categorical Graph Generation

Embedded Graph Flows (EGF) is a generative model for categorical graphs that learns continuous embeddings for node and unordered-edge categories and uses a permutation‑equivariant graph transformer to transport Gaussian noise toward these embeddings. A terminal readout then maps the embeddings back to discrete graph categories. EGF achieves competitive performance on molecular benchmarks, outperforming other methods on QM9 and maintaining low maximum mean discrepancy on ZINC250k.

By Ethan Ma, Zihan Wang, Chris Siu Yeung Chow, Xinguo Feng, Qingqing Li, Rui Jiang, Naipeng Dong, Guangdong Bai
arXiv Machine Learning
Sep 24

Variational Bayesian Flow Network for Graph Generation

The paper introduces Variational Bayesian Flow Network (VBFN), a graph generation model that lifts Bayesian updates to a joint Gaussian belief family with structured precisions, enabling coupled node and edge updates in a single fusion step. By constructing sample‑agnostic sparse precisions from a representation‑induced dependency graph, VBFN avoids label leakage while enforcing node‑edge consistency. Experiments on synthetic and molecular graph datasets show that VBFN improves fidelity and diversity over baseline methods.

By Yida Xiong, Jiameng Chen, Xiuwen Gong, Jia Wu, Shirui Pan, Wenbin Hu
arXiv Machine Learning
Jul 13

Autoregressive latent diffusion for 3D molecule generation

arXiv:2607. 09277v1 Announce Type: new Abstract: Three-dimensional (3D) molecule generation has been dominated by diffusion models, which achieve strong generation quality but typically require the molecular size to be specified a priori.

By Federico Ottomano, Gaopeng Ren, Yingzhen Li, Kim E. Jelfs, Alex M. Ganose
arXiv Machine Learning
Aug 28

Gromov-Monge Flow Matching for Equivariant Graph Generation

The paper introduces Gromov-Monge Flow Matching, a method that incorporates permutation-equivariance into generative graph models by aligning graph pairs up to node relabeling using the Gromov–Monge distance. It shows theoretically that quotient couplings can be lifted to aligned representatives without extra cost and that symmetrization yields equivariant flow-matching minimizers, even for categorical endpoints. Practically, the authors build minibatch couplings with Gromov–Wasserstein relaxations and optional outer assignments, improving sample quality in continuous graph and categorical molecular generation while remaining compatible with standard equivariant architectures.

By Moritz Piening, Christian Wald
arXiv Machine Learning
Jul 23

Boltzmann-Expected Molecular Design with Decoupled Annealing Flows

arXiv:2607. 19519v1 Announce Type: cross Abstract: Most 3D properties relevant to molecular design, including free energies and shape descriptors, are $\textit{expectations}$ over the Boltzmann distribution over 3D configurations of a molecular graph.

By Selma Moqvist, Richard Beckmann, Ross Irwin, Roc\'io Mercado, Simon Olsson
arXiv Machine Learning
Jul 8

Multimodal Molecular Representation Learning with Graph Neural Networks, Deep & Cross Networks, and SMILES Embeddings

arXiv:2607. 05736v1 Announce Type: new Abstract: Molecular property prediction often relies on isolated data modalities, where continuous 3D graph neural networks (GNNs) struggle to efficiently capture long-range topological dependencies and exact macroscopic heuristics.

By Qiwei Han, Chi Zhou, Ruobing Wang, Zheng Ma