Differentiable Model Selection for Ensemble Learning

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

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