Characterizing the Expressivity of Local Attention in Transformers
For NLP researchers, this work offers a theoretical justification for the empirical success of local attention, revealing its complementary expressivity to global attention.
The paper provides a formal explanation for why local attention improves transformer quality, proving that it adds a second temporal operator to the expressible fragment of linear temporal logic, strictly enlarging the class of recognizable regular languages. Experiments on formal language recognition and natural language modeling show hybrid global-local transformers outperform global-only counterparts.
The transformer is the most popular neural architecture for language modeling. The cornerstone of the transformer is its global attention mechanism, which lets the model aggregate information from all preceding tokens before generating the next token. One common variant of attention is called local attention, which restricts each token to aggregating information from a bounded window of predecessors, reducing the quadratic cost of global attention to linear. Although this restriction is usually motivated by efficiency, it has also been found to improve model quality, a phenomenon that has so far lacked a satisfactory explanation. We provide a formal account of this phenomenon in terms of recognizer expressivity. It has been shown that fixed-precision transformers with global attention correspond to a fragment of linear temporal logic containing a single past operator. We additionally prove that adding local attention introduces a second temporal operator, strictly enlarging the class of recognizable regular languages. Moreover, global and local attention are expressively complementary: neither subsumes the other, and combining them yields the richest fragment. Experiments on formal language recognition and natural language modeling corroborate the theory, showing that hybrid global--local transformers outperform their global-only counterparts.