In plain words: Transformers trained on strings of abstract relations learn the rules and apply them to symbols never seen before, which is proved for a broad family of tasks. Plain fully-connected networks fail at this, and a two-parameter tweak helps transformers learn it with less data.
Abstract
We investigate the capabilities of transformer models on relational reasoning tasks. In these tasks, models are trained on a set of strings encoding abstract relations, and are then tested out-of-distribution on data that contains symbols that did not appear in the training dataset. We prove that for any relational reasoning task in a large family of tasks, transformers learn the abstract relations and generalize to the test set when trained by gradient descent on sufficiently large quantities of training data. This is in contrast to classical fully-connected networks, which we prove fail to learn to reason. Our results inspire modifications of the transformer architecture that add only two trainable parameters per head, and that we empirically demonstrate improve data efficiency for learning to reason.
Enric Boix-Adsera, Omid Saremi, Emmanuel Abbe, Samy Bengio, Etai Littwin, Joshua Susskind
arXiv:2310.09753 · cs.CL, cs.AI, cs.LG · submitted Oct 15, 2023 · updated Apr 16, 2024
abstract · pdf · html · 25 figures