arXiv Machine Learning

Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generation

arXiv:2606. 01595v1 Announce Type: new Abstract: Bayesian inference provides a principled framework for modeling epistemic uncertainty in neural networks by treating predictions as distributions rather than deterministic values.

arXiv Machine Learning
Sep 23

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.

By Yida Xiong, Jiameng Chen, Kun Li, Hongzhi Zhang, Xiantao Cai, Lei Lei, 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
Jun 15

Smoothing Dark Areas in Molecular Latent Diffusion

arXiv:2606. 13955v1 Announce Type: new Abstract: Latent diffusion is a promising framework for scalable 3D molecular generation, but it requires a latent space that remains smooth, valid, and navigable beyond posterior samples.

By Xi Wang, Jiahan Li, Yuxuan Xia, Yingcheng Wu, Shaoyi Zheng, Shengjie Wang