Predictive Geometry of Hidden Trajectories in Transformers
This study addresses the unclear geometric properties of hidden states in Transformers trained solely with terminal losses, particularly regarding their constraints from downstream computation. We analyze the local second-order geometry of layer-wise losses and employ the pullback Fisher operator to identify output-sensitive directions and predictive null spaces, thereby constructing an observable residual stream subspace. Furthermore, we propose a token-level curvature score based on Fisher-weighted sensitivity, serving as a loss-aware alternative to attention magnitude for enabling non-uniform hierarchical rank allocation, which is efficiently estimated via matrix-free Jacobian-vector products. Evaluated on datasets such as WikiText, this approach effectively predicts perturbation sensitivity, provides competitive structured token pruning signals, and significantly enhances the recovery performance of low-rank student models during autoregressive distillation.