Research & Papers

Transformers can do Bayesian model selection, but only with meaningful tokens

A 2.8M-parameter transformer matches Bayesian optimal posteriors to within 0.01 bits—until arithmetic hits opaque symbols.

Deep Dive

Prior work showed transformers can do Bayesian filtering within a fixed hypothesis class. But can they identify the correct hypothesis class itself? In a new preprint titled "Bayesian Wind Tunnels for Model Selection," Siddhartha Dalal, Vishal Misra, and Abhay Parekh create controlled environments where ground-truth posteriors over hypothesis classes are known exactly. Using fixed-point-free involutions—a purely relational function where f(f(x))=x—a tiny 2.8M-parameter transformer achieves 0.01-bit entropy agreement with the Bayesian optimum across multiple seeds. The model also handles non-nested comparisons (involutions vs. 3-cycles) with class-posterior MAE under 0.001, proving genuine model selection beyond simplicity bias.

The key finding is a sharp perceptual access boundary: when the discriminative statistic requires arithmetic (modular addition or multiplication), model selection succeeds with integer tokens but fails completely with opaque symbols that change meaning each episode. This failure holds even after 112x scaling (2.8M to 316M parameters), but a stationarity control—where opaque tokens keep fixed relabelings—succeeds (0.009-bit MAE), showing stable semantics enable circuit compilation. Probing frontier LLMs reveals qualitative Bayesian behavior but a large calibration gap (~55x). The work has major implications for understanding when transformers truly generalize vs. exploit token meaning.

Key Points
  • A 2.8M-parameter transformer matches Bayesian optimum posteriors to within 0.01 bits on relational tasks (fixed-point-free involutions).
  • Model selection fails completely on arithmetic tasks when using opaque symbols instead of integers, even after 112x scaling to 316M parameters.
  • Frontier LLMs show qualitative Bayesian behavior but a 55x calibration gap compared to the Bayesian optimum.

Why It Matters

Reveals a fundamental limitation in transformer reasoning: they need stable token semantics, not just structure, to generalize.

📬 Get the top 10 AI stories daily