AI 快讯NVIDIA Kumo Tabular来了
模型上新

NVIDIA Kumo Tabular来了

2026-09-29T22:04:49.296Z
NVIDIA Kumo Tabular来了

NVIDIA近日发布预训练表格基础模型 Kumo Tabular,面向分类和回归任务,以少量标注样本作为上下文完成零样本或少样本预测,试图在传统模型的效率与深度模型的泛化能力之间找到新平衡。

NVIDIA Kumo Tabular来了:表格预测开始进入基础模型时代

NVIDIA 近日通过 Hugging Face 发布 Kumo Tabular,一款面向分类和回归任务的预训练表格基础模型。它不要求开发者为每个数据集重新训练一套模型,而是把带标签的数据样本作为上下文,直接预测新样本的类别概率或数值结果。

这件事的重点,不是 NVIDIA 又推出了一个“更大的表格模型”,而是表格预测正在从“每张表训练一个模型”转向“一个预训练模型适配多张表”。对于信用风控、用户流失预测、医疗风险评估、设备故障预警这类数据结构稳定、但任务经常变化的场景,Kumo Tabular 的价值可能比单纯追求更高的离线分数更现实。

NVIDIA Kumo Tabular 表格基础模型工作流程示意图,展示带标签上下文样本输入模型并输出分类概率或回归结果

Kumo Tabular是什么

**Kumo Tabular 是 NVIDIA 发布的预训练表格基础模型,能够基于上下文中的标注样本执行分类和回归预测。**它的输入不是一段自然语言提示,而是一组结构化特征、标签和待预测样本;它的输出也不是文本,而是类别概率或连续数值。

这使它与 GPT、Claude 这类语言模型有本质区别。语言模型处理的是 token 序列,Kumo Tabular 处理的是列、行、数据类型和标签之间的关系。它不需要把表格先拼成自然语言,也不依赖开发者为每一列编写提示词。

从 Hugging Face 模型仓库给出的示例看,Kumo Tabular 可以通过 structured-data-models Python 包调用。开发者需要先把 Pandas DataFrame 转换为 TableTensor,再明确指定分类或回归任务,将一部分带标签数据作为 x_context 和 y_context,将待预测部分作为 x_query 输入模型。

官方示例使用的是经典的 Breast Cancer Wisconsin 数据集,并将前 300 条样本作为上下文、剩余样本作为查询数据。模型通过 num_estimators=8 进行多次估计后输出分类概率。这个接口透露出一个重要设计:Kumo Tabular 的推理过程更接近“基于示例的预测”,而不是先训练、再部署的传统机器学习流程。

import sdm
from sklearn.datasets import load_breast_cancer

df = load_breast_cancer(as_frame=True).frame

table = sdm.TableTensor.from_pandas(
    df=df,
    stypes=sdm.infer_stypes(
        df,
        overrides={"target": "categorical"}
    ),
    device="cuda",
)

model = sdm.models.KumoTabular(
    task="classification",
    device="cuda",
)

probs = model(
    x_context=table[:300].drop_columns("target"),
    y_context=table[:300, "target"],
    x_query=table[300:].drop_columns("target"),
    num_estimators=8,
)

这段代码不是在访问远程推理服务,而是在本地通过结构化数据模型工具加载模型并执行推理。当前公开资料显示,Kumo Tabular 权重采用 OpenMDW 1.1 许可证发布,模型仓库位于 Hugging Face,配套推理包名为 structured-data-models。

它解决了传统表格机器学习的什么问题

**表格预测的核心问题,是单个任务通常拥有有限数据,但业务方又希望模型快速上线并保持稳定效果。**传统方法在这件事上并不差,XGBoost、LightGBM、CatBoost 仍然是工业界大量表格任务的默认选择,但它们通常需要针对每个数据集单独训练、调参和验证。

对于数据量充足、特征工程成熟的任务,这种方式很有效。问题出现在以下几类场景:

  • 新业务刚上线,只有几百到几千条标注样本;
  • 数据集经常变化,重新训练成本高;
  • 同一家公司有大量相似但不完全相同的预测任务;
  • 数据科学团队需要快速验证一个预测方向,而不是立即搭建完整训练流水线;
  • 任务本身是低频的,专门训练模型的工程成本超过了模型带来的收益。

Kumo Tabular 试图把预训练阶段的成本摊销到大量下游任务上。开发者不必从随机初始化开始训练,而是向模型提供当前任务的少量示例,让它利用预训练阶段学到的表格模式完成预测。

可以把它理解成两种工作方式的差别:传统模型像是每接手一个新项目就重新培养一名分析师;表格基础模型则像是一名已经看过大量不同数据表的分析师,拿到新项目后先阅读几十到几百条样本,再开始判断。

