Physics-Informed Laplace Neural Operator for Solving Partial Differential Equations

TL;DR

The paper introduces PILNO, a physics-informed neural operator that improves PDE solving by embedding physics into training, enhancing generalization, especially in small-data and out-of-distribution scenarios.

Contribution

It proposes PILNO, combining an advanced neural operator backbone with physics-informed training techniques like virtual inputs and temporal weighting for better PDE approximation.

Findings

PILNO outperforms data-driven baselines in small-data regimes.

It achieves stronger out-of-distribution generalization.

PILNO reduces variability across different training runs.

Abstract

Neural operators have emerged as fast surrogate solvers for parametric partial differential equations (PDEs). However, purely data-driven models often require extensive training data and can generalize poorly, especially in small-data regimes and under unseen (out-of-distribution) input functions that are not represented in the training data. To address these limitations, we propose the Physics-Informed Laplace Neural Operator (PILNO), which enhances the Laplace Neural Operator (LNO) by embedding governing physics into training through PDE, boundary condition, and initial condition residuals. To improve expressivity, we first introduce an Advanced LNO (ALNO) backbone that retains a pole-residue transient representation while replacing the steady-state branch with an FNO-style Fourier multiplier. To make physics-informed training both data-efficient and robust, PILNO further leverages (i) virtual inputs: an unlabeled ensemble of input functions spanning a broad spectral range that provides abundant physics-only supervision and explicitly targets out-of-distribution (OOD) regimes; and (ii) temporal-causality weighting: a time-decaying reweighting of the physics residual that prioritizes early-time dynamics and stabilizes optimization for time-dependent PDEs. Across four representative benchmarks -- Burgers' equation, Darcy flow, a reaction-diffusion system, and a forced KdV equation -- PILNO consistently improves accuracy in small-data settings (e.g., N_train <= 27), reduces run-to-run variability across random seeds, and achieves stronger OOD generalization than purely data-driven baselines.