ARC-STAR: Auditable Post-Hoc Correction for PDE Foundation Models
About
Partial differential equation (PDE) foundation models are pretrained networks that forecast how physical fields like velocity and pressure evolve from a single reusable solver. On unfamiliar flows their predictions drift step by step, errors concentrate in a few regions, yet retraining destabilizes the network and uniform post-hoc correction overlooks this spatial concentration. To address this, we propose a frozen-solver post-hoc correction framework, Adaptive Risk-Calibrated Spatial Triage for Auditable Refinement (ARC-STAR). ARC-STAR organizes correction into three stages: a global corrector removes broad solver bias, a blockwise local refiner cleans the post-global residual, and, at deployment, a label-free score routes refinement to high-risk blocks under a compute budget. The framework is designed to be (i) frozen-host, preserving the pretrained solver without fine-tuning; (ii) auditable, with global and local stages trained and evaluated separately for measurable contributions; and (iii) budget-aware, using a blockwise interface that either refines the full field or routes limited compute to high-risk regions. Across five flow benchmarks spanning ten regime cells, ARC-STAR is the only method that cuts velocity rollout error by at least 36x over raw Poseidon on every cell. The global stage reduces raw host error by 91-99%, and the local stage further reduces the remaining post-global residual by up to 94.4%.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 10-step incompressible flow rollout | NS-SL moderate viscosity (m) (test) | UV Relative L2 Ratio0.0036 | 30 | |
| 10-step incompressible flow rollout | NS-SL extreme viscosity (x) (test) | UV relative-L2 ratio0.0072 | 30 | |
| 10-step incompressible flow rollout | NS-PwC moderate viscosity (m) (test) | UV Relative L2 Ratio0.0047 | 15 | |
| 10-step incompressible flow rollout | NS-PwC extreme viscosity (x) (test) | UV Relative L2 Ratio0.0056 | 15 | |
| 10-step incompressible flow rollout | NS-Sines moderate viscosity (m) (test) | UV Relative L2 Ratio0.0162 | 15 | |
| 10-step incompressible flow rollout | NS-Sines extreme viscosity (x) (test) | UV Relative L2 Ratio0.0213 | 15 | |
| 10-step incompressible flow rollout | NS-G moderate viscosity (m) (test) | UV relative-L2 ratio0.0045 | 15 | |
| 10-step incompressible flow rollout | NS-G extreme viscosity (x) (test) | UV Relative L2 Ratio0.0224 | 15 | |
| PDE solving | Kolmogorov Flow (Extension) | Median Relative Error1.72 | 3 | |
| PDE solving | Navier-Stokes G Modified | Median Relative Error0.45 | 3 |