Robust Filter Attention: Self-Attention as Precision-Weighted State Estimation
Not on the map. Why?
Abstract (arXiv)
We introduce Robust Filter Attention (RFA), a formulation of self-attention as a robust state estimator. Each token is treated as a noisy observation of a latent trajectory governed by a linear stochastic differential equation (SDE), and attention weights are determined by consistency under this model rather than static feature similarity. Under isotropic noise and decay assumptions, RFA matches the computational complexity of standard attention. On language modeling benchmarks, RFA achieves lower perplexity than RoPE within the training window while remaining stable under zero-shot extrapolation to longer contexts. The framework also provides a dynamical interpretation of standard positional mechanisms, connecting rotational embeddings and recency biases to transport and uncertainty propagation induced by dynamics.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 29 of its 48 references that are in the map. First posted on arXiv 4 Sep 2025.
Builds on
29 of 48 references.
References · 29
- 2017Attention Is All You Needcited by 60,361
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
- 2020Hopfield Networks is All You Needcited by 475
- 2019Transformer Dissection: A Unified Understanding of Transformer's Attention via the Lens of Kernelcited by 213
17 more
- 2021KalmanNet: Neural Network Aided Kalman Filtering for Partially Known Dynamicscited by 132
- 2022A Length-Extrapolatable Transformercited by 127
- 2021Improving Transformers with Probabilistic Attention Keyscited by 28
- 2023Can a Transformer Represent a Kalman Filter?cited by 20
- 2020Kalman Filtering Attention for User Behavior Modeling in CTR Predictioncited by 15
- 2021ACE-NODE: Attentive Co-Evolving Neural Ordinary Differential Equationscited by 15
- 2021Probabilistic Attention for Interactive Segmentationcited by 12
- 2022Designing Robust Transformers using Robust Kernel Density Estimationcited by 11
- 2024Adaptive Kalman-Informed Transformercited by 11
- 2025Revisiting Kernel Attention with Correlated Gaussian Process Representationcited by 10
- 2022Learning Self-Modulating Attention in Continuous Time Space with Applications to Sequential Recommendationcited by 8
- 2022Neural Extended Kalman Filters for Learning and Predicting Dynamics of Structural Systemscited by 8
- 2024Elliptical Attentioncited by 7
- 2025Neural SDEs as a Unified Approach to Continuous-Domain Sequence Modelingcited by 6
- 2025Bayesian Attention Mechanism: A Probabilistic Framework for Positional Encoding and Context Length Extrapolationcited by 3
- 2024Nonlinear Kalman Filtering based on Self-Attention Mechanism and Lattice Trajectory Piecewise Linear Approximationcited by 2
- 2025Uncertainty Propagation Networks for Neural Ordinary Differential Equationscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026The Transformer as a Polar State Estimatorcited by 0
- 2026The Transformer as a Polar State Estimator10 May 2026