Source-linked AI summary

Learning-guided Kansa collocation for forward and inverse PDEs beyond linearity

Zheyuan Hu, Weitao Chen, Cengiz Öztireli, Chenliang Zhou, Fangcheng Zhong

arXiv:2602.07970v3cs.CEcs.AIcs.LGmath.NA

TL;DR

The paper addresses the computational and discretization challenges of PDE solving and the limited evidence on CNF solver behavior across broader settings. It extends learning-guided Kansa collocation to coupled and nonlinear PDEs, applies it to forward and inverse problems, and systematically compares solvers across benchmarks. The study concludes that learning-guided Kansa solvers are promising and flexible for coupled or nonlinear PDE systems, while noting optimization and scope limitations.

  • Problem

    PDE solvers face dimensionality, computational-cost, and domain-discretization challenges, while CNF performance comparisons and applications to inverse problems remain unclear.

  • Method

    The paper extends CNF-driven Kansa collocation to coupled and nonlinear PDEs, uses self-tuning objectives, and evaluates representative classical and neural solvers on benchmark problems.

  • Results

    The study provides a systematic empirical comparison of CNF solvers with classical and neural PDE solvers across accuracy, efficiency, convergence, and complexity.

  • Takeaways & Limitations

    Learning-guided Kansa solvers can serve as a promising and flexible tool for coupled or nonlinear PDE systems.

  • Takeaways & Limitations

    Local optimization can settle at different minima depending on the initial guess, while more precise predictions require multiple runs and higher computational cost.

Abstract

from arXiv · show

Partial Differential Equations are precise in modelling the physical, biological and graphical phenomena. However, the numerical methods suffer from the curse of dimensionality, high computation costs and domain-specific discretization. We aim to explore pros and cons of different PDE solvers, and apply them to specific scientific simulation problems, including forwarding solution, inverse problems and equations discovery. In particular, we extend the recent CNF (NeurIPS 2023) framework solver to coupled and non-linear settings, together with down-stream applications. The outcomes include implementation of selected methods, self-tuning techniques, evaluation on benchmark problems and a comprehensive survey of neural PDE solvers and scientific simulation applications.

1 INTRODUCTION

The paper extends a learning-guided Kansa/CNF PDE solver beyond single-variable linear equations to coupled and nonlinear PDEs, while evaluating selected solvers across scientific-simulation criteria.

  • Prior CNF-based Kansa work introduced automatic RBF shape-parameter tuning but focused on single-variable linear PDEs.
  • This paper extends the framework to multiple unknown functions and nonlinear PDEs for forward computation and inverse problems.
  • The comparison evaluates classical and neural PDE solvers using accuracy, efficiency, resource use, convergence speed, complexity, and scientific-simulation utility.

2 RELATED WORK

The related work positions PDE solvers across classical discretization-based and neural approaches, and identifies unresolved questions about CNF applications to inverse problems and solver comparisons.

  • Representative PDE benchmarks vary in operator linearity and the dimensionality of the solution field.
  • Classical FDM and FEM methods face dimensionality, computational-cost, and domain-discretization challenges, while PINNs and FNOs address high-dimensional or unseen-scenario settings.
  • Inverse problems estimate unknown parameters or inputs from solution observations, but CNF applications including differentiable-rendering connections remain unclear.

3 METHODOLOGY

The methodology combines mesh-free Kansa RBF collocation with extensions for coupled fields and nonlinear operators, using time-stepping, nonlinear residual solvers, and automatic hyperparameter tuning.

  • PDE formulation: The PDE formulation combines a differential operator and boundary operators with source and boundary-condition functions.
  • Kansa formulation: Kansa approximates PDE solutions as linear combinations of kernel functions centered at collocation points, with coefficients solved from the resulting system.
  • Coupled solution fields: Coupled multi-dimensional fields are represented with separate approximations and assembled through a coupling operator acting across solution dimensions.
  • Nonlinear operators: For nonlinear operators, the framework uses differentiable operator matrices and either time-stepping schemes or a fully nonlinear solver on PDE residuals.
  • Time-stepping schemes: The methodology includes forward Euler, IMEX, backward Euler, and Crank-Nicolson time-stepping, alongside nonlinear-system solution procedures.
  • Auto-tuning: The fully nonlinear approach directly minimizes collocation-point PDE residuals, solution total variation, and available training L2 loss through weighted penalties.

