Where Computation Lives Inside TabPFN: Causal Localisation of Attention Head Function

Atharva Gupta, Dhruv Kumar, Murari Mandal, Saurabh Deshpande
6/11/2026
cs.LG

No confident code implementation found for this paper.

Abstract

We present the first causal mechanistic analysis of a tabular foundation model, investigating how TabPFN 2.5's feature wise attention heads distribute computation across layers. Using activation patching, ablation, and attention entropy across two synthetic regression datasets, we find clear temporal specialisation: one head's causal necessity dominates that of the others by 2 to 5 times at peak layer, with its dominant layer shifting across tasks of different complexity, while the remaining heads exhibit symmetric late layer profiles. Attention entropy and patching provide convergent evidence for the computationally active layers of the dominant head. We additionally investigate inference time steerability via contrastive activation steering, which fails to transfer across samples. We attribute this result to TabPFN's in context learning mechanism, which encodes task structure through context dependent attention rather than the stable parametric directions that make steering tractable in language models.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{gupta2026where,
  title  = {Where Computation Lives Inside TabPFN: Causal Localisation of Attention Head Function},
  author = {Atharva Gupta and Dhruv Kumar and Murari Mandal and Saurabh Deshpande},
  year   = {2026},
  eprint = {2606.12917},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.12917v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.12917.svg)](https://codeofpaper.com/paper/2606.12917)

Discussion