Regularized dynamical parametric approximation of stiff evolution problems
Christian Lubich, Jörg Nick
TL;DR
This work addresses the challenge of evolving irregular nonlinear parametrizations, such as neural networks, in stiff evolution problems by developing regularized parametric time integrators. It introduces a regulated Dirac–Frenkel–type approach where the parameter time derivatives are obtained from a regularized nonlinear least-squares problem, and the time stepping uses implicit Euler, implicit midpoint, and Runge--Kutta methods solved via Gauss--Newton iterations. The authors establish error bounds linking the parametric iterates to the nonparametric Newton scheme, derive global error estimates that are robust to stiffness, and extend the framework to higher-order RK methods like Radau IIA and Gauß. Numerical experiments on transport and heat equations illustrate defect decay and long-time stability, while implementation details highlight practical considerations for quadrature, initialization, and adaptive regularization. Overall, the paper provides a principled, stiffness-robust methodology for simulating high-dimensional, irregular parametrizations in evolution equations with provable error control and practical algorithmic guidance.
Abstract
Evolutionary deep neural networks have emerged as a rapidly growing field of research. This paper studies numerical integrators for such and other classes of nonlinear parametrizations $ u(t) = Φ(θ(t)) $, where the evolving parameters $θ(t)$ are to be computed. The primary focus is on tackling the challenges posed by the combination of stiff evolution problems and irregular parametrizations, which typically arise with neural networks, tensor networks, flocks of evolving Gaussians, and in further cases of overparametrization. We propose and analyse regularized parametric versions of the implicit Euler method and higher-order implicit Runge--Kutta methods for the time integration of the parameters in nonlinear approximations to evolutionary partial differential equations and large systems of stiff ordinary differential equations. At each time step, an ill-conditioned nonlinear optimization problem is solved approximately with a few regularized Gauss--Newton iterations. Error bounds for the resulting parametric integrator are derived by relating the computationally accessible Gauss--Newton iteration for the parameters to the computationally inaccessible Newton iteration for the underlying non-parametric time integration scheme. The theoretical findings are supported by numerical experiments that are designed to show key properties of the proposed parametric integrators.