这种类比并不意味着模型理解了业务语义。列名、数据质量、标签定义和样本分布仍然决定了预测上限。它只是把“如何从结构化数据中寻找可预测关系”的一部分能力提前训练好了。

精度与效率,Kumo押注的是中间地带

**Kumo Tabular 的产品定位不是取代所有梯度提升树,而是在少样本适配、跨任务复用和推理效率之间寻找平衡。**这也是 NVIDIA 官方文章标题中“accuracy-efficiency frontier”的含义:模型并非只追求最高精度,而是试图在准确率、适配成本和推理速度之间取得更好的组合。

在表格领域,精度和效率往往存在现实冲突。大型深度模型可以学习更复杂的表示,但训练和推理代价更高;传统树模型训练速度快、解释路径清晰,却需要针对每个任务单独拟合。零样本或少样本模型可以减少训练环节,但单次推理可能需要处理更多上下文样本。

Kumo Tabular 采用 num_estimators 参数进行多次估计,示例值为 8。这个参数可以理解为让模型从多个估计结果中形成更稳定的预测。它可能带来更稳健的概率输出,但也意味着推理成本会随估计次数增加。生产环境不能只看单次准确率,还要同时测量延迟、显存占用和吞吐量。

目前公开页面并没有在模型卡中给出一套足以覆盖所有业务场景的统一价格表,也没有把 Kumo Tabular 变成按调用次数计费的云端产品。它更像是一个面向开发者和研究者的模型权重与本地推理组件。实际成本主要取决于显卡、数据规模、上下文样本数量以及估计次数。

与常见方案怎么选

**Kumo Tabular 适合被看作传统表格模型和表格基础模型之间的新选项,而不是 XGBoost 或 CatBoost 的自动替代品。**不同方案的适用边界并不相同。

| 方案 | 典型工作方式 | 训练成本 | 少样本适配 | 推理特点 | 更适合的场景 | |---|---|---:|---|---|---| | XGBoost / LightGBM | 针对每个任务单独训练 | 低到中 | 较弱 | 快,工程成熟 | 数据量充足、特征工程明确的生产任务 | | CatBoost | 针对每个任务单独训练,擅长类别特征 | 低到中 | 较弱到中等 | 快,对类别列友好 | 类别特征较多、需要稳定基线的任务 | | 深度表格模型 | 针对数据集训练神经网络 | 中到高 | 中等 | 取决于模型规模 | 有较多数据、需要复杂表示学习的任务 | | Kumo Tabular | 提供上下文样本后直接预测 | 前置预训练已完成 | 强 | 受上下文规模和估计次数影响 | 快速验证、低标注量、跨任务复用 | | 人工特征工程加传统模型 | 手动构造业务特征后训练 | 中 | 取决于特征质量 | 通常较快 | 业务规则清晰、可解释性要求高的任务 |

如果团队已经有一套经过多年验证的 LightGBM 流水线,Kumo Tabular 不会因为“基础模型”这个名字就自动胜出。树模型的训练速度、可解释性和线上稳定性仍然是强项,尤其是在百万级样本、特征分布稳定、每天定时重训的业务中。

但在数据量较小、任务变化快、需要同时评估几十个预测目标的场景,Kumo Tabular 的方法更有吸引力。它可以先帮助团队判断某个预测方向是否值得投入,而不是在特征工程和超参数搜索完成后才知道这个方向本身没有信号。

为什么表格基础模型现在受到关注

**表格基础模型是试图用一个预训练模型统一处理不同结构化数据集的技术路线。**它的研究价值在于,表格数据不像文本那样拥有统一的 token 体系,每张表的列名、类型、取值范围和业务含义都可能完全不同。

一列“年龄”可能是整数,一列“客户等级”可能是有序类别,一列“设备状态”可能是字符串,也可能被编码为 0、1、2。模型需要识别这些差异,还要处理缺失值、类别分布变化、异常值和不同列之间的交互关系。

这也是表格基础模型比语言模型直接套用到表格上更难的地方。把一行数据序列化成文本,确实可以让大语言模型读取,但这种方式会引入额外 token 成本,也容易丢失数据类型和数值距离信息。对于“100 和 101 很接近,但 100 和 10000 差别很大”这类关系,专门的结构化数据模型通常更自然。

NVIDIA 此前已经围绕结构化数据和关系型数据推出 Kumo Relational 等方向。Kumo Relational 面向多表、实体关系和图结构预测;Kumo Tabular 则聚焦单表或相对规整的表格任务。两者的边界可以简单理解为:前者关注“不同表之间如何连接”,后者关注“当前这张表如何预测”。

