about
A Generalist Neural Algorithmic Learner (arxiv.org)
92 points by tosh on Dec 8, 2022 | hide | past | pdf | 15 comments on HN

In plain words: A graph network that works on connected points learns to run algorithms—sorting, searching, path-finding, geometry—at once, instead of a model for each. Tweaks to its inputs and training boosted single-task performance by over 20% versus earlier work, and the combined model matched the specialists.

Abstract

The cornerstone of neural algorithmic reasoning is the ability to solve algorithmic tasks, especially in a way that generalises out of distribution. While recent years have seen a surge in methodological improvements in this area, they mostly focused on building specialist models. Specialist models are capable of learning to neurally execute either only one algorithm or a collection of algorithms with identical control-flow backbone. Here, instead, we focus on constructing a generalist neural algorithmic learner -- a single graph neural network processor capable of learning to execute a wide range of algorithms, such as sorting, searching, dynamic programming, path-finding and geometry. We leverage the CLRS benchmark to empirically show that, much like recent successes in the domain of perception, generalist algorithmic learners can be built by "incorporating" knowledge. That is, it is possible to effectively learn algorithms in a multi-task manner, so long as we can learn to execute them well in a single-task regime. Motivated by this, we present a series of improvements to the input representation, training regime and processor architecture over CLRS, improving average single-task performance by over 20% from prior art. We then conduct a thorough ablation of multi-task learners leveraging these improvements. Our results demonstrate a generalist learner that effectively incorporates knowledge captured by specialist models.

Borja Ibarz, Vitaly Kurin, George Papamakarios, Kyriacos Nikiforou, Mehdi Bennani, Róbert Csordás, Andrew Dudzik, Matko Bošnjak, Alex Vitvitskyi, Yulia Rubanova, Andreea Deac, Beatrice Bevilacqua, et al.
arXiv:2209.11142 · cs.LG, cs.AI, stat.ML · submitted Sep 22, 2022 · updated Dec 3, 2022
abstract · pdf · html · To appear at LoG 2022 (Spotlight talk). 23 pages, 11 figures

add comment on HN
Also discussed: Sep 2022 (1 point, 0 comments)

This is from DeepMind. Twitter thread: https://twitter.com/DeepMind/status/1600852768125726720
Check https://logconference.org/schedule-tutorials/#neural-algorit... for a tutorial happening in less than 24 hours.

LoG conference is free to attend and online :)

I don't think this is THAT world changing. Generalist / multitask learning is already displayed in many models across orgs like Deepmind and others. This Graph NN model is performing algorithms that typically rely on graph theory. It's unsurprising that there's little friction and high success in combining embedding spaces on various graph algorithms by representing those in a graph. If this model was realizing special success on other domains it would seem more significant.
The interesting part would be if they could put this somewhere on the path of a large language model, so that it could learn to apply logic to its transformations instead of just symbolically manipulate things. Then maybe we could get a language model that can do math.
Ha well that _would_ be interesting. This work is interesting on its own BUT yeah it doesn't yet do what you mentioned and, to me, its conclusions are not that surprising.
The question is whether this is a better approach than just giving the LLM access to calculator / python-interpreter.
Or a little recurrent ALU side-chain glued on. It would at the very least put a stop to the "It can't even multiple two four-digit numbers correctly," squad which I admit, I can't do in my head either.
Whats the difference between math and a language? Its all just literals (numbers) and verbs relating them (+,-,*,/).
Large Language Models are bad at a lot of the same things humans are bad at. For example, humans can't add or multiply large numbers in their head. They need a piece of paper as an axillary memory. Or a calculator. If we need to find the shortest path in a graph, we write a short program to do it.

Likewise a Large Language Model can easily write the source code to do addition for arbitrarily large numbers, or solve graph related problems. (See AlphaCode.) It's not clear that they need to be "Generalist Algorithmic Learners" as this paper suggests. Time will tell.

It feels like there's an interesting release from the one of the major AI labs nearly every week now.

Is this a hype wave before the next AI winter, or are we accelerating towards something truly profound?

The hype cycle wants to end, but the pace of technological advance is breaking it.
there many hype like releases, with bold claims you can't verify, but very few cases when results are good enough to be applied on real world tasks.
Are some of these labs under layoff pressure?
This could be a piece of the AGI puzzle.

Learning how to combine/integrate various kinds of ANNs trained on vastly different domains is something that will have to be done at some point.

At least this seems to be a reasonable path to explore.

Man the pace of AI progress these days is bonkers.