TabPFN extensions for interpretable geotechnical modelling
Geotechnical site characterisation relies on sparse, heterogeneous borehole data, where uncertainty quantification and interpretability matter as much as predictive accuracy. We evaluate TabPFN, a tabular foundation model, and its 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 (su, Eu, σ′p, Cc, Cv) 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 Cc and σ′p distributions through a one-dimensional consolidation model to obtain the reliability index β and serviceability exceedance probability Pf. 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.
Authors
- Yu Otake (ORCID: https://orcid.org/0000-0002-4879-540X)
- Daijiro MIZUTANI (ORCID: https://orcid.org/0000-0003-1028-2489)
- Stephen Wu (ORCID: https://orcid.org/0000-0002-7847-8106)
- Taiga Saito (ORCID: https://orcid.org/0000-0002-6662-2000)
Institutions
- The Graduate University for Advanced Studies, SOKENDAI (JP)
- Tohoku University (JP)
- The Institute of Statistical Mathematics (JP)
Publication Details
- Journal
- Georisk Assessment and Management of Risk for Engineered Systems and Geohazards
- Published
- 2026-07-28
- DOI
- https://doi.org/10.1080/17499518.2026.2707500
- Primary Topic
- Geological Modeling and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00