Momentum Attention: The Physics of In-Context Learning and Spectral Forensics for Mechanistic Interpretability
On the map: Mechanistic interpretability → Circuits and mechanistic analysis
Abstract (arXiv)
The Mechanistic Interpretability (MI) program has mapped the Transformer as a precise computational graph. We extend this graph with a conservation law and time-varying AC dynamics, viewing it as a physical circuit. We introduce Momentum Attention, a symplectic augmentation embedding physical priors via the kinematic difference operator pt = qt - qt-1, implementing the symplectic shear hatqt = qt + γ pt on queries and keys. We identify a fundamental Symplectic-Filter Duality: the physical shear is mathematically equivalent to a High-Pass Filter. This duality is our cornerstone contribution -- by injecting kinematic momentum, we sidestep the topological depth constraint (L ≥ 2) for induction head formation. While standard architectures require two layers for induction from static positions, our extension grants direct access to velocity, enabling Single-Layer Induction and Spectral Forensics via Bode Plots. We formalize an Orthogonality Theorem proving that DC (semantic) and AC (mechanistic) signals segregate into orthogonal frequency bands when Low-Pass RoPE interacts with High-Pass Momentum. Validated through 5,100+ controlled experiments (documented in Supplementary Appendices A--R and 27 Jupyter notebooks), our 125M Momentum model exceeds expectations on induction-heavy tasks while tracking a 350M baseline within ~2.9% validation loss. Dedicated associative recall experiments reveal a scaling law γ^* = 4.17 × N-0.74 establishing momentum-depth fungibility. We offer this framework as a complementary analytical toolkit connecting Generative AI, Hamiltonian Physics, and Signal Processing.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Circuits and mechanistic analysis, one of 13 directions in Mechanistic interpretability. First posted on arXiv 3 Feb 2026.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Certified Mechanistic Interpretability: Lifting Single-Input Findings to Bounded Neighbourhoods
- 2026Intrinsic Structure: Spectral Identifiability for Mechanistic Interpretability
- 2026PhysSAE: Mechanistic Interpretability of PINNs with Sparse Autoencoders
- 2026Sparse probes and murky physics: a case study of interpretability challenges in a foundation model for continuum dynamics
- 2025Interpretability in Parameter Space: Minimizing Mechanistic Description Length with Attribution-based Parameter Decomposition
- 2026Spectral Edge Dynamics Reveal Functional Modes of Learning
- 2026A Measure-Theoretic Analysis of Reasoning: Structural Generalization and Approximation Limits
- 2026Hidden not Deleted: How Networks Suppress Entangled Features