Introduction aux algorithmes des arbres de décision

Les algorithmes des arbres de décision ont longtemps été une pierre angulaire de l'extraction de données et de l'apprentissage machine, offrant des modèles d'interprétation pour les tâches de classification et de régression. Parmi les plus utilisés sont C4.5, CART et CHEID. Chaque algorithme apporte une approche distincte pour construire des arbres, différent dans la façon dont ils divisent les données, manipulent différents types d'attributs, et gèrent l'ajustement excessif.

Fondements de l'arbre de décision

Un arbre de décision est une structure de type diagramme de flux où chaque noeud interne représente un test sur un attribut, chaque branche représente un résultat de ce test, et chaque noeud de feuille détient une étiquette de classe ou une prédiction numérique. L'arbre est construit de façon récursive en sélectionnant le meilleur attribut pour diviser les données à chaque noeud, en fonction d'une mesure d'impureté choisie. Les principales différences entre C4.5, CART, et CHAID résident dans leurs critères de division, topologie de l'arbre (coupures binaires contre multidirectionnelles), capacité à gérer différents types de données, et stratégies de taille.

L'algorithme C4.5

Historique et développement

Développé par Ross Quinlan comme successeur de ID3, C4.5 est l'un des algorithmes de décision les plus influents de la littérature. Il a été conçu pour surmonter plusieurs limites de son prédécesseur, notamment dans la manipulation des attributs continus, des valeurs manquantes et de la taille des arbres. L'algorithme adopte une recherche descendante et gourmande dans l'espace des arbres possibles et utilise un critère de division basé sur le rapport de gain d'information.

Critère de division : Rapport de gain d'information

C4.5 utilise le rapport de gain d'information pour décider sur quel attribut diviser. Le gain d'information est dérivé de l'entropie, mesure de l'impureté de la théorie de l'information. Cependant, le gain d'information tend à favoriser les attributs avec de nombreuses valeurs distinctes (forte cardinalité). Pour corriger ce biais, Quinlan a introduit le rapport de gain, qui normalise le gain d'information par l'information intrinsèque de la fraction. L'attribut avec le rapport de gain le plus élevé est sélectionné.

Manipulation des attributs continus

Par exemple, si un attribut a des valeurs 1, 3, 5, 7, l'algorithme peut tester des scissions comme ≤3 vs >3, ≤5 vs >5, et ainsi de suite, choisir celui qui maximise le rapport de gain. Ce processus est répété à chaque nœud, ce qui rend C4.5 capable de manipuler des types de données mixtes sans discrétisation.

Valeurs manquantes et taille

C4.5 gère les valeurs d'attribut manquantes dans l'entraînement et la prédiction. Lorsqu'une valeur d'attribut est manquante, l'algorithme utilise une approche probabiliste, distribuant l'instance entre les branches proportionnellement à la distribution observée dans les données d'entraînement. Pour la prédiction, les valeurs inconnues sont traitées de la même façon en utilisant les mêmes probabilités. Pour éviter tout surajustement, C4.5 utilise une méthode post-élagage appelée élagage basé sur des erreurs. À partir des nœuds foliaires, il remplace un sous-arbre par une feuille si le taux d'erreur estimé n'augmente pas.

Principales forces et limites

