TasksSotADatasetsPapersMethodsSubmitAbout
Papers With Code 2

A community resource for machine learning research: papers, code, benchmarks, and state-of-the-art results.

Explore

Notable BenchmarksAll SotADatasetsPapersMethods

Community

Submit ResultsAbout

Data sourced from the PWC Archive (CC-BY-SA 4.0). Built by the community, for the community.

Papers/ANODE: Unconditionally Accurate Memory-Efficient Gradients...

ANODE: Unconditionally Accurate Memory-Efficient Gradients for Neural ODEs

Amir Gholami, Kurt Keutzer, George Biros

2019-02-27Multivariate Time Series Imputation
PaperPDFCodeCodeCodeCodeCode

Abstract

Residual neural networks can be viewed as the forward Euler discretization of an Ordinary Differential Equation (ODE) with a unit time step. This has recently motivated researchers to explore other discretization approaches and train ODE based networks. However, an important challenge of neural ODEs is their prohibitive memory cost during gradient backpropogation. Recently a method proposed in [8], claimed that this memory overhead can be reduced from O(LN_t), where N_t is the number of time steps, down to O(L) by solving forward ODE backwards in time, where L is the depth of the network. However, we will show that this approach may lead to several problems: (i) it may be numerically unstable for ReLU/non-ReLU activations and general convolution operators, and (ii) the proposed optimize-then-discretize approach may lead to divergent training due to inconsistent gradients for small time step sizes. We discuss the underlying problems, and to address them we propose ANODE, an Adjoint based Neural ODE framework which avoids the numerical instability related problems noted above, and provides unconditionally accurate gradients. ANODE has a memory footprint of O(L) + O(N_t), with the same computational cost as reversing ODE solve. We furthermore, discuss a memory efficient algorithm which can further reduce this footprint with a trade-off of additional computational cost. We show results on Cifar-10/100 datasets using ResNet and SqueezeNext neural networks.

Related Papers

CoSTI: Consistency Models for (a faster) Spatio-Temporal Imputation2025-01-31BRATI: Bidirectional Recurrent Attention for Time-Series Imputation2025-01-09SDA-GRIN for Adaptive Spatial-Temporal Multivariate Time Series Imputation2024-10-04Mining of Switching Sparse Networks for Missing Value Imputation in Multivariate Time Series2024-09-16Latent Space Score-based Diffusion Model for Probabilistic Multivariate Time Series Imputation2024-09-13MTSCI: A Conditional Diffusion Model for Multivariate Time Series Consistent Imputation2024-08-11REGIONAL DATASETS FOR AIR QUALITY MONITORING IN EUROPEAN CITIES2024-07-12Score-CDM: Score-Weighted Convolutional Diffusion Model for Multivariate Time Series Imputation2024-05-21