El modelo más reciente del benchmark AIM-Tabular es Kumo-Tabular.
AIM Tabular Index
v1.0Probamos 13 modelos tabulares en 19 conjuntos de datos y puntuamos 1.200 ejecuciones de validación cruzada. El AIM Tabular Index combina el benchmark de modelos tabulares de AIMultiple con dos clasificaciones públicas, TabArena y BeyondArena. Consulte AIM Tabular Index Methodology para más detalles.
Clasificación
Los modelos tabulares con mayor puntuación en todos los benchmarks, ordenados por Elo combinado.
# | Rango | Modelo | AIM Tabular Index | AIM-Tabular | TabArena | BeyondArena | Segundos por fold | Coste de GPU, ejecución completa (USD) | Licencia |
|---|---|---|---|---|---|---|---|---|---|
| 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 | No comercial |
| 3 | 1-6 | TabPFN-3.5 Prior Labs | 16191558-1692 | 1836 | 1645 | 1376 | 4.3 | 0.68 | No comercial |
| 4 | 1-6 | TabFM Google Research | 15461451-1647 | 1798 | 1566 | 1274 | 173.6 | 27.40 | No comercial |
| 5 | 2-6 | Causilo Nums AI | 15201467-1585 | 1665 | 1569 | 1327 | 1.4 | 0.22 | No 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 | No 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
El AIM Tabular Index combina el benchmark de modelos tabulares de AIMultiple con dos clasificaciones públicas, TabArena y BeyondArena.
Puntuación
Cada benchmark puntúa los modelos en Elo. TabArena fija el RandomForest por defecto en 1000 Elo, BeyondArena fija el XGBoost por defecto en 1000 y AIM-Tabular fija la media de sus modelos en 1000. Desplazamos cada benchmark para que XGBoost por defecto obtenga 1000 y después promediamos las puntuaciones desplazadas para obtener el índice. En un mismo benchmark, una ventaja de 200 puntos predice una victoria en unos 3 de cada 4 conjuntos de datos.
Intervalos y posición
Cada benchmark da a cada modelo un intervalo de confianza del 95 %, y promediamos esos intervalos igual que las puntuaciones. Solo ordenamos dos modelos cuando sus intervalos no se solapan. Si se solapan, la posición se muestra como un rango. Por ejemplo, 1-4 significa que el modelo puede quedar en cualquier puesto del primero al cuarto.
Configuración
Los benchmarks públicos aportan sus puntuaciones con la configuración por defecto. AIM-Tabular ejecuta cada modelo sin ajuste ni fine-tuning. En TabArena, el ajuste sube XGBoost 126 puntos por encima de su valor por defecto.
Velocidad y coste
AIM-Tabular informa de los segundos por fold (tiempo medio de entrenamiento y predicción en un fold de validación cruzada) y del coste de GPU (tiempo total de los folds a 5,98 $ por hora de GPU). Los modelos fundacionales se ejecutaron en una GPU NVIDIA B200. Los modelos clásicos se ejecutaron en CPU y no tienen coste de GPU.
Benchmarks que usamos
Resultados sin procesar de 13 puntos de datos de los benchmarks de AIMultiple en los que se basa este índice. Descárguelos como un archivo ZIP que contiene un archivo CSV y un README.
¿Quieres los datos granulares que hay detrás? Únete a Premium
Registro de cambios
4 actualizaciones- v1.0
5 modelos entraron en el índice: Causilo, Kumo-Tabular, LimiX-2, Mitra-v2, TabPFN-3.5.
- v1.0
7 modelos entraron en el índice: CatBoost, LightGBM, Logistic regression, TabFM, TabICLv2 y 2 más.
- v1.0
Se añadió Mitra al benchmark; aún no se clasifica en el índice.
- v1.0
AIM-Tabular: Se reemplazó TabPFN por TabPFN 3 en la lista de modelos evaluados.