Walker
Walker is a decoder-only language model that adds trajectory modeling to the Transformer.
The main idea is simple:
Attention looks at other tokens. Walker looks at how representations move.
How does attention work?
A Transformer represents each token as a vector. For every token, self-attention creates:
- Query (Q): what the token is looking for
- Key (K): what each token contains
- Value (V): the information that can be retrieved
The query is compared with other keys to determine which tokens are relevant:
In simple terms:
Attention asks: "What should I look at?"
A different approach: trajectory
Walker keeps attention, but asks another question:
"Where is my representation going?"
Consider the hidden states:
Instead of only using the states themselves, Walker measures their changes:
This is a discrete derivative of the hidden representation.
Now the model can observe a trajectory rather than just a sequence of points:
xβ β xβ β xβ β xβ β ...
β
differences
β
trajectory
Walker uses multiple scales of these differences to capture both short- and longer-range motion, along with changes in direction.
Why "Walker"?
Imagine the hidden states forming a path through a high-dimensional space.
A Transformer can look around the path using attention.
Walker also tries to walk along the path.
Transformer
β
βββ Attention
βββ "What should I look at?"
Walker
β
βββ Trajectory
βββ "Where am I going?"
The two mechanisms are complementary. Walker is not designed to replace attention; it adds trajectory features as a residual pathway alongside the Transformer.
Architecture
Input Tokens
β
Token Embedding
β
βββββββββββββ΄ββββββββββββ
β β
Attention Walker
β β
Token relationships Trajectory
β β
βββββββββββββ¬ββββββββββββ
β
Residual Update
β
βΌ
Next Block
The Walker pathway is initialized as an exact no-op, allowing the model to begin as a normal Transformer and gradually learn how much trajectory information to use.
Summary
Attention: What information is relevant?
Walker: How is the representation moving?
Walker is a Transformer that not only looks at the sequence, but also walks through the trajectory of its own representations.