Neural Flows: Efficient Alternative to Neural ODEs
About
Neural ordinary differential equations describe how values change in time. This is the reason why they gained importance in modeling sequential data, especially when the observations are made at irregular intervals. In this paper we propose an alternative by directly modeling the solution curves - the flow of an ODE - with a neural network. This immediately eliminates the need for expensive numerical solvers while still maintaining the modeling capability of neural ODEs. We propose several flow architectures suitable for different applications by establishing precise conditions on when a function defines a valid flow. Apart from computational efficiency, we also provide empirical evidence of favorable generalization performance via applications in time series modeling, forecasting, and density estimation.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Reconstruction | MuJoCo (test) | MSE4.217 | 51 | |
| Forecasting | MIMIC-III (test) | MSE0.477 | 43 | |
| Classification | Activity | Accuracy78.3 | 34 | |
| Irregularly Sampled Time Series Forecasting | Physionet 12 (test) | MSE0.326 | 28 | |
| Irregularly Sampled Time Series Forecasting | USHCN (test) | MSE0.414 | 26 | |
| Temporal Point Process modeling | MOOC real-world (test) | NLL-1.2379 | 25 | |
| Temporal Point Process modeling | Reddit real-world (test) | Negative Log-Likelihood-1.2962 | 25 | |
| Irregularly Sampled Time Series Forecasting | MIMIC-IV (test) | MSE0.354 | 25 | |
| Classification | PhysioNet | AUC Score0.788 | 23 | |
| Temporal Point Process modeling | Wiki real-world (test) | Negative Log-Likelihood-1.2907 | 18 |