On the Existence and Behavior of Secondary Attention Sinks
This paper identifies and characterizes "secondary attention sinks" as distinct, transient phenomena emerging in middle layers of transformer models due to specific MLP-induced vector alignments, revealing their deterministic patterns across model scales and their dynamic interplay with weakening primary sinks.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
Imagine a large language model (like the AI behind this chat) as a massive, multi-story library where information travels from the ground floor to the top floor. As a story or a math problem moves through this library, it passes through many "rooms" (layers), and in each room, a team of librarians (the attention mechanism) decides which parts of the story are most important to remember.
For a while, researchers knew about one specific librarian who always got a lot of attention: the BOS Token (the "Start of Sequence" token). Think of this as the Librarian at the Front Desk. No matter what story you tell, the Front Desk Librarian always gets a huge spotlight, even if the story is about something completely unrelated to the front desk. This is called an "Attention Sink." It's like a black hole that sucks up attention, and it helps the library run smoothly.
But this new paper, accepted to a top AI conference, discovered something surprising: There are other librarians who also get sucked into the spotlight, but they are very different.
Here is the breakdown of their discovery using simple analogies:
1. The "Secondary Sinks": The Pop-Up Stars
The authors found a new type of attention sink they call Secondary Sinks.
- The Old Way (Primary Sinks): The Front Desk Librarian (BOS) is always there from the very first room to the very last. They are the permanent star.
- The New Way (Secondary Sinks): Imagine that halfway up the building, in the middle floors, a random, boring-looking book (a token with no real meaning, like a comma or a space) suddenly gets a massive spotlight. It wasn't important at the start, but suddenly, in the middle of the story, the librarians start staring at it intensely.
- The Difference: These "Secondary Sinks" don't stay forever. They appear in the middle, shine brightly for a few floors, and then fade away. They are like pop-up stars that appear only during specific parts of the story.
2. How They Are Made: The "Transformer" Machine
How does a boring token suddenly become a star? The paper found the "factory" that makes them.
- The Factory: It happens in a specific machine in the middle of the building called an MLP (a type of processing unit).
- The Process: Imagine a conveyor belt carrying many different items (tokens). Most items stay normal. But there is a specific machine in the middle that grabs a few random items and paints them with a special "glow" that makes them look exactly like the Front Desk Librarian.
- The Result: Once these items pass through this machine, they align perfectly with the Front Desk Librarian's direction. They become "fake Front Desks" for a little while.
3. The "Sink Levels": A Hierarchy of Fame
The researchers noticed these pop-up stars aren't random; they follow a pattern.
- Level 1: The permanent Front Desk Librarian (BOS).
- Level 2, 3, etc.: Different groups of "pop-up stars" that appear at different heights in the building.
- The Rule: In bigger, smarter models (like the 32B or 14B parameter models mentioned), these levels are very organized. It's like a well-rehearsed play where specific actors know exactly when to step into the spotlight and when to leave. In smaller models, this doesn't happen as much.
4. The "Compensation" Effect: Why Do They Exist?
Why would the library create these temporary stars?
- The Theory: The permanent Front Desk Librarian (BOS) gets tired. As the story gets longer and more complex, the Front Desk Librarian's attention starts to fade or get "diluted" in the middle of the building.
- The Fix: The Secondary Sinks appear right when the Front Desk Librarian gets weak. They act as substitutes. They take over the job of holding the attention steady so the story doesn't fall apart. It's like a backup singer stepping in when the lead singer's voice gets a little raspy, ensuring the song keeps going smoothly.
5. The "Strength" of the Star
The paper also found that the "brightness" of these pop-up stars depends on how hard the factory machine works.
- If the machine pushes the token really hard (high mathematical "norm"), the star shines brighter and stays in the spotlight for more floors.
- If the machine is gentle, the star is dimmer and disappears quickly.
Why Does This Matter?
Understanding these "Secondary Sinks" is like finding the hidden wiring in a house.
- Better AI: If we know how these temporary stars work, we can build better AI that doesn't get confused by long stories or complex math problems.
- Efficiency: We might be able to save computer memory (by knowing which "stars" we can ignore) or make the AI faster.
- Reasoning: The paper noticed that models trained specifically for math and reasoning (like the Qwen and DeepSeek models they tested) have these secondary sinks much more often. This suggests that these "pop-up stars" might be a secret weapon the AI uses to solve hard problems.
In short: The AI isn't just staring at the start of the sentence. In the middle of a complex thought, it creates temporary "focus points" to help it keep its balance, acting like a team of substitute librarians stepping in to keep the story on track.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.