English

Diversifying Deep Ensembles: A Saliency Map Approach for Enhanced OOD Detection, Calibration, and Accuracy

Computer Vision and Pattern Recognition 2024-11-06 v5 Artificial Intelligence Machine Learning

Abstract

Deep ensembles are capable of achieving state-of-the-art results in classification and out-of-distribution (OOD) detection. However, their effectiveness is limited due to the homogeneity of learned patterns within ensembles. To overcome this issue, our study introduces Saliency Diversified Deep Ensemble (SDDE), a novel approach that promotes diversity among ensemble members by leveraging saliency maps. Through incorporating saliency map diversification, our method outperforms conventional ensemble techniques and improves calibration in multiple classification and OOD detection tasks. In particular, the proposed method achieves state-of-the-art OOD detection quality, calibration, and accuracy on multiple benchmarks, including CIFAR10/100 and large-scale ImageNet datasets.

Keywords

Cite

@article{arxiv.2305.11616,
  title  = {Diversifying Deep Ensembles: A Saliency Map Approach for Enhanced OOD Detection, Calibration, and Accuracy},
  author = {Stanislav Dereka and Ivan Karpukhin and Maksim Zhdanov and Sergey Kolesnikov},
  journal= {arXiv preprint arXiv:2305.11616},
  year   = {2024}
}