Learning to Fluctuate: Statistical Foundations for Causal Tabular Pretraining
Abstract
Causal tabular foundation models amortize effect estimation across synthetic mechanisms, but supervision by the mechanism-level effect , which we call latent-effect supervision, rewards posterior shrinkage instead of directly encoding the repeated-sample response needed in a fixed deployment population. We introduce fluctuation-supervised pretraining (FSP): each synthetic table is labeled by its average treatment effect plus its efficient influence-function fluctuation, while deployment remains a single frozen forward pass. Along the path , we prove an endpoint transition: every fixed retains label ambiguity of order , whereas full fluctuation makes the Gaussian label observable and reduces optimal finite-stratum causal label-prediction risk to order . One finite-pretraining bound combines label, network, episode-sampling, and optimization errors; its resulting sampling defect controls fixed-mechanism bias, mean squared error, variance, Gaussian approximation, and, with variance-head accuracy, studentized coverage. Complementary lower bounds separate the local ATE risk that deployment observations cannot erase from the excess risk of a generic finite-dictionary episode-learning problem. Experiments trace the learned sampling response. Across 24 nonlinear continuous-covariate cells at trained context lengths, continuous-row FSP lowers checkpoint-mean macro RMSE by 7.0% versus S-learner and wins all 12 weak-overlap cells; validation-selected Summary FSP deploys faster per table than S-learner and faster than the released CausalPFN-S pipeline in our warm one-thread CPU benchmark. Under effect shift, matched Raw FSP lowers mean-checkpoint RMSE by 54.2% and teacher defect by 99.0% versus latent-effect supervision, and RMSE by 10.2% versus the released CausalPFN-S checkpoint. Known-effect semisynthesis tests coverage; two randomized-study evaluations show that lower RMSE can coexist with residual attenuation.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.