O modelo mais recente do benchmark AIM-Tabular é Kumo-Tabular.
AIM Tabular Index
v1.0Testamos 13 modelos tabulares em 19 conjuntos de dados e pontuamos 1.200 execuções de validação cruzada. O AIM Tabular Index combina o benchmark de modelos tabulares da AIMultiple com dois rankings públicos, TabArena e BeyondArena. Consulte AIM Tabular Index Methodology para mais detalhes.
Ranking
Os modelos tabulares com maior pontuação em todos os benchmarks, ordenados pelo Elo combinado.
# | Faixa | Modelo | AIM Tabular Index | AIM-Tabular | TabArena | BeyondArena | Segundos por fold | Custo de GPU, execução completa (USD) | Licença |
|---|---|---|---|---|---|---|---|---|---|
| 1 | 1-4 | Kumo-Tabular NVIDIA | 16921610-1794 | 1911 | 1739 | 1426 | 7.6 | 1.19 | OpenMDW-1.1 |
| 2 | 1-6 | LimiX-2 StableAI | 16651571-1774 | 1976 | 1732 | 1288 | 111.9 | 17.65 | Não comercial |
| 3 | 1-6 | TabPFN-3.5 Prior Labs | 16191558-1692 | 1836 | 1645 | 1376 | 4.3 | 0.68 | Não comercial |
| 4 | 1-6 | TabFM Google Research | 15461451-1647 | 1798 | 1566 | 1274 | 173.6 | 27.40 | Não comercial |
| 5 | 2-6 | Causilo Nums AI | 15201467-1585 | 1665 | 1569 | 1327 | 1.4 | 0.22 | Não comercial |
| 6 | 2-8 | Mitra-v2 Amazon | 14981410-1600 | 1450 | 1546 | - | 1.5 | 0.24 | Apache-2.0 |
| 7 | 6-8 | TabPFN 3 Prior Labs | 13621298-1432 | 1434 | 1417 | 1234 | 4.2 | 0.66 | Não comercial |
| 8 | 6-8 | TabICLv2 Inria SODA | 13401269-1414 | 1495 | 1352 | 1172 | 3.1 | 0.48 | BSD-3-Clause |
| 9 | 9-10 | CatBoost Yandex | 11341081-1188 | 1133 | 1151 | 1118 | 0.6 | CPU | Apache-2.0 |
| 10 | 9-11 | LightGBM Microsoft | 1037987-1090 | 1146 | 975 | 990 | 0.5 | CPU | MIT |
AIM Tabular Index Methodology
O AIM Tabular Index combina o benchmark de modelos tabulares da AIMultiple com dois rankings públicos, TabArena e BeyondArena.
Pontuação
Cada benchmark classifica os modelos em Elo. O TabArena fixa o RandomForest padrão em 1000 Elo, o BeyondArena fixa o XGBoost padrão em 1000 e o AIM-Tabular fixa a média dos seus modelos em 1000. Deslocamos cada benchmark para que o XGBoost padrão obtenha 1000 e depois calculamos a média das pontuações deslocadas para obter o índice. Num mesmo benchmark, uma vantagem de 200 pontos prevê uma vitória em cerca de 3 em cada 4 conjuntos de dados.
Intervalos e posição
Cada benchmark dá a cada modelo um intervalo de confiança de 95%, e calculamos a média desses intervalos tal como das pontuações. Só ordenamos dois modelos quando os seus intervalos não se sobrepõem. Caso contrário, a posição aparece como um intervalo. Por exemplo, 1-4 significa que o modelo pode ficar em qualquer lugar entre o primeiro e o quarto.
Configuração
Os benchmarks públicos contribuem com as suas pontuações na configuração padrão. O AIM-Tabular executa cada modelo sem ajuste nem fine-tuning. No TabArena, o ajuste eleva o XGBoost 126 pontos acima do seu valor padrão.
Velocidade e custo
O AIM-Tabular reporta os segundos por fold (tempo médio de treino e previsão num fold de validação cruzada) e o custo de GPU (tempo total dos folds a 5,98 $ por hora de GPU). Os modelos fundacionais correram numa GPU NVIDIA B200. Os modelos clássicos correram em CPU e não têm custo de GPU.
Benchmarks que usamos
Resultados brutos de 13 pontos de dados dos benchmarks da AIMultiple em que este índice se baseia. Baixe-os como um arquivo ZIP contendo um arquivo CSV e um README.
Quer os dados granulares por trás disso? Assine o Premium
Registro de alterações
4 atualizações- v1.0
5 modelos entraram no índice: Causilo, Kumo-Tabular, LimiX-2, Mitra-v2, TabPFN-3.5.
- v1.0
7 modelos entraram no índice: CatBoost, LightGBM, Logistic regression, TabFM, TabICLv2 e mais 2.
- v1.0
Mitra foi adicionado ao benchmark; ainda não classificado no índice.
- v1.0
AIM-Tabular: Substituído o TabPFN pelo TabPFN 3 na lista de modelos avaliados.