跳到正文
arXiv · cs.AI· Tejas Vaidhya·· 17 天前精选AI 评分54

Tab-Loop 用循环权重共享替换 TabICLv2 的 ICL 层,参数减少 62% 性能持平

Recurrent In-Context Learning for Tabular Foundation Models

来源与原文

arXiv · cs.AI

作者:Tejas Vaidhya

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2610.14998v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

Tejas Vaidhya 在 arXiv 论文中提出 Tab-Loop,将 TabICLv2 的 12 个非共享 ICL 块替换为 4 个共享块循环应用 k 次,其余编码器、解码器、合成先验和训练配方不变。在相同预训练算力 18.2 EFLOP 与相同推理算力下,参数量从 27.6M 降到 10.4M,held-out log loss 0.4583 对 0.4588,200 个 TALENT 数据集平均准确率 84.10% 对 83.92%,38 个 TabArena 数据集 Elo 1409 对 1396。由于权重共享,深度可在推理时选择,在训练范围内增加循环次数可提升精度且超出范围仍成立,用更深的循环继续训练可在 4–5 倍推理算力下达到基线 stage-2 成绩。

推荐理由

论文在相同预训练与推理算力下对比权重共享与独立参数,给出参数量、精度和 Elo 的具体数字,可迁移到 ICL 模型的参数效率设计。

来源:arXiv · cs.AI · arxiv.org