English

TabPFN Extensions for Interpretable Geotechnical Modelling

Computational Engineering, Finance, and Science 2026-05-21 v2 Machine Learning

Abstract

Geotechnical site characterisation relies on sparse, heterogeneous borehole data, where uncertainty quantification and interpretability matter as much as predictive accuracy. We evaluate TabPFN~\citep{Hollmann2025}, a tabular foundation model, and its \texttt{tabpfn-extensions} library on two geotechnical tasks: (1) soil-type classification from N-value and shear-wave velocity data as a controlled illustrative case, and (2) iterative imputation of five mechanical parameters (sus_\mathrm{u}, EuE_{\mathrm{u}}, σp{\sigma'}_\mathrm{p}, CcC_\mathrm{c}, CvC_\mathrm{v}) in BM/AirportSoilProperties/2/2025. Without retraining, we apply cosine-similarity analysis to TabPFN embeddings, visualise predictive distributions, and compute SHAP attributions. On the regression benchmark we compare TabPFN with mean imputation, linear regression, random forests, XGBoost, and HBM; introduce a proxy decomposition of predictive uncertainty across context-perturbation classes; and propagate marginal CcC_\mathrm{c} and σp{\sigma'}_\mathrm{p} distributions through a one-dimensional consolidation model to obtain the reliability index β\beta and serviceability exceedance probability PfP_\mathrm{f}. Embeddings exhibit label-consistent Clay/Sand grouping; iterative imputation reduces RMSE for all five targets, with TabPFN lowest on four; SHAP attributions are consistent with the Skempton compression-index correlation and the inverse preconsolidation-pressure-water-content dependence; the within-posterior component is largest in the proxy decomposition. We position the contribution as a worked evaluation workflow that may complement established methods for data-scarce geotechnics, not as algorithmic innovation.

Cite

@article{arxiv.2603.21033,
  title  = {TabPFN Extensions for Interpretable Geotechnical Modelling},
  author = {Taiga Saito and Yu Otake and Daijiro Mizutani and Stephen Wu},
  journal= {arXiv preprint arXiv:2603.21033},
  year   = {2026}
}
R2 v1 2026-07-01T11:31:52.213Z