Has anyone tried/thought about logit lenses in recurrent models/Astra?
Astra uses two passes to think and then decodes the result of the second loop into tokens. I’m reasonably sure that the hidden state at the end of the first loop would show something interesting if put through the unembedding matrix.
Even if there’s some kind of switch that gets thrown in the second loop to indicate that results must decode to tokens, it might be a linear direction that can be added to first loop residual states to make them more comprehensible
Do you have code/data from this publicly available?