Stiff Neural Ordinary Differential Equations
Not on the map. Why?
Abstract (arXiv)
Neural Ordinary Differential Equations (ODE) are a promising approach to learn dynamic models from time-series data in science and engineering applications. This work aims at learning Neural ODE for stiff systems, which are usually raised from chemical kinetic modeling in chemical and biological systems. We first show the challenges of learning neural ODE in the classical stiff ODE systems of Robertson's problem and propose techniques to mitigate the challenges associated with scale separations in stiff systems. We then present successful demonstrations in stiff systems of Robertson's problem and an air pollution problem. The demonstrations show that the usage of deep networks with rectified activations, proper scaling of the network outputs as well as loss functions, and stabilized gradient calculations are the key techniques enabling the learning of stiff neural ODE. The success of learning stiff neural ODE opens up possibilities of using neural ODEs in applications with widely varying time-scales, like chemical dynamics in energy conversion, environmental engineering, and the life sciences.
Key facts. Cited by 77 papers in the map (data through 7 Oct 2026). Builds on 14 of its 43 references that are in the map. First posted on arXiv 29 Mar 2021.
Builds on
14 of 43 references.
References · 14
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2019Deep Equilibrium Modelscited by 603
- 2017OptNet: Differentiable Optimization as a Layer in Neural Networkscited by 588
- 2020Universal Differential Equations for Scientific Machine Learningcited by 345
- 2020Understanding and mitigating gradient pathologies in physics-informed neural networkscited by 134
- 2019ANODE: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEscited by 83
- 2020STEER: Simple Temporal Regularization For Neural ODEscited by 58
- 2019DiffEqFlux.jl - A Julia Library for Neural Differential Equationscited by 33
- 2020Autonomous Discovery of Unknown Reaction Pathways from Data by Chemical Reaction Neural Networkcited by 27
Built on by
Citing papers per year: 2021: 7; 2022: 12; 2023: 12; 2024: 15; 2025: 19; 2026: 12.
77 papers.
Sort:
By area:
- 2024KAN-ODEs: Kolmogorov-Arnold Network Ordinary Differential Equations for Learning Dynamical Systems and Hidden Physicscited by 58
- 2023Effectively Modeling Time Series with Simple Discrete State Spacescited by 54
- 2021Neural Controlled Differential Equations for Online Prediction Taskscited by 46
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraintscited by 46
- 2022Deep Latent State Space Models for Time-Series Generationcited by 37
- 2023Learning stiff chemical kinetics using extended deep neural operatorscited by 30
- 2021DAE-PINN: A Physics-Informed Neural Network Model for Simulating Differential-Algebraic Equations with Application to Power Networkscited by 28
- 2022Neural Laplace: Learning diverse classes of differential equations in the Laplace domaincited by 24
- 2022Interpretable Polynomial Neural Ordinary Differential Equationscited by 24
- 2022Differentiable Programming for Earth System Modelingcited by 21
- 2025Comprehensive Review of Neural Differential Equations for Time Series Analysiscited by 20
- 2021DySMHO: Data-Driven Discovery of Governing Equations for Dynamical Systems via Moving Horizon Optimizationcited by 19
38 more
- 2022Parsimonious Physics-Informed Random Projection Neural Networks for Initial-Value Problems of ODEs and index-1 DAEscited by 19
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraintscited by 15
- 2022Residual-based physics-informed transfer learning: A hybrid method for accelerating long-term CFD simulations via deep learningcited by 14
- 2022Homotopy-based training of NeuralODEs for accurate dynamics discoverycited by 14
- 2024When are dynamical systems learned from time series data statistically accurate?cited by 14
- 2021Finite volume method network for acceleration of unsteady computational fluid dynamics: non-reacting and reacting flowscited by 12
- 2024Training Stiff Neural Ordinary Differential Equations with Implicit Single-Step Methodscited by 11
- 2022On the Forward Invariance of Neural ODEscited by 10
- 2024From Fourier to Neural ODEs: Flow Matching for Modeling Complex Systemscited by 9
- 2025DUE: A Deep Learning Framework and Library for Modeling Unknown Equationscited by 8
- 2024AdjointDEIS: Efficient Gradients for Diffusion Modelscited by 7
- 2024Learning Hamiltonian neural Koopman operator and simultaneously sustaining and discovering conservation lawcited by 7
- 2023Learning Energy Conserving Dynamics Efficiently with Hamiltonian Gaussian Processescited by 6
- 2023Bayesian polynomial neural networks and polynomial neural ordinary differential equationscited by 6
- 2024Learning Neural Differential Algebraic Equations via Operator Splittingcited by 6
- 2024Training Stiff Neural Ordinary Differential Equations with Explicit Exponential Integration Methodscited by 5
- 2025Mitigating mode collapse in normalizing flows by annealing with an adaptive schedule: Application to parameter estimationcited by 5
- 2023Locally Regularized Neural Differential Equations: Some Black Boxes Were Meant to Remain Closed!cited by 4
- 2025ChemKANs for Combustion Chemistry Modeling and Accelerationcited by 4
- 2022A memory-efficient neural ODE framework based on high-level adjoint differentiationcited by 3
- 2023Do We Need an Encoder-Decoder to Model Dynamical Systems on Networks?cited by 3
- 2023Tasks Makyth Models: Machine Learning Assisted Surrogates for Tipping Pointscited by 3
- 2025SPIN-ODE: Stiff Physics-Informed Neural ODE for Chemical Reaction Rate Estimationcited by 3
- 2025The Vanishing Gradient Problem for Stiff Neural Differential Equationscited by 3
- 2022A Neural Ordinary Differential Equation Model for Visualizing Deep Neural Network Behaviors in Multi-Parametric MRI based Glioma Segmentationcited by 2
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamicscited by 2
- 2024Semi-Implicit Neural Ordinary Differential Equationscited by 2
- 2025Greed is Good: A Unifying Perspective on Guided Generationcited by 2
- 2025Symbolic Neural Ordinary Differential Equationscited by 2
- 2025AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kineticscited by 2
- 2025Physics-Informed Neural ODEs with Scale-Aware Residuals for Learning Stiff Biophysical Dynamicscited by 2
- 2021Incorporating NODE with Pre-trained Neural Differential Operator for Learning Dynamicscited by 1
- 2021Numerical Solution of Stiff ODEs with Physics-Informed RPNNscited by 1
- 2021Abstract Interpretation of Fixpoint Iterators with Applications to Neural Networkscited by 1
- 2022Efficient Climate Simulation via Machine Learning Methodcited by 1
- 2023Jacobian-Scaled K-means Clustering for Physics-Informed Segmentation of Reacting Flowscited by 1
- 2023A Posteriori Evaluation of a Physics-Constrained Neural Ordinary Differential Equations Approach Coupled with CFD Solver for Modeling Stiff Chemical Kineticscited by 1
- 2024FlamePINN-1D: Physics-informed neural networks to solve forward and inverse problems of 1D laminar flamescited by 1
50 most cited of 77 All 77 (JSON)
- 2026Helix-FNO: Spectral-Domain Operator Learning Coupled with a High-Fidelity Mechanistic Model for Fast Surrogate Simulation23 Aug 2026
- 2026Synergizing Physically Constrained MCMC and Chemical-Informed Gaussian Processes for Reaction Network Discovery22 Jun 2026
- 2026Learning Hybrid Biophysical Neuron Models with Neural ODEs15 Jun 2026
- 2026Watch your neighbors: Training statistically accurate chaotic systems with local phase space information14 May 2026
- 2026Continuity Laws for Sequential Models8 May 2026
- 2026Model synthesis and identifiability analysis of stiff chemical reaction systems with inVAErt networks5 May 2026
- 2026Acceleration of horizontal numerical advection for atmospheric modeling through surrogate modeling with temporal coarse-graining30 Apr 2026
- 2026Trajectory-Optimized Time Reparameterization for Learning-Compatible Reduced-Order Modeling of Stiff Dynamical Systems17 Mar 2026
- 2026Inverse Neural Operator for ODE Parameter Optimization12 Mar 2026
- 2026A neural operator for predicting vibration frequency response curves from limited data10 Mar 2026
- 2026Solving stiff dark matter equations via Jacobian Normalization with Physics-Informed Neural Networks25 Feb 2026
- 2026From Prompt to Protocol: Fast Charging Batteries with Large Language Models14 Jan 2026
38 more
- 2025NORi: An ML-Augmented Ocean Boundary Layer Parameterization4 Dec 2025
- 2025Physics-Informed Neural ODEs with Scale-Aware Residuals for Learning Stiff Biophysical Dynamics13 Nov 2025
- 2025Learning solutions of parameterized stiff ODEs using Gaussian processes8 Nov 2025
- 2025Zero-Shot Function Encoder-Based Differentiable Predictive Control7 Nov 2025
- 2025AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kinetics15 Oct 2025
- 2025The Vanishing Gradient Problem for Stiff Neural Differential Equations2 Aug 2025
- 2025Fredholm Neural Networks for inverse problems in elliptic PDEs8 Jul 2025
- 2025Stable CDE Autoencoders with Acuity Regularization for Offline Reinforcement Learning in Sepsis Treatment17 Jun 2025
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraints4 Jun 2025
- 2025Automatic and Structure-Aware Sparsification of Hybrid Neural ODEs25 May 2025
- 2025SPIN-ODE: Stiff Physics-Informed Neural ODE for Chemical Reaction Rate Estimation8 May 2025
- 2025Mitigating mode collapse in normalizing flows by annealing with an adaptive schedule: Application to parameter estimation6 May 2025
- 2025ChemKANs for Combustion Chemistry Modeling and Acceleration17 Apr 2025
- 2025Predicting Wave Dynamics using Deep Learning with Multistep Integration Inspired Attention and Physics-Based Loss Decomposition15 Apr 2025
- 2025DUE: A Deep Learning Framework and Library for Modeling Unknown Equations14 Apr 2025
- 2025Symbolic Neural Ordinary Differential Equations11 Mar 2025
- 2025Comprehensive Review of Neural Differential Equations for Time Series Analysis14 Feb 2025
- 2025Greed is Good: A Unifying Perspective on Guided Generation11 Feb 2025
- 2025LiLaN: A Linear Latent Network as the Solution Operator for Real-Time Solutions to Stiff Nonlinear Ordinary Differential Equations14 Jan 2025
- 2024Semi-Implicit Neural Ordinary Differential Equations15 Dec 2024
- 2024Training Stiff Neural Ordinary Differential Equations with Explicit Exponential Integration Methods2 Dec 2024
- 2024When are dynamical systems learned from time series data statistically accurate?9 Nov 2024
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics3 Nov 2024
- 2024A comparative study of NeuralODE and Universal ODE approaches to solving Chandrasekhar White Dwarf equation19 Oct 2024
- 2024Modeling chaotic Lorenz ODE System using Scientific Machine Learning9 Oct 2024
- 2024Training Stiff Neural Ordinary Differential Equations with Implicit Single-Step Methods8 Oct 2024
- 2024Neural Differential Appearance Equations23 Sep 2024
- 2024KAN-ODEs: Kolmogorov-Arnold Network Ordinary Differential Equations for Learning Dynamical Systems and Hidden Physics5 Jul 2024
- 2024FlamePINN-1D: Physics-informed neural networks to solve forward and inverse problems of 1D laminar flames7 Jun 2024
- 2024Learning Hamiltonian neural Koopman operator and simultaneously sustaining and discovering conservation law4 Jun 2024
- 2024AdjointDEIS: Efficient Gradients for Diffusion Models23 May 2024
- 2024From Fourier to Neural ODEs: Flow Matching for Modeling Complex Systems19 May 2024
- 2024Forecasting with an N-dimensional Langevin Equation and a Neural-Ordinary Differential Equation12 May 2024
- 2024Learning Neural Differential Algebraic Equations via Operator Splitting19 Mar 2024
- 2023A Posteriori Evaluation of a Physics-Constrained Neural Ordinary Differential Equations Approach Coupled with CFD Solver for Modeling Stiff Chemical Kinetics22 Nov 2023
- 2023Time-vectorized numerical integration for systems of ODEs12 Oct 2023
- 2023Tasks Makyth Models: Machine Learning Assisted Surrogates for Tipping Points25 Sep 2023
- 2023Bayesian polynomial neural networks and polynomial neural ordinary differential equations17 Aug 2023
50 newest of 77 All 77 (JSON)
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraintscited by 46
- 2024From Fourier to Neural ODEs: Flow Matching for Modeling Complex Systemscited by 9
- 2024AdjointDEIS: Efficient Gradients for Diffusion Modelscited by 7
- 2025Greed is Good: A Unifying Perspective on Guided Generationcited by 2
- 2025Physics-Constrained Flow Matching: Sampling Generative Models with Hard Constraints4 Jun 2025
- 2025Greed is Good: A Unifying Perspective on Guided Generation11 Feb 2025
- 2024AdjointDEIS: Efficient Gradients for Diffusion Models23 May 2024
- 2024From Fourier to Neural ODEs: Flow Matching for Modeling Complex Systems19 May 2024
- 2022Deep Latent State Space Models for Time-Series Generationcited by 37
- 2022Deep Latent State Space Models for Time-Series Generation24 Dec 2022
- 2024KAN-ODEs: Kolmogorov-Arnold Network Ordinary Differential Equations for Learning Dynamical Systems and Hidden Physicscited by 58
- 2023Effectively Modeling Time Series with Simple Discrete State Spacescited by 54
- 2021Neural Controlled Differential Equations for Online Prediction Taskscited by 46
- 2023Learning stiff chemical kinetics using extended deep neural operatorscited by 30
- 2021DAE-PINN: A Physics-Informed Neural Network Model for Simulating Differential-Algebraic Equations with Application to Power Networkscited by 28
- 2022Neural Laplace: Learning diverse classes of differential equations in the Laplace domaincited by 24
- 2022Interpretable Polynomial Neural Ordinary Differential Equationscited by 24
- 2022Differentiable Programming for Earth System Modelingcited by 21
- 2025Comprehensive Review of Neural Differential Equations for Time Series Analysiscited by 20
- 2021DySMHO: Data-Driven Discovery of Governing Equations for Dynamical Systems via Moving Horizon Optimizationcited by 19
- 2022Parsimonious Physics-Informed Random Projection Neural Networks for Initial-Value Problems of ODEs and index-1 DAEscited by 19
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraintscited by 15
12 most cited of 71 All 71 (JSON)
- 2026Helix-FNO: Spectral-Domain Operator Learning Coupled with a High-Fidelity Mechanistic Model for Fast Surrogate Simulation23 Aug 2026
- 2026Synergizing Physically Constrained MCMC and Chemical-Informed Gaussian Processes for Reaction Network Discovery22 Jun 2026
- 2026Learning Hybrid Biophysical Neuron Models with Neural ODEs15 Jun 2026
- 2026Watch your neighbors: Training statistically accurate chaotic systems with local phase space information14 May 2026
- 2026Continuity Laws for Sequential Models8 May 2026
- 2026Model synthesis and identifiability analysis of stiff chemical reaction systems with inVAErt networks5 May 2026
- 2026Acceleration of horizontal numerical advection for atmospheric modeling through surrogate modeling with temporal coarse-graining30 Apr 2026
- 2026Trajectory-Optimized Time Reparameterization for Learning-Compatible Reduced-Order Modeling of Stiff Dynamical Systems17 Mar 2026
- 2026Inverse Neural Operator for ODE Parameter Optimization12 Mar 2026
- 2026A neural operator for predicting vibration frequency response curves from limited data10 Mar 2026
- 2026Solving stiff dark matter equations via Jacobian Normalization with Physics-Informed Neural Networks25 Feb 2026
- 2026From Prompt to Protocol: Fast Charging Batteries with Large Language Models14 Jan 2026
12 newest of 71 All 71 (JSON)