OrthoFormer: Instrumental Variable Estimation in Transformer Hidden States via Neural Control Functions
Not on the map. Why?
Abstract (arXiv)
Transformer architectures excel at sequential modeling yet remain fundamentally limited by correlational learning - they capture spurious associations induced by latent confounders rather than invariant causal mechanisms. We identify this as an epistemological challenge: standard Transformers conflate static background factors (intrinsic identity, style, context) with dynamic causal flows (state evolution, mechanism), leading to catastrophic out-of-distribution failure. We propose OrthoFormer, a causally grounded architecture that embeds instrumental variable estimation directly into Transformer blocks via neural control functions. Our framework rests on four theoretical pillars: Structural Directionality (time-arrow enforcement), Representation Orthogonality (latent-noise separation), Causal Sparsity (Markov Blanket approximation), and End-to-End Consistency (gradient- detached stage separation). We prove that OrthoFormer achieves bias strictly less than OLS for any valid instrument lag, with residual bias decaying geometrically as O(rhok ). We characterize the bias-variance-exogeneity trilemma inherent in self-instrumenting and identify the neural forbidden regression - where removing gradient detachment improves prediction loss while destroying causal validity. Experiments confirm all theoretical predictions. OrthoFormer represents a paradigm shift from correlational to causal sequence modeling, with implications for robustness, interpretability, and reliable decision-making under distribution shift.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 20 references that are in the map. First posted on arXiv 8 Mar 2026.
Builds on
6 of 20 references.
References · 6
- 2017Attention Is All You Needcited by 60,361
- 2019Invariant Risk Minimizationcited by 1,854
- 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 941
- 2019Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecastingcited by 761
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 123
- 2018Adversarial Generalized Method of Momentscited by 34
Built on by
Not cited yet.