Le modèle le plus récent du benchmark AIM-Tabular est Kumo-Tabular.
AIM Tabular Index
v1.0Nous avons testé 13 modèles tabulaires sur 19 jeux de données et noté 1 200 exécutions de validation croisée. L'AIM Tabular Index réunit le benchmark des modèles tabulaires d'AIMultiple et deux classements publics, TabArena et BeyondArena. Consultez AIM Tabular Index Methodology pour plus de détails.
Classement
Les modèles tabulaires les mieux notés sur l'ensemble des benchmarks, classés par Elo combiné.
# | Plage | Modèle | AIM Tabular Index | AIM-Tabular | TabArena | BeyondArena | Secondes par fold | Coût GPU, exécution complète (USD) | Licence |
|---|---|---|---|---|---|---|---|---|---|
| 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 | Non commercial |
| 3 | 1-6 | TabPFN-3.5 Prior Labs | 16191558-1692 | 1836 | 1645 | 1376 | 4.3 | 0.68 | Non commercial |
| 4 | 1-6 | TabFM Google Research | 15461451-1647 | 1798 | 1566 | 1274 | 173.6 | 27.40 | Non commercial |
| 5 | 2-6 | Causilo Nums AI | 15201467-1585 | 1665 | 1569 | 1327 | 1.4 | 0.22 | Non commercial |
| 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 | Non commercial |
| 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
L'AIM Tabular Index réunit le benchmark des modèles tabulaires d'AIMultiple et deux classements publics, TabArena et BeyondArena.
Notation
Chaque benchmark note les modèles en Elo. TabArena fixe le RandomForest par défaut à 1000 Elo, BeyondArena fixe le XGBoost par défaut à 1000 et AIM-Tabular fixe la moyenne de ses modèles à 1000. Nous décalons chaque benchmark pour que XGBoost par défaut obtienne 1000, puis nous faisons la moyenne des notes décalées pour obtenir l'indice. Sur un même benchmark, une avance de 200 points prédit une victoire sur environ 3 jeux de données sur 4.
Intervalles et rang
Chaque benchmark donne à chaque modèle un intervalle de confiance à 95 %, et nous faisons la moyenne de ces intervalles comme pour les notes. Nous ne départageons deux modèles que si leurs intervalles ne se chevauchent pas. Sinon, le rang s'affiche sous forme de plage. Par exemple, 1-4 signifie que le modèle peut se classer n'importe où entre la première et la quatrième place.
Configuration
Les benchmarks publics fournissent leurs notes en configuration par défaut. AIM-Tabular exécute chaque modèle sans réglage ni fine-tuning. Sur TabArena, le réglage fait gagner à XGBoost 126 points au-dessus de sa valeur par défaut.
Vitesse et coût
AIM-Tabular indique les secondes par fold (temps moyen d'entraînement et de prédiction sur un fold de validation croisée) et le coût GPU (temps total des folds à 5,98 $ par heure de GPU). Les modèles de fondation ont tourné sur un GPU NVIDIA B200. Les modèles classiques ont tourné sur CPU et n'ont pas de coût GPU.
Benchmarks utilisés
Résultats bruts de 13 points de données issus des benchmarks AIMultiple sur lesquels repose cet indice. Téléchargez-les sous forme de fichier ZIP contenant un fichier CSV et un README.
Vous voulez les données détaillées derrière ? Rejoindre Premium
Journal des modifications
4 mises à jour- v1.0
5 modèles sont entrés dans l'indice : Causilo, Kumo-Tabular, LimiX-2, Mitra-v2, TabPFN-3.5.
- v1.0
7 modèles sont entrés dans l'indice : CatBoost, LightGBM, Logistic regression, TabFM, TabICLv2 et 2 de plus.
- v1.0
Mitra a été ajouté au benchmark ; pas encore classé dans l'indice.
- v1.0
AIM-Tabular: Remplacement de TabPFN par TabPFN 3 dans la liste des modèles évalués.