Saved in:
Bibliographic Details
Main Authors: Mollaali, Amirhossein, Kim, Bongseok, Moya, Christian, Lin, Guang
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.16757
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908894419222528
author Mollaali, Amirhossein
Kim, Bongseok
Moya, Christian
Lin, Guang
author_facet Mollaali, Amirhossein
Kim, Bongseok
Moya, Christian
Lin, Guang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2603_16757
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle pADAM: A Plug-and-Play All-in-One Diffusion Architecture for Multi-Physics Learning
Mollaali, Amirhossein
Kim, Bongseok
Moya, Christian
Lin, Guang
Machine Learning
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.
title pADAM: A Plug-and-Play All-in-One Diffusion Architecture for Multi-Physics Learning
topic Machine Learning
url https://arxiv.org/abs/2603.16757