4 EVALUATION

The evaluation compares Kansa-based solvers across forward and inverse advection, coupled Lotka–Volterra and Maxwell systems, and nonlinear Burgers’ equations. Kansa achieves strong accuracy and convergence results, while computational cost, local minima, and retraining remain practical constraints.

  • Metrics: L2 risk averages discretized solver error over Ntest spatial-temporal test points, while relative L2 risk normalizes this error by the ground-truth L2 norm.
  • Advection equation: All evaluated solvers improve test-point precision as the training-data scaling factor Cscale increases.
  • Advection equation: Kansa achieves the least advection error, up to 10^-6, with Cscale = 42, and outperforms other methods in accuracy and convergence speed.
  • Advection equation: Above Cscale = 102, Kansa exceeds the available memory limit because of increasing computational cost.
  • Coupled PDEs: For coupled equations, Kansa errors converge as Cscale increases; training and inference take 0.4034 and 0.0001 seconds on Lotka–Volterra, and 0.4486 and 0.0005 seconds on Maxwell’s equations.
  • Inverse problems: SciPy Powell successfully infers four Lotka–Volterra parameters with high accuracy and decent computational cost, whereas local optimization methods can become trapped in initial-guess-dependent local minima.
  • Nonlinear PDEs: For Burgers’ equation, the fully nonlinear approach outperforms the time-stepping schemes in accuracy, while Crank–Nicolson is more accurate than IMEX and backward Euler.
  • Nonlinear PDEs: The fully nonlinear Burgers’ solver has longer training because of heavier computation and nonlinear solvers, but reduced inference time through coefficient reuse; retraining is required for each new parameter.

5 CONCLUSIONS

The paper extends learning-guided Kansa solvers to coupled and nonlinear PDEs and evaluates them alongside classical and neural alternatives. It positions these methods for forward and inverse scientific simulation while identifying open theoretical and integration directions.

  • Contributions: Learning-guided Kansa solvers are generalized from linear PDEs to coupled and nonlinear systems, addressing the loss property of linear reordering.The extension broadens CNF-driven self-tuning mesh-free solvers to forward modeling and inverse problem formulations.
  • Method context: Kansa methods support mesh-free PDE computation without relying on domain-specific discretization.The method is grounded in radial basis functions and is described as a mesh-free solver approach.
  • Contributions: The study systematically compares CNF solvers with classical and neural PDE solvers across accuracy, efficiency, convergence, and complexity.The comparisons are conducted on benchmark problems and are intended to clarify strengths and limitations within the broader solver landscape.
  • Conclusions: Kansa-based methods are presented as a promising and flexible tool for coupled or nonlinear PDE systems.The stated future directions include theoretical analysis of error and convergence, neural-field applications, and integration with differentiable scientific pipelines.
  • PDE formulation: Boundary conditions specify behavior on the domain boundary, while initial conditions specify behavior at t = 0.These additional conditions enforce uniqueness for differential-equation solutions containing integration constants.
  • Benchmarks: The benchmark set includes advection and Lotka-Volterra predator-prey equations, with solutions visualized across representative settings.The supplied figures show advection in 1D, 2D, and 3D and the Lotka-Volterra solution with phase space.

B.5 MAXWELL’S EQUATIONS

