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 arXivComments on this paper