TorchEnsemble-Community / Ensemble-Pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.
https://ensemble-pytorch.readthedocs.io
BSD 3-Clause "New" or "Revised" License
1.07k stars 95 forks source link

Can we use multiple baseestimators? #49

Closed EternalConfession closed 3 years ago

EternalConfession commented 3 years ago

In the demo and DOCs, it seems that we can only use one type of model as basemodel.

Can we use different models as base estimators?

xuyxu commented 3 years ago

Hi @EternalConfession, I think you are referring to heterogeneous ensemble.

This feature request is still on-going (#40), mainly because training protocols on different base estimators could be quite different, and therefore needs careful design.

EternalConfession commented 3 years ago

Hi @EternalConfession, I think you are referring to heterogeneous ensemble.

This feature request is still on-going (#40), mainly because training protocols on different base estimators could be quite different, and therefore needs careful design.

Thanks for your reply.

Is there any related work on heterogeneous ensemble methods?

xuyxu commented 3 years ago

Here are two latest papers published in MLSys 2020, which are related to building heterogeneous ensembles on top of the neural network:

In my point of view, recent works tend to focus more on how to reduce the training costs on the ensemble, especially when using large neural networks as the base estimator. Research on novel ensemble algorithms is lacking in recent years. Correct me if you have other opinions ;-)

xuyxu commented 3 years ago

Closed due to inactivity.