Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces LinearPFN, a prior-data fitted transformer network designed to amortize spike-and-slab inference for linear models incorporating main effects and pairwise interactions. By pretraining on synthetic data from a conjugate prior, LinearPFN enables a single forward pass on new datasets to yield posterior inclusion probabilities, mean coefficients, and predictive distributions without per-dataset fitting. Empirical results on social science datasets demonstrate that LinearPFN outperforms classical baselines in variable selection accuracy, even when model parameters deviate from the training prior.
Amortized spike-and-slab inference for linear models with interactions achieves state-of-the-art variable selection accuracy with a single forward pass, eliminating costly per-dataset MCMC.
Spike-and-slab regression is a standard Bayesian formulation of variable selection: it returns a posterior distribution over which candidate effects are active rather than a single selected subset, so that every candidate effect carries an inclusion probability. Its cost grows exponentially with the number of candidate effects, so the posterior can be enumerated exactly only when the number of predictors is small. Beyond that reach, the posterior has to be approximated, typically by Markov chain Monte Carlo over the model space, which requires a fresh run for every dataset and, within a fixed budget of steps, may fail to converge. We present LinearPFN, a prior-data fitted transformer network that amortizes spike-and-slab inference for linear models with main effects and pairwise interactions. The network is pretrained once on synthetic datasets, drawn from an explicitly specified prior, and a single forward pass over a new dataset returns posterior inclusion probabilities, posterior-mean coefficients and posterior predictive distributions with no per-dataset fitting. The prior is conjugate by design, so that the posterior for each fixed set of active effects has a closed form, and wherever the exact posterior is still computable by enumeration we verify the network's outputs against it. On real predictor matrices from published social-science datasets, with outcomes drawn from the prior so that the true active set is known, LinearPFN attains a higher per-dataset selection AUC and a higher F1 under the median probability model rule than five classical baselines. The lead holds when the coefficients, the interactions or the noise depart from the prior. Code: https://github.com/schiekiera/LinearPFN. Trained model: https://huggingface.co/schiekiera/LinearPFN.