Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Adaptive Physics Transformer with Fused Global-Local Attention for Subsurface Energy Systems

About

The Earth's subsurface is a cornerstone of modern society, providing essential energy resources like hydrocarbons, geothermal, and minerals while serving as the primary reservoir for $CO_2$ sequestration. However, full physics numerical simulations of these systems are notoriously computationally expensive due to geological heterogeneity, high resolution requirements, and the tight coupling of physical processes with distinct propagation time scales. Here we propose the $\textbf{Adaptive Physics Transformer}$ (APT), a geometry-, mesh-, and physics-agnostic neural operator that explicitly addresses these challenges. APT fuses a graph-based encoder to extract high-resolution local heterogeneous features with a global attention mechanism to resolve long-range physical impacts. Our results demonstrate that APT outperforms state-of-the-art architectures in subsurface tasks across both regular and irregular grids with robust super-resolution capabilities. Notably, APT is the first architecture that learns directly from HR-adaptive mesh refinement simulations. We also demonstrate APT's favorable scaling behavior and cross-dataset learning capability, positioning it as a robust and scalable backbone for large-scale subsurface foundation model development.

Xin Ju, Nok Hei (Hadrian) Fung, Yuyan Zhang, Carl Jacquemyn, Matthew Jackson, Randolph Settgast, Sally M. Benson, Gege Wen• 2026

Related benchmarks

TaskDatasetResultRank
Subsurface Flow Simulationdiscontinuous von Karman OOD (test)
R^20.9107
5
Subsurface Flow Simulationdiscontinuous von Karman (train)
R^20.9351
5
Gas saturation predictionSubsurface Energy Systems 19-step rollout (test)
Time (s)0.017
4
Spatiotemporal field predictionATES adaptive mesh
R299
4
Basin-scale CO2 storage simulationBasin-scale CO2 storage with LGR
Pressure Change (δΔPg)56
4
Subsurface Flow SimulationSubsurface Energy Systems simulation--
4
Showing 6 of 6 rows

Other info

Follow for update