Emergent Equivariance in Deep Ensembles
Jan E. Gerken, Pan Kessel
2024Year
14Citations
6Top-tier citations
Abstract
We show that deep ensembles become equivariant for all inputs and at all training times by simply using full 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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f4a91e4d-fb41-4c79-ab70-b2bf2440ca10Cited by top-tier papers6
- Safety Depth in Large Language Models: A Markov Chain PerspectiveChing-Chia Kao, Chia-Mu Yu, Chun-Shien Lu, Chu-Song ChenNeurIPS 2025 · 2 citations
- How Ensembles of Distilled Policies Improve Generalisation in Reinforcement LearningMax Weltevrede, Moritz A. Zanger, Matthijs T. J. Spaan, Wendelin BoehmerNeurIPS 2025 · 1 citation
- Finite-Width Neural Tangent Kernels from Feynman DiagramsMax Guillen, Philipp Misof, Jan GerkenICML 2026 · 1 citation
- Equivariant Neural Tangent KernelsPhilipp Misof, Pan Kessel, Jan E. GerkenICML 2025
- Equivariant Covariance Tensors: Guaranteed SPD Uncertainty for Tensor-Valued Geometric LearningRuihan Liu, Yu Ji, Jianbo Yu, Shifu Yan et al.ICML 2026
Builds on11
- Neural Tangents: Fast and Easy Infinite Neural Networks in PythonRoman Novak, Lechao Xiao, Jiri Hron, Jaehoon Lee et al.ICLR 2020 · 254 citations
- Frame Averaging for Invariant and Equivariant Network DesignOmri Puny, Matan Atzmon, Edward J. Smith, Ishan Misra et al.ICLR 2022 · 177 citations
- Dynamics of Deep Neural Networks and Neural Tangent HierarchyJiaoyang Huang, Horng-Tzer YauICML 2020 · 167 citations
- A Program to Build E(N)-Equivariant Steerable CNNsGabriele Cesa, Leon Lang, Maurice WeilerICLR 2022 · 133 citations
- Equivariance with Learned Canonicalization FunctionsSékou-Oumar Kaba, Arnab Kumar Mondal, Yan Zhang, Yoshua Bengio et al.ICML 2023 · 109 citations
Related papers
- Disentangling the Predictive Variance of Deep Ensembles through the Neural Tangent KernelSeijin Kobayashi, Pau Vilimelis Aceituno, Johannes von OswaldNeurIPS 2022 · 4 citations
- Identifiable Equivariant Networks are Layerwise EquivariantVahid Shahverdi, Giovanni Luca Marchetti, Georg Bökman, Kathlén KohnICML 2026
- Bayesian Deep Ensembles via the Neural Tangent KernelBobby He, Balaji Lakshminarayanan, Yee Whye TehNeurIPS 2020 · 136 citations
- Neural (Tangent Kernel) CollapseMariia Seleznova, Dana Weitzner, Raja Giryes, Gitta Kutyniok et al.NeurIPS 2023 · 23 citations
- Symmetries in Overparametrized Neural Networks: A Mean Field ViewJavier Maass Martínez, Joaquín FontbonaNeurIPS 2024 · 4 citations
