Skip to yearly menu bar Skip to main content


Poster

Deep Equilibrium Algorithmic Reasoning

Dobrik Georgiev · Joseph Wilson · Davide Buffelli · Pietro Lió


Abstract:

Neural algorithmic reasoning (NAR) research has demonstrated that graph neural networks (GNNs) could learn to execute classical algorithms. However, most previous approaches have always used a recurrent architecture, where each iteration of the GNN aligns with an iteration of the algorithm. In this paper we study neurally solving algorithms from a different perspective: since the algorithm’s solution is often an equilibrium, it is possible to find the solution directly by solving an equilibrium equation. This both improves the alignment of GNNs to algorithms and is also a step towards speeding up NAR models. Our empirical evidence, leveraging algorithms from the CLRS-30 benchmark validates that one can train a network to solve algorithmic problems by directly finding the equilibrium. We discuss the practical implementation of such models and propose regularisations to improve the performance of these equilibrium reasoners.

Live content is unavailable. Log in and register to view live content