The Maxwell benchmark reduces the source-free electromagnetic system to one-dimensional wave equations for electric and magnetic fields. Analytical solutions are obtained through characteristic variables and d’Alembert’s formula, then visualized for specified initial fields.

  • Formulation: Maxwell’s equations relate electric and magnetic fields to spatial position, time, charge density, and current density.The fields are E(r, t) and B(r, t), while ρ(r, t) and J(r, t) describe charge and current densities.
  • Formulation: In general, Maxwell’s equations are linear with respect to E and B, and time-varying magnetic and electric fields induce one another.The equations also include electric charge sources and current density in their differential form.
  • Reduction: The derivation converts Maxwell’s equations into inhomogeneous wave equations with source terms before restricting the benchmark to a source-free 1D case.The simplification assumes ρ = 0 and J = 0, with E aligned along z and B along y while propagation occurs along x.
  • Analytical solution: Characteristic variables u = Ez + By and v = Ez − By propagate along x − ct and x + ct, respectively.The propagation relations follow d’Alembert’s formula and use c as the speed of light in vacuum.
  • Analytical solution: The analytical fields Ez and By are reconstructed from the characteristic solutions through sums and differences of f(x ± ct) and g(x ± ct).The reconstruction is given in the analytical solution equations for the reduced 1D system.
  • Visualization: The visualization uses f(x) = sin(2πx) + 0.5 sin(4πx) and g(x) = cos(2πx) + 0.5 cos(4πx) on x ∈ [0, 1] and t ∈ [0, 0.5].These initial fields define the solution shown in Figure 3.

B.6 VISCOUS BURGERS’ EQUATION

The viscous Burgers’ equation combines nonlinear advection and diffusion. The paper derives an analytical solution through the Cole–Hopf transformation, reducing the transformed problem to the standard heat equation before visualization.

  • Formulation: Viscous Burgers’ equation captures nonlinear advection and diffusion phenomena in dynamics.The viscosity parameter ν is a positive constant, and the equation is posed with an initial condition u0(x).
  • Cole–Hopf transformation: The Cole–Hopf transformation converts the unknown function u(x, t) into a transformed function ϕ(x, t).The derivation computes the required spatial and temporal derivatives and substitutes them into the Burgers’ equation.
  • Analytical solution: The resulting transformed equation is reduced to the standard heat equation.This reduction follows substitution of the rescaled derivatives into the derived equation.
  • Transformation: Rescaling ϕ by a time-dependent factor preserves the Cole–Hopf transformation because the factor is independent of x.The transformed derivatives are then substituted into the intermediate equation.
  • Analytical solution: The analytical Burgers’ solution is obtained by substituting the heat-equation solution through the preceding transformed representations.The solution can also be formed using the heat kernel convolved with the initial condition.
  • Visualization: The visualized solution uses u0(−∞) = 1, u0(∞) = 0, and ν = 0.5, producing a steadily propagating wave.The stated initial condition and viscosity define the solution shown in Figure 4.
  • Solver context: PDE solvers are categorized into numerical-analysis and neural-based methods, with constraints represented as soft penalties or hard requirements.The solver summary distinguishes data-driven, hybrid, mesh-free, and CNF approaches by training and constraint characteristics.

B.7.1 FINITE DIFFERENCE METHOD (FDM)

Finite differences discretize PDE derivatives on sampled space-time grids, while neural solvers minimize data, boundary, and residual losses. Operator-learning variants map parameterized PDE inputs to solutions but face sampling and generalization challenges.

  • Finite-difference discretization: Finite difference operators approximate spatial and temporal derivatives on an N×M discretized sequence of u(x,t).N and M denote sampled spatial and temporal points, respectively.
  • Finite-difference discretization: The upwind scheme selects directional differences according to wave propagation, using neighboring values based on the sign of β.When β>0, the wave propagates rightward and u_i^j is influenced by u_{i−1}^j.
  • Stability: Explicit forward Euler schemes require stability conditions, whereas implicit schemes such as backward Euler are unconditionally stable.The CFL condition limits the allowable time step for a given spatial spacing.
  • Physics-informed neural networks: PINNs approximate unknown PDE solutions with neural networks and optimize supervised, residual, and boundary-condition losses.The PDE residual is computed through automatic differentiation and minimized during training.
  • Operator learning: Basic operator learning minimizes supervised error to learn a map from parameterized PDE inputs to solution functions, while PINO additionally combines supervised and residual losses.The operator-learning formulation connects input and solution function spaces through G(a,f)=u.
  • Operator learning: Parameterization and grid interpolation can suffer from sampling, non-uniqueness, poor generalization, or inconsistency between discrete and continuous functions.A continuous-function generalization is presented as another solution strategy.

