skip to main content
Idioma:

Better trees: an empirical study on hyperparameter tuning of classification decision tree induction algorithms

Gomes Mantovani, Rafael ; Horváth, Tomáš ; Rossi, André L. D. ; Cerri, Ricardo ; Barbon Junior, Sylvio ; Vanschoren, Joaquin ; Carvalho, André C. P. L. F. de

Data mining and knowledge discovery, 2024-05, Vol.38 (3), p.1364-1416 [Periódico revisado por pares]

New York: Springer US

Texto completo disponível

Citações Citado por
  • Título:
    Better trees: an empirical study on hyperparameter tuning of classification decision tree induction algorithms
  • Autor: Gomes Mantovani, Rafael ; Horváth, Tomáš ; Rossi, André L. D. ; Cerri, Ricardo ; Barbon Junior, Sylvio ; Vanschoren, Joaquin ; Carvalho, André C. P. L. F. de
  • Assuntos: Algorithms ; Artificial Intelligence ; Chemistry and Earth Sciences ; Classification ; Computer Science ; Configurations ; Data Mining and Knowledge Discovery ; Datasets ; Decision trees ; Information Storage and Retrieval ; Legislation ; Machine learning ; Optimization techniques ; Performance prediction ; Physics ; Statistics for Engineering ; Tuning
  • É parte de: Data mining and knowledge discovery, 2024-05, Vol.38 (3), p.1364-1416
  • Descrição: Machine learning algorithms often contain many hyperparameters whose values affect the predictive performance of the induced models in intricate ways. Due to the high number of possibilities for these hyperparameter configurations and their complex interactions, it is common to use optimization techniques to find settings that lead to high predictive performance. However, insights into efficiently exploring this vast space of configurations and dealing with the trade-off between predictive and runtime performance remain challenging. Furthermore, there are cases where the default hyperparameters fit the suitable configuration. Additionally, for many reasons, including model validation and attendance to new legislation, there is an increasing interest in interpretable models, such as those created by the decision tree (DT) induction algorithms. This paper provides a comprehensive approach for investigating the effects of hyperparameter tuning for the two DT induction algorithms most often used, CART and C4.5. DT induction algorithms present high predictive performance and interpretable classification models, though many hyperparameters need to be adjusted. Experiments were carried out with different tuning strategies to induce models and to evaluate hyperparameters’ relevance using 94 classification datasets from OpenML. The experimental results point out that different hyperparameter profiles for the tuning of each algorithm provide statistically significant improvements in most of the datasets for CART, but only in one-third for C4.5. Although different algorithms may present different tuning scenarios, the tuning techniques generally required few evaluations to find accurate solutions. Furthermore, the best technique for all the algorithms was the Irace. Finally, we found out that tuning a specific small subset of hyperparameters is a good alternative for achieving optimal predictive performance.
  • Editor: New York: Springer US
  • Idioma: Inglês
Voltar para lista de resultados

  • Realização: ABCD-USP USP   Logos de Redes Sociais: Freepik

Buscando em bases de dados remotas. Favor aguardar.