A repulsive last-layer ensemble trained with function-space diversity on OOD or augmented samples gives competitive uncertainty estimates at a fraction of deep-ensemble cost.
Stein variational gradient descent: A general purpose Bayesian inference algorithm,
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
citation-role summary
method 1
citation-polarity summary
fields
cs.LG 1years
2024 1verdicts
CONDITIONAL 1roles
method 1polarities
support 1representative citing papers
citing papers explorer
-
Function Space Diversity for Uncertainty Prediction via Repulsive Last-Layer Ensembles
A repulsive last-layer ensemble trained with function-space diversity on OOD or augmented samples gives competitive uncertainty estimates at a fraction of deep-ensemble cost.