the.ai

Attention / Behaviour

verified

Attention Sink

Trained transformers dump a surprising amount of attention on the very first token, regardless of what it is. The softmax has to put its weight somewhere even when a position has nothing it needs, and the first token becomes the place it goes. Evict it from the cache and quality collapses, for a token carrying no information.

Viz primitive · attention-heatmaptokens = 8 · heads = 1
query ↓key →

8 queries against 8 keys; a brighter cell means more of that query's attention went to that key. Each query sees itself and everything before it, and nothing after — the upper triangle is masked. The first key takes 95% of the average row on its own.

Attention weights with a bright first column every row falls back on. Drag the length to see the sink persist however far the sequence runs.

8

Reviewed by opendroid · 2026-08-04