16
8

One-layer transformers fail to solve the induction heads task

Abstract

A simple communication complexity argument proves that no one-layer transformer can solve the induction heads task unless its size is exponentially larger than the size sufficient for a two-layer transformer.

View on arXiv
Comments on this paper