论文

研究解析 TabPFN-3 逐层推理:深度并非对每个输入都有帮助

When Does Depth Help? Understanding Layerwise Inference in Tabular Foundation Models

精选理由

TabPFN-3 逐层拆开做了 272 个数据集的实验,发现模糊样本才需要深层计算,研究模型内部机制的可以看看这篇。

arXiv 论文研究了 Transformer 表格基础模型 TabPFN-3 全部 24 层上下文学习层的推理行为,覆盖 272 个分类数据集。分析发现表征呈扩展-压缩轨迹,后期维度收缩越强,类原型分离提升越大。对提前稳定样本的分析显示,预测类别稳定后表征仍在显著变化,而局部模糊样本从浅层到深层获得的准确率提升大于分离清晰的样本。重复单个预训练块或打乱层序平均会降低预测性能,说明简单的事后层替换不可行。

原文 · arXiv cs.LG

When Does Depth Help? Understanding Layerwise Inference in Tabular Foundation Models

Transformer-based tabular foundation models (TFMs) achieve state-of-the-art performance on small to medium-sized tabular prediction tasks while processing diverse inputs through a shared stack of layers. Recent layerwise analyses identify strong predictive performance in early TFM representations and substantial overlap in computation across layers. However, early predictive accessibility alone does not establish whether later layers are dispensable for every input, and aggregate accuracy can obscure which inputs benefit from additional computation. We study all 24 in-context-learning layers of TabPFN-3 across 272 classification datasets, combining representation geometry, prediction trajectories, truncation, and structural interventions, while also examining related representation dynamics in four additional TFMs. We uncover a pronounced expansion-compression trajectory, with stronger late dimensional contraction associated with larger increases in class-prototype separation. A targeted analysis of early-stabilizing samples shows that substantial representational changes persist after predicted classes stabilize, while locally ambiguous queries obtain larger accuracy gains from shallow to late layers than well-separated queries. Repeating a single pretrained block or perturbing layer order reduces predictive performance on average, constraining simple post-hoc substitutions in the frozen model. These findings distinguish predictive accessibility, class stability, and the benefits of continuing computation, supporting an input-dependent assessment of depthwise redundancy in TFMs.