Mechanisms underlying attention sinks in alternative Transformer architectures
Investigate which mechanisms form attention sinks in Transformer architectures that omit learned query biases or use alternative positional schemes such as RoPE or ALiBi.
References
The broader Transformer ecosystem includes architectures that omit such biases or use alternative positional schemes (e.g., RoPE, ALiBi). While we do show that the specific components constituting the GPT-2 circuit are absent or altered in those settings, we leave the investigation of what mechanisms do form in those models to future work.
Our interventions reduce the sink substantially but do not eliminate it entirely: nullifying $b_Q$ or zeroing the massive-activation rows of $W_k$ leaves a residual sink. This indicates that secondary contributors beyond the three main components we isolate also play a role. Identifying them is a natural extension of our work which we leave to future work.
We do not claim a definitive functional rationale for why such a sink is beneficial or harmful across tasks. Establishing the downstream utility or cost of the sink, and the conditions under which it is selected by optimization, is left for future work.