NVIDIA Kumo Tabular 是一個值得試跑的公開下載單表格模型,但以目前證據來看,還不足以取代 CatBoost、LightGBM 或 TabPFN,直接用於正式環境。
實際結論:可以試跑 Kumo Tabular,但先別切換正式流量
如果資料本來就集中在單一表格、手上也有標籤,而且想確認 in-context prediction 能否省下建模工作,Kumo Tabular 值得納入測試。至於嚴格延遲要求、只能使用 CPU 的服務環境、託管式部署,或需要處理多張關聯表格的情境,應該先視為測試門檻,不要預設 Kumo Tabular 一定符合需求。
官方權重與 API 文件都已公開,但目前針對 Kumo 的公開比較仍然有限。是否適合你的場景,還是要用自己的 holdout 資料驗證。
NVIDIA 實際釋出了什麼
NVIDIA 的 Kumo-Tabular model card 將它描述為一個支援分類與迴歸的預訓練模型。官方的 KumoTabular API 文件則展示了 in-context 工作流程:以帶有標籤的資料列提供上下文,接著直接對查詢資料列產生預測,不必針對每個資料集重新訓練權重。
公開套件提供 small、medium 與 large 三種變體。NVIDIA 的 structured-data model catalog 顯示,這些變體的參數量約從 2,700 萬到 2.16 億不等。模型卡列出 OpenMDW-1.1 授權條款,並提供 Python 安裝與推論流程。若要商業部署或重新散布,仍應仔細檢查授權內容,不要把「公開權重」直接視為可以無條件使用。
不要把這次釋出內容與 KumoRFM 混為一談。Kumo Tabular 的目標是單一表格;NVIDIA 的 Kumo Relational overview則將 KumoRFM 描述為另一個針對關聯式資料設計的模型。Kumo Tabular 文件也明確標示,目前不提供相關表格支援。
| 問題 | Kumo Tabular 的答案 |
|---|---|
| 主要任務 | 分類與迴歸 |
| 輸入形式 | 包含上下文與查詢資料列的單一表格 |
| 是否需要針對每個資料集訓練 | 使用 in-context prediction 流程時不需要 |
| 相關表格 | 公開的 KumoTabular API 不支援 |
| 模型權重 | 公開於 Hugging Face |
| 模型卡所列授權 | OpenMDW-1.1 |
| 託管式推論 | 目前沒有列出 Hugging Face Inference Provider |
本文檢視的官方模型卡與 API 頁面,沒有公布簡單明確的最大資料列數或最大特徵數。不過,文件提供了模型設定與任務限制,因此在試跑期間,應實際記錄資料列數、特徵數、上下文與查詢資料的組成、記憶體使用量,以及可接受的類別數,不要直接套用其他表格模型的規格。
決定它是否適用的三個限制
Kumo Tabular 是否適合你的環境,主要取決於三項營運限制:
- 資料形狀:Kumo Tabular 無法原生處理關聯表格。如果預測欄位分散在客戶、訂單、產品與客服事件等資料表中,應先定義並驗證攤平或彙總規則,再進行模型比較。
- 推論規模:公開文件說明了模型變體與支援的任務,但沒有提供經獨立驗證的正式環境吞吐量、GPU 記憶體或單次預測成本表。你需要實測上下文大小對延遲與記憶體的影響。
- 部署方式:模型權重與 Python 介面雖然公開,但這不等於具備 SLA 的託管式 API。如果團隊需要區域路由、配額保證或由供應商代管的營運模式,應把這些需求明確列為試跑門檻。
Kumo Tabular 與 TabPFN、傳統訓練模型的比較方式
根據本文檢視的來源,目前沒有可靠的公開 apples-to-apples 結果,能證明 Kumo Tabular 勝過現行的 TabPFN、CatBoost、LightGBM 或 XGBoost。以下引用的第三方基準測試適合用來設計試跑,不應拿來宣稱 Kumo 的效能。
| 工作負載 | 第一組比較對象 | 評估目的 |
|---|---|---|
| 乾淨的單一表格、小型或中型標籤資料集 | Kumo Tabular 對上 TabPFN | 在相同切分下比較兩者以表格為核心的 in-context prediction |
| 類別特徵占比較高的表格 | Kumo Tabular 對上 CatBoost | 以 CatBoost 作為訓練式類別資料基準 |
| 穩定且重複執行的批次評分 | Kumo Tabular 對上 LightGBM 或 XGBoost | 比較 in-context 模型與訓練式模型的服務表現 |
| 訊號分散在多張關聯表格中 | 攤平表格基準對上關聯式方法 | 衡量所採用的彙總方式是否遺失有用結構 |
| 快速探索資料欄位與 schema | 試跑 Kumo Tabular | 測試省略特定任務訓練迴圈是否能節省實際時間 |
AnoFox 的比較在一台 8-core CPU 上測試了數個表格基礎模型,結果顯示它們在五個小型資料集上的表現,比受測的 scikit-learn 模型高出約 2–7%,但執行時間可能高出許多。其中一次 churn 測試中,Mitra 的 warm inference 花了 19.3 秒,logistic regression 則是 0.035 秒。
另一項 AIMultiple 基準測試顯示,TabFM 在 19 個資料集中的 15 個拿下最佳結果,但所需的運算量約為 TabPFN 3 或 TabICLv2 的 40 倍。該測試估算 TabFM 完整執行一次在 B200 GPU 上約需 $27,相較之下 TabPFN 3 約為 $0.65。再次強調,這些數字應該被視為需要蒐集的測量項目,而不是 Kumo 的測試結果。
一套站得住腳的第一次測試
使用固定的 holdout,讓 Kumo Tabular 和訓練式基準正面比較,看看它是否值得留下。
- 在嘗試任何模型前,先固定一組符合時間順序或分層抽樣的 train/test split。預測時不要讓模型接觸測試標籤。
- 先驗證表格 schema:目標欄位、缺失值、類別欄位、重複實體,以及任何在預測時間點之後才產生的欄位。
- 在原始且受支援的表格上執行 Kumo Tabular,記錄模型變體、上下文與查詢資料列、特徵數、可接受的類別數、批次大小、硬體、冷啟動時間、warm latency,以及峰值記憶體。
- 使用相同資料列與目標欄位執行 CatBoost 或 LightGBM。預處理時間應與訓練、預測時間分開記錄。
- 選擇符合任務的指標進行比較:二元分類使用 AUROC 與校準度;不平衡多分類使用 macro-F1;迴歸則使用 MAE 或 RMSE。
- 以較小與較大的上下文樣本重複測試。如果品質維持穩定,但延遲大幅上升,這個模型可能更適合探索,而不是正式服務。
- 事先定義三項通過標準:最低指標提升幅度、最高 p95 延遲,以及每批次最高基礎架構成本。只有在 Kumo Tabular 三項都達標時,才進入下一階段。
不要把供應商基準測試當成檢查資料洩漏的替代品。「不需要特徵工程」不代表「不需要驗證資料」。