Differentiable Model Selection for Ensemble Learning

James Kotary, Vincenzo Di Vito, Ferdinando Fioretto

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence

Main Track. Pages 1954-1962.

https://doi.org/10.24963/ijcai.2023/217

PDF BibTeX

Abstract

Model selection is a strategy aimed at creating accurate and robust models by identifying the optimal model for classifying any particular input sample. This paper proposes a novel framework for differentiable selection of groups of models by integrating machine learning and combinatorial optimization.

The framework is tailored for ensemble learning with a strategy that learns to combine the predictions of appropriately selected pre-trained ensemble models. It does so by modeling the ensemble learning task as a differentiable selection program trained end-to-end over a pretrained ensemble to optimize task performance. The proposed framework demonstrates its versatility and effectiveness, outperforming conventional and advanced consensus rules across a variety of classification tasks.

Keywords

  • Constraint Satisfaction and Optimization: CSO: Constraint optimization
  • Machine Learning: ML: Applications
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards