Let's say that the forward pass that selected "Aha" produces activations that indicate a wrong assumption, and a plausible explanation.
It puts learned projections of the activation into the KV Cache and outputs Aha.
Both the cached projections and the current Aha token can now influence further activations in an additional Forward pass that the Aha bought the model.
Beats me how it works, honestly can't wrap my head around it.
From what I understand, at position Aha in each layer it's constructing a query based on the current activation and looking at the key of each other token position for that layer, in order to decide how much attention to pay to the value.
In this way it attends to the previous values, such as perhaps the incorrect assumption and plausible explanation.
Let's say that the forward pass that selected "Aha" produces activations that indicate a wrong assumption, and a plausible explanation.
It puts learned projections of the activation into the KV Cache and outputs Aha.
Both the cached projections and the current Aha token can now influence further activations in an additional Forward pass that the Aha bought the model.
At least that's how I thought it works.