Joint leaf-refinement and ensemble pruning through L1 regularization

Ensembles are among the state-of-the-art in many machine learning applications. With the ongoing integration of ML models into everyday life, e.g., in the form of the Internet of Things, the deployment and continuous application of models become more and more an important issue. Therefore, small models that offer good predictive performance and use small amounts of memory are required. Ensemble pruning is a standard technique for removing unnecessary classifiers from a large ensemble that reduces the overall resource consumption and sometimes improves the performance of the original ensemble. Similarly, leaf-refinement is a technique that improves the performance of a tree ensemble by jointly re-learning the probability estimates in the leaf nodes of the trees, thereby allowing for smaller ensembles while preserving their predictive performance. In this paper, we develop a new method that combines both approaches into a single algorithm. To do so, we introduce L1
regularization into the leaf-refinement objective, which allows us to jointly prune and refine trees at the same time. In an extensive experimental evaluation, we show that our approach not only offers statistically significantly better performance than the state-of-the-art but also offers a better accuracy-memory trade-off. We conclude our experimental evaluation with a case study showing the effectiveness of our method in a real-world setting.

  • Published in:
    Data Mining and Knowledge Discovery
  • Type:
  • Authors:
    Buschjäger, Sebastian; Morik, Katharina
  • Year:

Citation information

Buschjäger, Sebastian; Morik, Katharina: Joint leaf-refinement and ensemble pruning through L1 regularization, Data Mining and Knowledge Discovery, 2023, 37, 1230--1261,, Buschjaeger.Morik.2023a,

Associated Lamarr Researchers

lamarr institute person Buschjager Sebastian - Lamarr Institute for Machine Learning (ML) and Artificial Intelligence (AI)

Dr. Sebastian Buschjäger

Scientific Coordinator Resource-aware ML to the profile
lamarr institute person Morik Katharina e1663924705259 - Lamarr Institute for Machine Learning (ML) and Artificial Intelligence (AI)

Prof. Dr. Katharina Morik

Founding Director to the profile