Neural expectation-maximization and meta-models
Not on the map. Why?
Abstract (arXiv)
We interpret Transformer networks as learned approximations to a hybrid probabilistic-inference procedure that combines max-product Loopy Belief Propagation (LBP) with Expectation--Maximization (EM). In this view, a Transformer performs inference in partially observed Markov Random Fields (MRFs) whose structure and parameters are not given a priori but are inferred from data. We show that self-attention implements a differentiable approximation to max-product LBP on an input-conditioned, dynamically structured MRF, with attention weights encoding learned structural affinities. We further establish an EM correspondence at the level of depth: each layer updates latent representations in an E-step-like forward pass, while the transition θ(l) → θ(l+1) across layers acts as an implicit M-step. Building on this interpretation, we introduce meta-models, a design paradigm in which inputs generate instance-specific parameters, unifying mechanisms such as self-attention, mixture-of-experts, and gated units within a single framework. Empirical results support the proposed perspective.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 32 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
22 of 32 references.
References · 22
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2017Graph Attention Networkscited by 8,923
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,261
- 2024M3-Embedding: Multi-Linguality, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillationcited by 1,073
- 2016HyperNetworkscited by 1,027
- 2022In-context Learning and Induction Headscited by 964
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
10 more
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2016Language Modeling with Gated Convolutional Networkscited by 702
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 684
- 2017Exploring Generalization in Deep Learningcited by 600
- 2022TimeLMs: Diachronic Language Models from Twittercited by 131
- 2023Factor Graph Neural Networkscited by 15
- 2025Interplay Between Belief Propagation and Transformer: Differential-Attention Message Passing Transformercited by 6
- 2021Graphical Models with Attention for Context-Specific Independence and an Application to Perceptual Groupingcited by 3
- 2025Transformers as Unsupervised Learning Algorithms: A study on Gaussian Mixturescited by 3
- 2025Mitigating Attention Localization in Small Scale: Self-Attention Refinement via One-step Belief Propagationcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Constrained belief updates explain geometric structures in transformer representations
- 2025Is Random Attention Sufficient for Sequence Modeling? Disentangling Trainable Components in the Transformer
- 2025Sigmoid Self-Attention has Lower Sample Complexity than Softmax Self-Attention: A Mixture-of-Experts Perspective
- 2026Hybrid Linear Attention Done Right: Efficient Distillation and Effective Architectures for Extremely Long Contexts
- 2021Sum-Product-Attention Networks: Leveraging Self-Attention in Probabilistic Circuits
- 2026A Comparative Analysis of Attention versus State-Space Models for In-Context Learning
- 2024Interpreting Attention Layer Outputs with Sparse Autoencoders
- 2026ITNet: A Learnable Integral Transform That Subsumes Convolution, Attention, and Recurrence