Training Stiff Neural Ordinary Differential Equations with Explicit Exponential Integration Methods

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Fronk, Colby, Petzold, Linda
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929610434805760
author Fronk, Colby
Petzold, Linda
author_facet Fronk, Colby
Petzold, Linda
contents Stiff ordinary differential equations (ODEs) are common in many science and engineering fields, but standard neural ODE approaches struggle to accurately learn these stiff systems, posing a significant barrier to widespread adoption of neural ODEs. In our earlier work, we addressed this challenge by utilizing single-step implicit methods for solving stiff neural ODEs. While effective, these implicit methods are computationally costly and can be complex to implement. This paper expands on our earlier work by exploring explicit exponential integration methods as a more efficient alternative. We evaluate the potential of these explicit methods to handle stiff dynamics in neural ODEs, aiming to enhance their applicability to a broader range of scientific and engineering problems. We found the integrating factor Euler (IF Euler) method to excel in stability and efficiency. While implicit schemes failed to train the stiff Van der Pol oscillator, the IF Euler method succeeded, even with large step sizes. However, IF Euler's first-order accuracy limits its use, leaving the development of higher-order methods for stiff neural ODEs an open research problem.
format Preprint
id arxiv_https___arxiv_org_abs_2412_01181
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training Stiff Neural Ordinary Differential Equations with Explicit Exponential Integration Methods
Fronk, Colby
Petzold, Linda
Numerical Analysis
Artificial Intelligence
Machine Learning
Symbolic Computation
Stiff ordinary differential equations (ODEs) are common in many science and engineering fields, but standard neural ODE approaches struggle to accurately learn these stiff systems, posing a significant barrier to widespread adoption of neural ODEs. In our earlier work, we addressed this challenge by utilizing single-step implicit methods for solving stiff neural ODEs. While effective, these implicit methods are computationally costly and can be complex to implement. This paper expands on our earlier work by exploring explicit exponential integration methods as a more efficient alternative. We evaluate the potential of these explicit methods to handle stiff dynamics in neural ODEs, aiming to enhance their applicability to a broader range of scientific and engineering problems. We found the integrating factor Euler (IF Euler) method to excel in stability and efficiency. While implicit schemes failed to train the stiff Van der Pol oscillator, the IF Euler method succeeded, even with large step sizes. However, IF Euler's first-order accuracy limits its use, leaving the development of higher-order methods for stiff neural ODEs an open research problem.
title Training Stiff Neural Ordinary Differential Equations with Explicit Exponential Integration Methods
topic Numerical Analysis
Artificial Intelligence
Machine Learning
Symbolic Computation
url https://arxiv.org/abs/2412.01181