用表格基础模型提升地质工程数据少时的预测与可解释性。
TabPFN Extensions for Interpretable Geotechnical Modelling
- 用嵌入相似度和SHAP分析增强模型可解释性,无需重新训练。
- 五项力学参数迭代填补误差降低,四种参数表现最优。
- 适合数据稀疏的地质工程场景,可辅助传统方法决策。
地质勘察依赖稀疏、异构的钻孔数据,不确定性量化与可解释性与预测精度同等重要。我们评估了表格基础模型TabPFN及其扩展库在两个地质任务上的表现:(1)基于N值和剪切波速进行土类分类,作为受控示范案例;(2)对五个力学参数($s_ ext{u}$, $E_ ext{u}$, ${σ'}_ ext{p}$, $C_ ext{c}$, $C_ ext{v}$)进行迭代填补,数据集为BM/AirportSoilProperties/2/2025。无需再训练,通过余弦相似度分析嵌入、可视化预测分布并计算SHAP归因。在回归基准上,将TabPFN与均值填补、线性回归、随机森林、XGBoost和HBM对比,提出一种基于上下文扰动类别的不确定性分解方法,并将$C_ ext{c}$与${σ'}_ ext{p}$的边际分布传播至一维固结模型,获得可靠度指标$β$与服务性超越概率$P_ ext{f}$。嵌入显示粘土/砂土标签一致性聚类;迭代填补使五项参数均降低RMSE,其中四项以TabPFN最低;SHAP归因符合Skempton压缩指数相关性及预压应力-含水率反相关性;后验内成分在不确定性分解中占比最大。本工作定位为可复用的评估流程,补充数据稀缺地质工程中的现有方法,非算法创新。
原文摘要 · Abstract (English)
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 ($s_\mathrm{u}$, $E_{\mathrm{u}}$, ${σ'}_\mathrm{p}$, $C_\mathrm{c}$, $C_\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 $C_\mathrm{c}$ and ${σ'}_\mathrm{p}$ distributions through a one-dimensional consolidation model to obtain the reliability index $β$ and serviceability exceedance probability $P_\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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。