On the Abilities of Mathematical Extrapolation with Implicit Models

Juliette Decugis, Alicia Y. Tsai, Ashwin Ganesh, Max Emerling, Laurent El Ghaoui

Keywords: implicit deep learning models, OOD extrapolation

TL;DR: Show implicit models' unique advantages for sequence extrapolation due to their flexible and selective framework

Abstract: Deep neural networks excel on a variety of different tasks, often surpassing human intelligence. However, when presented with out-of-distribution data, these models tend to break down even on the simplest tasks. In this paper, we compare robustness in sequence modeling of implicitly-defined and classical deep learning models on a series of extrapolation tasks where the models are tested with out-of-distribution samples during inference time. Throughout our experiments, implicit models greatly outperform classical deep learning networks that overfit the training distribution. We showcase implicit models' unique advantages for sequence extrapolation thanks to their flexible and selective framework. Implicit models, with potentially unlimited depth, not only adapt well to out-of-distribution inputs but also understand the underlying structure of inputs much better.