本文探讨表格基础模型能否摊销统计推断,提出了一种不同于传统逐问题设计估计器的新范式。作者聚焦置信区间构建这一经典任务,开发了TabCon系统,基于表格基础模型,通过单次前向传播即可为新数据集生成置信区间。其方法核心是稀疏专家混合架构与基于强化学习的后训练,用以将区间校准至目标覆盖水平。在大量基准数据集上,TabCon在保持接近名义覆盖率的同时间隔更短,并在推理效率上显著优于经典自助法,即使后者仅使用50次自助采样,TabCon仍快约50倍。该工作表明,摊销式推断有望在保证统计有效性的前提下大幅降低计算成本,为统计推断的规模化与自动化提供了新思路。
| TabCon | 本文提出的基于表格基础模型的摊销推断系统,通过前向传播直接为新数据集生成置信区间。 |
| 摊销推断 | 一种将推断过程预先训练成模型、使得对新数据集的推断只需一次前向传播即可完成的范式。 |
| 表格基础模型 | 在大规模表格数据上预训练、可迁移到多种表格任务的深度学习模型。 |
| 稀疏专家混合 | 一种模型架构,包含多个专家子网络,并通过稀疏门控仅激活部分专家,以提升效率和容量。 |
| 基于强化学习的后训练 | 在预训练模型基础上,使用强化学习进一步微调,以校准置信区间达到目标覆盖水平。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