Sensitivity-Constrained Neural Operators for Data-Efficient Forward and Inverse Modeling of Partial Differential Equation Systems

📅 2026-08-30
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
研究通过引入敏感性约束的神经算子(SC-NOs),利用求解器导出的雅可比矩阵监督,提高高维偏微分方程系统正向和反向建模的效率与准确性。
📝 Abstract
Neural operators provide fast surrogates for partial differential equation (PDE) solvers, but their reliability can degrade for high-dimensional spatial inputs and inverse or repeated inference. State-only training constrains solution values but not the learned input--output response. We study sensitivity-constrained neural operators (SC-NOs), which augment standard training with sampled solver-derived Jacobian supervision. Selected sensitivities from differentiable solvers or discrete adjoints are matched during training, allowing response information to be amortized across minibatches without imposing the full Jacobian at every update. We evaluate SC-NO on advection--diffusion and RANS--Spalart--Allmaras benchmarks, input-dimensionality scaling tests, long-horizon autoregressive rollout, and a shallow-water Tohoku tsunami source-inversion case. Sensitivity supervision improves forward prediction and yields larger gains in gradient-based inverse reconstruction of distributed fields. Scaling experiments show an improved accuracy--cost tradeoff for high-dimensional gridded inputs, while ablations indicate that state values and Jacobian information provide complementary supervision. In the tsunami case, SC-FNO reconstructs gridded seafloor deformation from sparse early gauge observations and forecasts subsequent wave propagation in a near-real-time proof-of-concept workflow. These results support sampled sensitivity supervision as a practical way to improve neural PDE surrogates when forward accuracy, inverse stability, robustness, and computational cost must be considered together.
Problem

Research questions and friction points this paper is trying to address.

Neural Operators
Sensitivity-Constrained
Partial Differential Equations
Inverse Modeling
Forward Prediction
Innovation

Methods, ideas, or system contributions that make the work stand out.

Sensitivity-Constrained Neural Operators
Jacobian Supervision
High-Dimensional Inputs
Inverse Reconstruction
Computational Cost
A
Abdolmehdi Behroozi
Department of Civil and Environmental Engineering, Penn State University, University Park, PA 16802, USA
Chaopeng Shen
Chaopeng Shen
Professor in Water Resources Engineering, Pennsylvania State University
AI/MLDifferentiable ModelingHydrology & FloodsEcosystemWater quality
Daniel Kifer
Daniel Kifer
Penn State University
privacymachine learning
K
Kathryn Lawson
Department of Civil and Environmental Engineering, Penn State University, University Park, PA 16802, USA