← Back to context

Comment by magicalhippo

4 hours ago

That's not how I read it.

For each position t and offset Δ∈ such that t+Δ≤L, we form a pair (z_t,z_t+Δ).

For a given t, this produces up to m pairs:

(z_t,z_t+Δ_1),(z_t,z_t+Δ_2),….

We treat these as local neighborhoods at multiple scales. Note that the pairing is causal: we only pair t with positions strictly to its right (future), consistent with left-to-right language modeling.

So z_t is a token in the input, and z_t+Δ is some later token. Guess they could have flipped that around to (z_t-Δ, z_t) with t+Δ>=0.

2 comments

magicalhippo

Reply
mapontosevenths  3 hours ago

> So z_t is a token in the input, and z_t+Δ is some later token.

I think you get what I'm saying, but when they compute the gate they do:

  h_t_new = alpha_t * h_t + (1 - alpha_t) * g_t

If I'm reading it right g_t = geometric_mix(z_t, z_(t+delta))

Which would mean that g_t includes t+Δ, which is a future token. It's leaking future tokens into the current math.

So it could just be a sign error and still be right in the code, or I could just be getting it wrong.

If I'm not though, and given that they got worse results than a standard transformer I really hope it's fixed in the code. If they do worse even when knowing the future it's bad news for the idea.