@inproceedings{
zhang2023semiimplicit,
title={Semi-Implicit Neural Ordinary Differential Equations for Learning Chaotic Systems},
author={Hong Zhang and Ying Liu and Romit Maulik},
booktitle={NeurIPS 2023 Workshop Heavy Tails in Machine Learning},
year={2023},
url={https://openreview.net/forum?id=wI4SXLFqVK}
}
Classical neural ordinary differential equations (ODEs) trained by using explicit methods are intrinsically constrained by stability,
severely affecting their efficiency and robustness in learning complex spatiotemporal dynamics, particularly those displaying chaotic behavior. 
In this work we propose a semi-implicit neural ODE approach that capitalizes on the partitionable structure of the underlying dynamics.
In our method the neural ODE is partitioned into a linear part treated implicitly for enhanced stability and a nonlinear part treated explicitly.
We apply this approach to learn chaotic trajectories of the Kuramoto--Sivashinsky equation.
Our results demonstrate that our approach significantly outperforms existing approaches for coarse-resolution data and remains efficient for fine-resolution data where existing techniques become intractable.
Semi-Implicit Neural Ordinary Differential Equations for Learning Chaotic Systems
Hong Zhang Ying Liu Romit Maulik