C ARCHITECTURE DETAILS

The implementations use a multi-scale feed-forward architecture for PINNs and a spectral-convolution architecture for FNOs, with experiments run on a single-core Mac M1 CPU.

  • PINN architecture: The PINN architecture has three hidden layers with 64 neurons and multiple-frequency input encoding layers.The multi-scale design augments the original feed-forward architecture.
  • FNO architecture: The FNO retains 16 spectral modes and uses a latent feature dimension of 64.These settings define the adopted spectral-convolution configuration.
  • Experimental setup: Measurements use a Mac M1 with a single-core 3.2 GHz CPU, and data points are uniformly sampled unless otherwise specified.The hardware and sampling policy apply to the reported measurements.

E BURGERS’ EQUATION STABILITY EXPERIMENT

The Burgers’ equation experiment evaluates forward-Euler stability under the CFL condition and identifies forward Euler as the only unstable time-discretization scheme among four tested schemes.

  • Stability result: Only forward Euler is unstable when the time step exceeds the stability limit under the CFL condition.The experiment is reported in Table 15 for the forward Euler Kansa method on Burgers’ equation.
  • Interpretation: The result illustrates why explicit forward Euler requires a time-step restriction to avoid numerical instability.Implicit schemes do not share this stated conditional-stability requirement.
  • Stability result: The tested stability cases use CFL scale values of 1 and 2.These cases are reported as following the CFL condition.

F.1 FUNCTIONAL ANALYSIS

The paper introduces functional-analysis concepts for its PDE-solvers discussion and later learning-theory treatment, with additional learning-theory material referenced in Appendix §F.2.

  • Scope: Functional-analysis concepts are introduced for PDE solvers and later learning theory.The concepts support both parts of the paper’s technical development.
  • Scope: The functional-analysis discussion is connected explicitly to the paper’s later learning-theory material.This connection is stated as part of the section’s purpose.
  • Scope: Additional concepts used for learning theory are referenced in Appendix §F.2.The appendix provides further material beyond the introductory discussion.

F.2 FUNCTIONAL ANALYSIS ADDENDUM

The addendum formalizes function smoothness, Sobolev regularity, neural-network approximation, and error decomposition for PDE solvers. Its PINN analysis links residual, quadrature, training, and total errors under stated regularity and coercivity conditions.

  • Function spaces: Smoothness counts continuous derivatives, while Sobolev spaces extend this framework to functions with weak, p-integrable derivatives.C^k requires continuous derivatives through order k; W^k,p requires weak derivatives through order k in L^p(Ω).
  • Approximation theory: For fixed approximation error ϵ, neural-network parameter requirements scale as M = O(ϵ^-d/s), exposing exponential dependence on input dimension d.Greater smoothness s improves approximation, but the dimensional dependence produces the curse of dimensionality.
  • Error analysis: The total neural-network approximation error separates into approximation, optimization, and training components associated with network expressivity, optimization, and quadrature.Training error measures the difference between integral and discretized risks caused by quadrature.
  • PINN learning theory: PINN analysis reduces total-error control to PDE residual control through a coercivity condition, while quadrature bounds connect training error to residual error.The argument assumes the relevant constants are finite and uses sufficient quadrature points and smooth activations to make residual error small.
  • PINN learning theory: The resulting bound is EPINN(θ) < CC′[ϵ + o(1/NR)], with the extra quadrature term converging faster than 1/NR as NR increases.Thus, under the stated assumptions, total error converges as training error converges.
  • PINN learning theory: Practical failure modes include too few quadrature points, insufficient training, violated coercivity, and large constants in the error bounds.These respectively enlarge training or optimization error, invalidate the residual-to-total-error link, or weaken the resulting bound.
Loading 2602.07970v3…