← Back to Machine Learning (Statistics)
stat.ML

Why one-step generative models work for physics-based probability distributions

Likun Lin, Zhongjian Wang, Jack Xin, Zhiwen Zhang

May 20, 2026

Generative models excel empirically but lack theory for scientific computing tasks like sampling from PDE solutions. This work proves that probability measures arising from PDEs have special structure—they satisfy doubling conditions—which guarantees smooth transport maps between simple distributions and complex PDE-induced ones. This regularity justifies training a single neural network to learn the transport map, and the authors derive concrete error bounds for the DeepParticle model showing convergence rates.
Published as On the Regularity and Generalization of One-Step Wasserstein-guided Generative Models for PDE-Induced Measures arXiv:2605.21388
Read the original paper →