Can you do that in the middle of the layers? And if you do, would that word be that meaningful to the final output? Genuinely curious.
One problem with this technique is that the model wasn't trained with intermediate layers being mapped to logits in the first place, so it's not clear why the LMHead should be able to map them to anything sensible. But alas, like everything in DL research, they threw science at the wall and a bit stuck.