Proceedings

Taylor-Lagrange Neural Ordinary Differential Equations: Toward Fast Training and Evaluation of Neural ODEs

Video #1 Video #2 Download BibTeX Cite

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence. Main Track. Pages 2923-2929.

https://doi.org/10.24963/ijcai.2022/405

Abstract

Neural ordinary differential equations (NODEs) -- parametrizations of differential equations using neural networks -- have shown tremendous promise in learning models of unknown continuous-time dynamical systems from data. However, every forward evaluation of a NODE requires numerical integration of the neural network used to capture the system dynamics, making their training prohibitively expensive. Existing works rely on off-the-shelf adaptive step-size numerical integration schemes, which often require an excessive number of evaluations of the underlying dynamics network to obtain sufficient accuracy for training. By contrast, we accelerate the evaluation and the training of NODEs by proposing a data-driven approach to their numerical integration. The proposed Taylor-Lagrange NODEs (TL-NODEs) use a fixed-order Taylor expansion for numerical integration, while also learning to estimate the expansion's approximation error. As a result, the proposed approach achieves the same accuracy as adaptive step-size schemes while employing only low-order Taylor expansions, thus greatly reducing the computational cost necessary to integrate the NODE. A suite of numerical experiments, including modeling dynamical systems, image classification, and density estimation, demonstrate that TL-NODEs can be trained more than an order of magnitude faster than state-of-the-art approaches, without any loss in performance.

  1. [Bettencourt et al., 2019] J. Bettencourt, Matthew J. John- son, and D. Duvenaud. Taylor-mode automatic differen- tiation for higher-order derivatives in jax. In Workshop on Program Transformations for ML, NeurIPS, 2019.
  2. [Chen et al., 2018] Ricky TQ Chen, Yulia Rubanova, Jesse Bettencourt, and David K Duvenaud. Neural ordinary dif- ferential equations. Advances in Neural Information Pro- cessing Systems, 2018.
  3. [Cranmer et al., 2020] M. Cranmer, Sam Greydanus, Stephan Hoyer, Peter W. Battaglia, David N. Spergel, and Shirley Ho. Lagrangian neural networks. ArXiv, abs/2003.04630, 2020.
  4. [Deng, 2012] Li Deng. The mnist database of handwritten digit images for machine learning research. IEEE Signal Processing Magazine, 2012.
  5. [Djeumou et al., 2021] Franck Djeumou, Abraham P Vinod, Eric Goubault, Sylvie Putot, and Ufuk Topcu. On-the-fly control of unknown smooth systems from limited data. In 2021 American Control Conference. IEEE, 2021.
  6. [Djeumou et al., 2022a] Franck Djeumou, Cyrus Neary, Eric Goubault, Sylvie Putot, and Ufuk Topcu. Neural net- works with physics-informed architectures and constraints for dynamical systems modeling. In Learning for Dynam- ics and Control. PMLR, 2022.
  7. [Djeumou et al., 2022b] Franck Djeumou, Cyrus Neary, Eric Goubault, Sylvie Putot, and Ufuk Topcu. Taylor-lagrange neural ordinary differential equations: Toward fast train- ing and evaluation of neural odes. ArXiv, abs/2201.05715, 2022.
  8. [Finlay et al., 2020] Chris Finlay, J. Jacobsen, L. Nurbekyan, and Adam M. Oberman. How to train your neural ode: the world of jacobian and kinetic regularization. In International Conference on Machine Learning. PMLR, 2020.
  9. [Finzi et al., 2020] Marc Finzi, Ke Alexander Wang, and Andrew G Wilson. Simplifying hamiltonian and la- grangian neural networks via explicit constraints. Ad- vances in Neural Information Processing Systems, 2020.
  10. [Ghosh et al., 2020] Arna Ghosh, Harkirat Singh Behl, Em- ilien Dupont, Philip H. S. Torr, and Vinay Namboodiri. Steer : Simple temporal regularization for neural odes. ArXiv, abs/2006.10711, 2020.
  11. [Grathwohl et al., 2019] Will Grathwohl, Ricky T. Q. Chen, Jesse Bettencourt, Ilya Sutskever, and David Kristjan- son Duvenaud. Ffjord: Free-form continuous dynam- ics for scalable reversible generative models. ArXiv, abs/1810.01367, 2019.
  12. [Greydanus et al., 2019] Samuel Greydanus, Misko Dzamba, and Jason Yosinski. Hamiltonian neural networks. Advances in Neural Information Processing Systems, 2019.
  13. [Griewank and Walther, 2008] Andreas Griewank and An- drea Walther. Evaluating derivatives: principles and tech- niques of algorithmic differentiation. SIAM, 2008.
  14. [Gupta et al., 2020] Jayesh K Gupta, Kunal Menda, Zachary Manchester, and Mykel Kochenderfer. Structured mechan- ical models for robot learning and control. In Learning for Dynamics and Control. PMLR, 2020.
  15. [Kelly et al., 2020] Jacob Kelly, Jesse Bettencourt, Matthew J Johnson, and David K Duvenaud. Learn- ing differential equations that are easy to solve. Advances in Neural Information Processing Systems, 2020.
  16. [Mathieu and Nickel, 2020] Emile Mathieu and Maximilian Nickel. Riemannian continuous normalizing flows. ArXiv, abs/2006.10605, 2020.
  17. [Menda et al., 2019] Kunal Menda, Jayesh K Gupta, Zachary Manchester, and Mykel J Kochenderfer. Struc- tured mechanical models for efficient reinforcement learning. In Workshop on Structure and Priors in Reinforcement Learning, International Conference on Learning Representations, 2019.
  18. [Pal et al., 2021] Avik Pal, Yingbo Ma, Viral Shah, and Christopher V Rackauckas. Opening the blackbox: Ac- celerating neural differential equations by regularizing in- ternal solver heuristics. In International Conference on Machine Learning. PMLR, 2021.
  19. [Papamakarios et al., 2017] George Papamakarios, Iain Murray, and Theo Pavlakou. Masked autoregressive flow for density estimation. ArXiv, abs/1705.07057, 2017.
  20. [Poli et al., 2020] Michael Poli, Stefano Massaroli, Atsushi Yamashita, Hajime Asama, and Jinkyoo Park. Hyper- solvers: Toward fast continuous-depth models. ArXiv, abs/2007.09601, 2020.
  21. [Roe et al., 2005] Byron P Roe, Hai-Jun Yang, Ji Zhu, Yong Liu, Ion Stancu, and Gordon McGregor. Boosted decision trees as an alternative to artificial neural networks for par- ticle identification. Nuclear Instruments and Methods in Physics Research Section A: Accelerators, Spectrometers, Detectors and Associated Equipment, 2005.
  22. [Salman et al., 2018] Hadi Salman, Payman Yadollahpour, Tom Fletcher, and Nematollah Batmanghelich. Deep dif- feomorphic normalizing flows. ArXiv, abs/1810.03256, 2018.
  23. [Zhong et al., 2021] Yaofeng Desmond Zhong, Biswadip Dey, and Amit Chakraborty. Benchmarking energy- conserving neural networks for learning dynamics from data. In Learning for Dynamics and Control. PMLR, 2021.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards