Understanding the Role of Functional Diversity in Weight-Ensembling with Ingredient Selection and Multidimensional Scaling
Not on the map. Why?
Abstract (arXiv)
Weight-ensembles are formed when the parameters of multiple neural networks are directly averaged into a single model. They have demonstrated generalization capability in-distribution (ID) and out-of-distribution (OOD) which is not completely understood, though they are thought to successfully exploit functional diversity allotted by each distinct model. Given a collection of models, it is also unclear which combination leads to the optimal weight-ensemble; the SOTA is a linear-time ``greedy" method. We introduce two novel weight-ensembling approaches to study the link between performance dynamics and the nature of how each method decides to use apply the functionally diverse components, akin to diversity-encouragement in the prediction-ensemble literature. We develop a visualization tool to explain how each algorithm explores various domains defined via pairwise-distances to further investigate selection and algorithms' convergence. Empirical analyses shed perspectives which reinforce how high-diversity enhances weight-ensembling while qualifying the extent to which diversity alone improves accuracy. We also demonstrate that sampling positionally distinct models can contribute just as meaningfully to improvements in a weight-ensemble.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 13 of its 18 references that are in the map. First posted on arXiv 4 Sep 2024.
Builds on
13 of 18 references.
References · 13
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
- 2021SWAD: Domain Generalization by Seeking Flat Minimacited by 431
- 2020What is being transferred in transfer learning?cited by 365
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2023A Unified Theory of Diversity in Ensemble Learningcited by 71
- 2021Diversity and Generalization in Neural Network Ensemblescited by 32
1 more
- 2023Pathologies of Predictive Diversity in Deep Ensemblescited by 16
Built on by
Not cited yet.