NVIDIA Kumo Tabular estabelece uma nova fronteira de precisão e eficiência para previsão tabular
huggingface ·
A NVIDIA lançou o Kumo Tabular, um modelo fundamental de código aberto projetado para previsão em dados tabulares que opera sem treinamento ou engenharia de recursos. Essa arquitetura baseada em Transformer utiliza aprendizado in-contexto para prever rótulos de novas linhas em uma única passagem direta, suportando tarefas tanto de classificação quanto de regressão. O modelo foi pré-treinado exclusivamente em tabelas artificiais geradas por Modelos Causais Estruturais, permitindo lidar com diversas imperfeições nos dados, como valores ausentes e outliers. Está disponível em três tamanhos, variando de 28 milhões a 215 milhões de parâmetros, e roda por meio da biblioteca de código aberto structured-data-models da NVIDIA. Avaliações de desempenho mostram que o Kumo Tabular ocupa a primeira posição em benchmarks principais, incluindo TabArena, BeyondArena, TALENT e ScoringBench. Notavelmente, ele estabelece um novo estado da arte na frente de Pareto entre precisão e eficiência, enquanto roda significativamente mais rápido do que modelos concorrentes como o LimiX-2. Embora atualmente limitado a colunas numéricas e categóricas, o modelo oferece tratamento robusto de estruturas complexas de tabelas por meio de mecanismos de atenção especializados. Lançado sob a licença OpenMDW-1.1, ele fornece aos desenvolvedores acesso imediato a capacidades de IA de alto desempenho para dados tabulares.
huggingface