Backend
Memory by Design: Probabilistic Sequence Layers
arXiv stat.ML0 views
arXiv:2605.31163v4 Announce Type: replace Abstract: We introduce the \emph{design-model framework}: a way to derive efficient recurrent sequence maps from explicit assumptions about memory. A design model writes evidence into memory by exact Bayesian filtering; a query- dependent readout produces a predictive distribution whose mean is the layer output. In our linear-Gaussian instantiation, the \emph{Bayesian Layer} propagates both a mean and a covariance: the covariance tracks uncertainty over stored associations, steering writes toward uncertain directions, attenuating gains as evidence accumulates, and preserving confident memories. The same framework unifies several sub-quadratic recurrences: linear attention, GLA, and Mamba-2/SSD are exact filters under a latent-input design model, whereas DeltaNet and related Delta-rule models are covariance-reset reductions of the Bayesian Layer's design model. Restoring covariance propagation yields closed-form predictions for retrieval dynamics, which we verify empirically, and improves robustness beyond the training regime in controlled collision studies, learned associative recall, and the Zoology MQAR benchmark. Training from scratch on WikiText-103 under matched state budgets lowers perplexity on associative-recall hits. Distilling Bayesian Layers into a pretrained 340M Gated DeltaNet improves RULER long-context retrieval over a matched-compute control, at a 2.5--2.7\% held-out perplexity cost.
Source:https://arxiv.org/abs/2605.31163PreviousKeeping Score: Adaptive, Tuning-Free Loss Weighting for Score-Augmented Neural Ratio EstimationNextMinimax PAC Bounds for Learning in Exogenous Contextual MDPs
Related posts
Actuator-Aware Observer Inputs for Cascade ADRC of Quadrotors Under Rotor Lag, Measurement Delay and Sensor Noise
0arXiv cs.RO
Keeping the Franka Emika Panda alive: a ROS 2 stack with a reliable position interface
0arXiv cs.RO
Rapid co-design of Buoyancy-assisted robots for Challenging Locomotion using Gaussian Evolutionary Specialists
0arXiv cs.RO
Specification-Aware Distribution Shaping for Robotics Foundation Models
0arXiv cs.RO
Sensorium: An Open-Source Multimodal Data Collection and Curation Library for Embodied Surgical AI
0arXiv cs.RO

Copilot Fusion Takes Local Models, and the Offline Boundary Is Somewhere Else
0dev.to AI