UniversalAGI
Blog

Epistemic Uncertainty in LIFT with Monte Carlo Sampling

A physics foundation model evaluates a design at much lower cost than a computational fluid dynamics (CFD) solver, but it reports no residuals, mesh sensitivity, or discretization error to indicate how far the prediction can be trusted. The design engineer needs to know where the model is reliable, how uncertain its predictions are, and which cases still require CFD. An uncertainty estimate can help allocate such simulation budget.

Uncertainty is commonly divided into two categories [1]. Aleatoric uncertainty comes from noise or variability not captured by the available inputs, such as measurement noise or unobserved physical conditions. Epistemic uncertainty reflects uncertainty about the model, often caused by limited or unrepresentative training data. Several models may fit the training data but disagree in unfamiliar regions of the input space. This disagreement can help prioritize model predictions for evaluation with CFD.

The Latent Interaction Field Transformer (LIFT) is UniversalAGI's in-house model architecture. It predicts physical fields at selected query points from geometry and operating conditions. We estimate model disagreement by adapting Monte Carlo dropout [3] to DropPath, also called stochastic depth [4], in LIFT's residual branches. Keeping DropPath active at inference produces predictions from a family of effective subnetworks. The spread between predictions shows how much the result depends on the sampled paths. We use that spread to estimate epistemic uncertainty. Figure 1 shows the schematic of the process.