English

Emergent Equivariance in Deep Ensembles

Machine Learning 2024-06-18 v2

Abstract

We show that deep ensembles become equivariant for all inputs and at all training times by simply using data augmentation. Crucially, equivariance holds off-manifold and for any architecture in the infinite width limit. The equivariance is emergent in the sense that predictions of individual ensemble members are not equivariant but their collective prediction is. Neural tangent kernel theory is used to derive this result and we verify our theoretical insights using detailed numerical experiments.

Keywords

Cite

@article{arxiv.2403.03103,
  title  = {Emergent Equivariance in Deep Ensembles},
  author = {Jan E. Gerken and Pan Kessel},
  journal= {arXiv preprint arXiv:2403.03103},
  year   = {2024}
}

Comments

11 pages + 17 pages appendices

R2 v1 2026-06-28T15:10:00.497Z