Le C4.5 est très interprétable et produit souvent des arbres plus petits et plus précis que ses prédécesseurs. Il supporte à la fois la classification et la régression (par l'intermédiaire de la variante M5) et fonctionne bien avec des données hétérogènes. Cependant, il peut être calculablement coûteux pour de très grands ensembles de données en raison de sa recherche dynamique de seuil.

Pour plus de détails sur le C4.5, voir l'ouvrage original de Quinlan : C4.5 : Programmes d'apprentissage automatique.

Le PANIER Algorithme

Historique et développement

Leo Breiman, Jérôme Friedman, Richard Olshen et Charles Stone ont introduit les arbres de la classification et de la régression (CART) dans leur livre de 1984. Contrairement à C4.5, le CART produit des arbres strictement binaires, ce qui signifie que chaque fraction divise le nœud en deux nœuds d'enfant exactement. Cette nature binaire simplifie de nombreux aspects de la construction et de l'interprétation des arbres.

Critère de division: Impureté de Gini

Pour les tâches de classification, le CART utilise la mesure Gini impurity pour sélectionner la meilleure division. L'impureté de Gini quantifie la probabilité de classification erronée d'un élément choisi au hasard s'il était étiqueté selon la distribution des étiquettes de classe dans le nœud. Elle est calculée comme où p i est la proportion de classe i. Un indice de Gini inférieur indique un noeud plus homogène. Pour la régression, le CART utilise l'écart le moins élevé des carrés[ (réduction de la variation) comme critère de division. L'algorithme évalue toutes les divisions possibles pour chaque attribut, à la fois en fonction du seuil pour les variables continues et les combinaisons de catégories pour les variables catégoriques, et choisit celui qui minimise le plus l'impureté.

Structure et taille des arbres

Parce que CART construit des arbres binaires, il peut créer plusieurs scissions sur le même attribut le long de différentes branches, manipulant efficacement des interactions non linéaires. Après avoir construit un grand arbre qui surpasse les données, CART applique la taille de complexité des coûts[. Cette méthode introduit un paramètre de complexité (α) qui pénalise la taille des arbres. L'algorithme génère une séquence de sous-arbres imbriqués et sélectionne celui avec la plus petite erreur validée croisée. Cette technique de taille est particulièrement robuste et est souvent considérée comme un repère pour d'autres algorithmes.

Traitement des types de données et des valeurs manquantes

Pour les variables catégorisées avec de nombreuses catégories, il peut évaluer toutes les partitions binaires possibles des catégories. Les valeurs manquantes sont traitées en utilisant scissions de substitution[: lorsque l'attribut de fraction primaire est manquant, l'algorithme utilise l'attribut de substitution le mieux corrélé pour décider de la direction de l'instance. Cette approche préserve bien les données et maintient la puissance prédictive même avec des enregistrements incomplets.

Principales forces et limites

Le CART est très robuste et efficace sur le plan informatique pour les ensembles de données de taille modérée. Ses fractions binaires réduisent la fragmentation des données par rapport aux fractions multi-voies. La manipulation intégrée des valeurs manquantes par les substituts est un avantage majeur dans les données du monde réel. Cependant, le CART peut produire des arbres plus profonds que nécessaire, et l'algorithme peut être biaisé vers des attributs avec des valeurs plus distinctes si pas correctement régularisées.

Pour une compréhension plus approfondie, voir le texte classique de Breiman et al.: Tariers de classification et de régression.

L'algorithme du CHEID

Historique et développement

Le CHAID (Chi-squared Automatic Interaction Detector) a été développé par Gordon V. Kass en 1980 comme technique de segmentation et de classification. Contrairement au C4.5 et au CART, le CHAID utilise un test de signification statistique – en particulier le test d'indépendance chi-square – pour décider des scissions, ce qui le rend particulièrement adapté aux données catégoriques et aux applications de recherche de marché où il est important de comprendre les interactions entre les variables.

Critère de division: Essais de Chi-Square

Le CHAID examine chaque variable prédicteur et fusionne des catégories qui ne sont pas significativement différentes par rapport à la variable cible, en se basant sur un test chi carré (pour les cibles nominales) ou un test F (pour les cibles ordinales). Il sélectionne ensuite le prédicteur qui produit la fraction la plus significative, c'est-à-dire la plus petite valeur p. Ce processus permet de s'assurer que l'arbre résultant ne fait que des fractions statistiquement justifiables. L'algorithme supporte les fractions multidirectionnelles, ce qui signifie qu'un prédicteur catégorique peut être divisé en plusieurs groupes, chacun contenant une ou plusieurs catégories originales qui sont semblables dans leur relation à la cible.

Manipulation des données et de la construction des arbres

Le CHAID est conçu principalement pour les tâches de classification avec des prédicteurs numériques catégoriques ou discrétés. Bien qu'il puisse gérer des variables continues, elles sont généralement regroupées en catégories avant l'analyse. L'algorithme n'exige pas de définition manuelle des catégories; il fusionne automatiquement des bacs adjacents basés sur des tests statistiques. Les valeurs manquantes peuvent être traitées comme une catégorie distincte ou imputées en utilisant le mode. La construction d'arbres s'arrête lorsqu'aucune autre division significative n'est trouvée selon un niveau de signification spécifié par l'utilisateur (souvent α = 0,05).

Principales forces et limites

La principale force du CHAID est sa rigueur statistique, qui le rend idéal pour l'analyse exploratoire et les tests d'hypothèse dans des domaines comme le marketing, la sociologie et les soins de santé. Les scissions multidirectionnels produisent souvent des arbres plus clairs et faciles à interpréter. Parce qu'il fusionne automatiquement des catégories non significatives, l'arbre peut révéler des regroupements naturels dans les données. Cependant, le CHAID est moins adapté aux tâches de régression (bien qu'il existe une extension appelée CHAID pour la régression).

Pour référence sur le CHAID, voir: Une technique exploratoire pour enquêter sur les grandes quantités de données catégoriques (Kass, 1980).

Analyse comparative des principales caractéristiques

Le tableau suivant résume les différences les plus importantes entre C4.5, CART et CHAID.

Feature C4.5 CART CHAID
Splitting Criterion Information gain ratio Gini impurity (classification), variance reduction (regression) Chi-square test (classification), F-test (ordinal)
Tree Structure Multi-way splits possible Binary splits only Multi-way splits (auto-merging categories)
Supported Target Types Categorical (classification), continuous (with modifications) Categorical and continuous Primarily categorical; continuous via binning
Handling Continuous Predictors Dynamic threshold search Dynamic threshold search Bin into categories (user-defined or automatic)
Missing Values Probabilistic distribution Surrogate splits Treated as separate category or mode imputation
Pruning Method Error-based pruning Cost-complexity pruning Stopping rule via significance level (no explicit pruning)
Scalability Moderate; expensive for large numeric datasets Good for moderate-sized datasets Slower with many categories
Interpretability High (often compact trees) High (binary splits easy to follow) High (statistically justified splits)
Overfitting Control Strong via pruning Strong via cost-complexity pruning Moderate; controlled by significance threshold

Au-delà de ces différences techniques, les algorithmes varient aussi dans la façon dont ils traitent les interactions de fonctionnalités. Les scissions binaires du CART lui permettent de modéliser des interactions complexes qui peuvent nécessiter une division répétée sur le même attribut. Les scissions multi-voies du CHAID peuvent capturer les interactions directement en une seule scission si les catégories fusionnées reflètent une interaction avec la cible.

Lignes directrices pour la sélection de l'algorithme

Choisir le bon algorithme d'arbre de décision dépend des caractéristiques spécifiques de votre ensemble de données et des objectifs de votre analyse. Utilisez les lignes directrices suivantes:

  • Choisissez C4.5 quand: Vous avez besoin d'un algorithme polyvalent qui gère les données continues et catégoriques, les valeurs manquantes sont présentes, et vous voulez un arbre facile à interpréter. C4.5 est un bon choix par défaut pour de nombreuses tâches de classification.
  • Choisissez CART lorsque: Vous avez besoin d'un algorithme robuste pour la classification et la régression, vos données comprennent de nombreuses valeurs manquantes, ou vous préférez la simplicité des fractions binaires. Les fractions de substitution de CART sont puissantes pour les données du monde réel avec une absence de motif.
  • Choisissez CHAID quand: Votre intérêt principal est d'explorer les relations entre les variables catégorisées, vous avez besoin d'un arbre statistiquement justifié, ou vous voulez fusionner automatiquement des catégories pour réduire la dimensionnalité.

C4.5 et CART produisent souvent des arbres plus profonds qui peuvent nécessiter une taille soigneuse, alors que la règle d'arrêt basée sur la signification du CHAID tend à produire des arbres plus faibles. Si les ressources de calcul sont limitées, CART est généralement plus rapide que C4.5 pour les grands ensembles de données numériques. Pour des attributs catégoriques de très haute cardinalité, le CHAID peut être lent en raison des calculs chi-carré; une bonne alternative peut être aux catégories de bin avant d'appliquer un autre algorithme.

Considérations pratiques de mise en œuvre

Les trois algorithmes sont disponibles dans les outils d'extraction de données populaires et les bibliothèques de programmation. C4.5 est implémenté dans Weka (comme J48), tandis que CART est disponible dans R (paquet rpart), Python (scikit-learn's DecisionTreeClassifier with par défaut Gini), et beaucoup d'autres plateformes. CHAID est implémenté dans SPSS et dans R (paquet RAID). Lors de la mise en œuvre de ces modèles, attention aux hyperparamètres: pour C4.5, le facteur de confiance dans le taillement affecte la profondeur des arbres; pour CART, le paramètre de complexité (cp) contrôle la taille; pour CHAID, le niveau de signification et la taille minimale des feuilles empêchent le surajustement.

Conclusion

C4.5 excelle avec son rapport de gain d'information, sa capacité à gérer des données continues et manquantes, et la taille basée sur les erreurs. Le CTA fournit un cadre binaire robuste avec la taille d'impureté et de complexité de coûts de Gini, ce qui le rend idéal pour les tâches de classification et de régression. Le CTAID apporte une rigueur statistique par le biais de tests chi-carré et de fusion automatique de catégorie, particulièrement adapté pour l'analyse exploratoire des données catégoriques. Comprendre les différences dans les critères de division, la structure des arbres et la manipulation des données permet aux praticiens de choisir l'algorithme le plus approprié pour leur problème. En alignant les forces de l'algorithme avec les caractéristiques de l'ensemble de données, on peut construire des modèles efficaces et interprétables qui fournissent des informations exploitables.