GAGA: Gaussianity-Aware Gaussian Approximation for Efficient 3D Molecular Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qu, Jingxiang, Gao, Wenhan, Xu, Ruichen, Liu, Yi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915808350830592
author Qu, Jingxiang
Gao, Wenhan
Xu, Ruichen
Liu, Yi
author_facet Qu, Jingxiang
Gao, Wenhan
Xu, Ruichen
Liu, Yi
contents Gaussian Probability Path based Generative Models (GPPGMs) generate data by reversing a stochastic process that progressively corrupts samples with Gaussian noise. Despite state-of-the-art results in 3D molecular generation, their deployment is hindered by the high cost of long generative trajectories, often requiring hundreds to thousands of steps during training and sampling. In this work, we propose a principled method, named GAGA, to improve generation efficiency without sacrificing training granularity or inference fidelity of GPPGMs. Our key insight is that different data modalities obtain sufficient Gaussianity at markedly different steps during the forward process. Based on this observation, we analytically identify a characteristic step at which molecular data attains sufficient Gaussianity, after which the trajectory can be replaced by a closed-form Gaussian approximation. Unlike existing accelerators that coarsen or reformulate trajectories, our approach preserves full-resolution learning dynamics while avoiding redundant transport through truncated distributional states. Experiments on 3D molecular generation benchmarks demonstrate that our GAGA achieves substantial improvement on both generation quality and computational efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2507_09043
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle GAGA: Gaussianity-Aware Gaussian Approximation for Efficient 3D Molecular Generation
Qu, Jingxiang
Gao, Wenhan
Xu, Ruichen
Liu, Yi
Machine Learning
Gaussian Probability Path based Generative Models (GPPGMs) generate data by reversing a stochastic process that progressively corrupts samples with Gaussian noise. Despite state-of-the-art results in 3D molecular generation, their deployment is hindered by the high cost of long generative trajectories, often requiring hundreds to thousands of steps during training and sampling. In this work, we propose a principled method, named GAGA, to improve generation efficiency without sacrificing training granularity or inference fidelity of GPPGMs. Our key insight is that different data modalities obtain sufficient Gaussianity at markedly different steps during the forward process. Based on this observation, we analytically identify a characteristic step at which molecular data attains sufficient Gaussianity, after which the trajectory can be replaced by a closed-form Gaussian approximation. Unlike existing accelerators that coarsen or reformulate trajectories, our approach preserves full-resolution learning dynamics while avoiding redundant transport through truncated distributional states. Experiments on 3D molecular generation benchmarks demonstrate that our GAGA achieves substantial improvement on both generation quality and computational efficiency.
title GAGA: Gaussianity-Aware Gaussian Approximation for Efficient 3D Molecular Generation
topic Machine Learning
url https://arxiv.org/abs/2507.09043