Search papers, labs, and topics across Lattice.
The paper introduces pADAM, a novel diffusion-based generative framework designed to learn a shared probabilistic prior across diverse partial differential equation (PDE) families. This unified architecture enables both forward prediction and inverse inference without retraining, leveraging a learned joint distribution of system states and physical parameters. Experiments across various benchmarks, including scalar diffusion and Navier-Stokes equations, demonstrate pADAM's accuracy, uncertainty quantification capabilities, and ability to perform probabilistic model selection from sparse data.
Forget retraining for every new physics problem – pADAM learns a single generative model that handles forward prediction, inverse inference, and even identifies governing laws across different PDEs.
Generalizing across disparate physical laws remains a fundamental challenge for artificial intelligence in science. Existing deep-learning solvers are largely confined to single-equation settings, limiting transfer across physical regimes and inference tasks. Here we introduce pADAM, a unified generative framework that learns a shared probabilistic prior across heterogeneous partial differential equation families. Through a learned joint distribution of system states and, where applicable, physical parameters, pADAM supports forward prediction and inverse inference within a single architecture without retraining. Across benchmarks ranging from scalar diffusion to nonlinear Navier--Stokes equations, pADAM achieves accurate inference even under sparse observations. Combined with conformal prediction, it also provides reliable uncertainty quantification with coverage guarantees. In addition, pADAM performs probabilistic model selection from only two sparse snapshots, identifying governing laws through its learned generative representation. These results highlight the potential of generative multi-physics modeling for unified and uncertainty-aware scientific inference.