这条产品线说明 NVIDIA 正在把基础模型能力从文本、图像和代码扩展到企业数据中更常见、但关注度相对较低的结构化数据。对于企业 AI 来说,真正影响决策的往往不是一段文本,而是客户、交易、库存、设备、订单和风险记录。

开发者需要关注的几个限制

**Kumo Tabular 的少样本能力并不等于可以忽略数据治理和验证流程。**表格预测中的数据泄漏、标签延迟和分布偏移,仍然会让一个看似准确的模型在生产环境中失效。

第一,模型上下文中的样本必须具有代表性。如果上下文数据只覆盖某个地区、某个时间段或某类客户,模型对查询样本的预测可能出现明显偏差。少样本方法降低了训练成本,却提高了样本选择的重要性。

第二,分类概率不能直接等同于可用的业务决策。风控、医疗和运维场景通常需要校准后的概率、阈值策略和代价敏感评估。开发者至少应该同时查看 AUROC、AUPRC、准确率、召回率、校准误差和不同人群之间的性能差异,而不是只看一个排行榜分数。

第三,表格模型的推理延迟需要单独测量。上下文样本越多、num_estimators 越高,通常意味着更多计算量。对于离线分析,这个成本可能可以接受;对于每秒数千次请求的实时推荐或风控系统,则需要评估批处理、缓存和模型蒸馏等方案。

第四,开源权重不代表所有部署条件都已经成熟。开发者需要核对 OpenMDW 1.1 的使用限制、模型依赖、显卡显存要求和企业内部合规要求。特别是医疗、金融等场景,还要确认训练数据来源、隐私边界和模型输出是否满足审计要求。

现在值得用它做什么

**Kumo Tabular 最适合先作为新的基线和探索工具,而不是直接替换现有生产模型。**开发者可以从三个方向开始验证。

一是小样本任务。把现有训练集拆出不同规模的上下文集,例如 32、64、128、256 和 512 条样本,观察模型在样本数量增加时的准确率、概率校准和延迟变化。这样才能判断它是否真正适合当前业务的数据规模。

二是跨数据集迁移。选择结构相近但业务来源不同的表格,比较 Kumo Tabular、LightGBM 和 CatBoost 在相同数据预算下的表现。关键不是只比较最高分,而是比较达到目标指标需要多少标注数据和多少工程时间。

三是快速筛选预测目标。对于同时存在多个候选标签的项目,可以用 Kumo Tabular 快速测试哪些目标具有可预测信号,再把资源投入到真正值得生产化的任务上。

一个合理的评估流程应该包含以下步骤:

  1. 明确任务类型、标签定义和预测时间窗口,避免把未来信息泄漏到特征中。
  2. 按时间或业务实体划分训练、验证和测试数据,而不是只做随机切分。
  3. 用 Kumo Tabular 建立少样本和全量上下文两组结果。
  4. 同时训练 LightGBM、CatBoost 等强传统基线。
  5. 记录准确率、召回率、概率校准、显存占用、单批延迟和吞吐量。
  6. 检查不同数据子群的性能,并进行错误样本分析。
  7. 只有在离线结果、稳定性和部署成本都满足要求后,才考虑进入线上灰度。

判断:它的真正价值不在“击败所有树模型”

**Kumo Tabular 的真正价值,是把表格预测从一次性训练任务推进成可复用的模型能力。**这比“在某个公开数据集上刷新多少个百分点”更值得关注。

表格领域长期存在一个现实矛盾:传统模型已经足够强,但每换一个任务就要重新做一遍数据处理、特征工程和训练;深度模型拥有更强的统一建模潜力,却经常需要更多数据和更复杂的调参。Kumo Tabular 选择的路径,是用预训练换取下游任务的适配效率。

它能否建立长期优势,取决于三个问题。第一,官方评测中的精度优势能否在真实企业数据上复现;第二,模型推理成本能否控制在传统模型可接受的范围内;第三,面对脏数据、缺失值、类别漂移和长尾标签时,模型是否足够稳定。

截至 2026 年 9 月 29 日,Kumo Tabular 更适合被视为表格基础模型方向的一次重要公开落地,而不是已经终结传统表格机器学习的最终答案。对于开发者,它值得加入基准测试;对于数据团队,它适合用于少样本探索和多任务筛选;对于生产系统,则仍然需要与 LightGBM、CatBoost 以及现有特征工程体系进行严格对照。

如果它能够在不显著增加推理成本的前提下,稳定减少新任务的标注量和训练周期,那么“一个模型服务多张表”就不再只是研究论文里的概念,而可能成为企业结构化数据建模的常规工作流。

相关推荐

查看全部