Differentiable Model Selection for Ensemble Learning
Differentiable Model Selection for Ensemble Learning
复制标题
DOI:
10.24963/ijcai.2023/217
复制
发表时间:
2022-11
期刊:
影响因子:
--
通讯作者:
James Kotary;Vincenzo Di Vito;Ferdinando Fioretto
中科院分区:
文献类型:
--
作者:
James Kotary;Vincenzo Di Vito;Ferdinando Fioretto
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.