mlsanityData to 7 Oct 2026

Mechanistic Evidence for Spectral Structures in Prior-Data Fitted Networks

arXiv 2601.21731 ↗First posted 29 Jan 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Prior-Data Fitted Networks (PFNs) perform approximate Bayesian inference in a single forward pass, and tabular foundation models (TFMs) built on them are now widely used. To understand what networks infer internally, recent mechanistic studies of TFMs locate where predictions form, but treat these models as tabular predictors rather than as PFNs. It therefore remains unknown whether PFNs represent the spectral content of their context, the quantity that specifies a stationary kernel, and whether this content can be read out as an explicit kernel. We answer both questions. First, across seven PFNs, including four pretrained TFMs and a model trained only on a decision-tree prior, a linear probe on the residual stream recovers the frequency of the context with R2 ≥ 0.95. This structure is led by a single principal direction. Second, activation and subspace patching show that the network uses the structure through a low-dimensional subspace, where a few spectral directions move predictions far more than random ones. This holds even for the decision-tree model, so a spectral training prior is not required. On real datasets with up to 499 features, 64 of the 192 directions of TabPFN, chosen without labels, carry 85 to 95% of the causal effect of the context in all but one pair. Third, we introduce a Filter Bank Decoder that turns frozen PFN representations into an explicit stationary kernel through Bochner's theorem. Without any test-time optimization, the decoded kernel supports Gaussian process regression competitive with deep kernel learning and random Fourier features at about 200× lower cost. PFN latents therefore hold spectral structure that is causally used and recoverable as a portable kernel.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 26 references that are in the map. First posted on arXiv 29 Jan 2026.

Builds on

18 of 26 references.

References · 18

  1. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  2. 2020Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domainscited by 2,170
  3. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  4. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  5. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  6. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
  7. 2019Designing and Interpreting Probes with Control Taskscited by 608
  8. 2021Causal Abstractions of Neural Networkscited by 401
  9. 2021Transformers Can Do Bayesian Inferencecited by 338
  10. 2023Finding Alignments Between Interpretable Causal Variables and Distributed Neural Representationscited by 197
  11. 2013Gaussian Process Kernels for Pattern Discovery and Extrapolationcited by 131
  12. 2013Structure Discovery in Nonparametric Regression through Compositional Kernel Searchcited by 86
6 more
  1. 2024TuneTables: Context Optimization for Scalable Prior-Data Fitted Networkscited by 47
  2. 2014Automatic Construction and Natural-Language Description of Nonparametric Regression Modelscited by 40
  3. 2025TabPFN Unleashed: A Scalable and Effective Solution to Tabular Classification Problemscited by 33
  4. 2025Position: The Future of Bayesian Prediction Is Prior-Fittedcited by 19
  5. 2023Amortized Inference for Gaussian Process Hyperparameters of Structured Kernelscited by 10
  6. 2025Decoupled-Value Attention for Prior-Data Fitted Networks: GP Inference for Physical Equationscited by 1

Built on by

Not cited yet.