Machine Learning: Science and Technology (Jan 2024)

Spectral-bias and kernel-task alignment in physically informed neural networks

  • Inbar Seroussi,
  • Asaf Miron,
  • Zohar Ringel

DOI
https://doi.org/10.1088/2632-2153/ad652d
Journal volume & issue
Vol. 5, no. 3
p. 035048

Abstract

Read online

Physically informed neural networks (PINNs) are a promising emerging method for solving differential equations. As in many other deep learning approaches, the choice of PINN design and training protocol requires careful craftsmanship. Here, we suggest a comprehensive theoretical framework that sheds light on this important problem. Leveraging an equivalence between infinitely over-parameterized neural networks and Gaussian process regression, we derive an integro-differential equation that governs PINN prediction in the large data-set limit—the neurally-informed equation. This equation augments the original one by a kernel term reflecting architecture choices. It allows quantifying implicit bias induced by the network via a spectral decomposition of the source term in the original differential equation.

Keywords