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.09k stars 95 forks source link

[ENH] Add Adversarial Training as an Ensemble Method #18

Closed xuyxu closed 3 years ago

xuyxu commented 3 years ago

Reference: Algorithm 1 in the paper "Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles" URL: http://papers.neurips.cc/paper/7219-simple-and-scalable-predictive-uncertainty-estimation-using-deep-ensembles.pdf

TODO: