小米开源表格数据模型 TabLDM:四大基准评测进入第一梯队,OpenML-CTR23 回归排名第一
小米已开源通用表格数据基础模型 Xiaomi-TabLDM。该模型完全基于结构因果模型生成的大规模合成数据进行预训练,旨在无需针对每个数据集重新训练或调参,即可完成分类和回归任务。其架构采用双流特征分组、轻量级 Attention Residual、稀疏混合专家和 Test-Time Scaling。根据小米公布的评测结果,TabLDM 在 OpenML-CTR23 回归任务中排名第一,并在 TALENT、TabArena 和 BCCO 等四大基准评测中均处于第一梯队。相关代码和模型权重已通过 GitHub 与 Hugging Face 发布。
本文来源:IT之家,仅供学习参考,版权归原作者所有。