How Should Reasoning Be Organized in a Transformer's Latent Space?
Experiments with two-layer and GPT-2 Transformers reproduce the predicted width advantage and show that unequal weights fail first on the states that receive the least weight, suggesting a important principle: keep reached states equally weighted, and restore equal weights as computation proceeds.