Augmented Neural ODEs

Emilien Dupont, Arnaud Doucet, Yee Whye Teh

Abstract: We show that Neural Ordinary Differential Equations (ODEs) learn representations that preserve the topology of the input space and prove that this implies the existence of functions Neural ODEs cannot represent. To address these limitations, we introduce Augmented Neural ODEs which, in addition to being more expressive models, are empirically more stable, generalize better and have a lower computational cost than Neural ODEs.

Code Link: https://github.com/EmilienDupont/augmented-neural-odes

CMT Num: 1771