Loss Surface Simplexes for Mode Connecting Volumes and Fast Ensembling
Not on the map. Why?
Abstract (arXiv)
With a better understanding of the loss surfaces for multilayer networks, we can build more robust and accurate training procedures. Recently it was discovered that independently trained SGD solutions can be connected along one-dimensional paths of near-constant training loss. In this paper, we show that there are mode-connecting simplicial complexes that form multi-dimensional manifolds of low loss, connecting many independently trained models. Inspired by this discovery, we show how to efficiently build simplicial complexes for fast ensembling, outperforming independently trained deep ensembles in accuracy, calibration, and robustness to dataset shift. Notably, our approach only requires a few training epochs to discover a low-loss simplex, starting from a pre-trained solution. Code is available at https://github.com/g-benton/loss-surface-simplexes.
Key facts. Cited by 75 papers in the map (data through 7 Oct 2026). Builds on 23 of its 28 references that are in the map. First posted on arXiv 25 Feb 2021.
Builds on
23 of 28 references.
References · 23
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,274
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 555
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
11 more
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
- 2014The Loss Surfaces of Multilayer Networkscited by 278
- 2014Qualitatively characterizing neural network optimization problemscited by 186
- 2016Topology and Geometry of Half-Rectified Network Optimizationcited by 102
- 2019Subspace Inference for Bayesian Deep Learningcited by 94
- 2021Learning Neural Network Subspacescited by 82
- 2019Large Scale Structure of Neural Network Loss Landscapescited by 68
- 2019Understanding Generalization through Visualizationscited by 54
- 2020Rethinking Parameter Counting in Deep Models: Effective Dimensionality Revisitedcited by 50
- 2019Loss Patterns of Neural Networkscited by 10
- 2019A Deep Neural Network's Loss Surface Contains Every Low-dimensional Patterncited by 5
Built on by
Citing papers per year: 2021: 6; 2022: 11; 2023: 13; 2024: 22; 2025: 17; 2026: 6.
75 papers.
Sort:
By area:
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2023Deep Model Fusion: A Surveycited by 89
- 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Expertscited by 85
- 2021Prior and Posterior Networks: A Survey on Evidential Deep Learning Methods For Uncertainty Estimationcited by 71
- 2023Knowledge is a Region in Weight Space for Fine-tuned Language Modelscited by 58
- 2021Which Shortcut Cues Will DNNs Choose? A Study from the Parameter-Space Perspectivecited by 56
- 2022Linear Connectivity Reveals Generalization Strategiescited by 55
- 2025Deep Learning is Not So Mysterious or Differentcited by 55
- 2022ColD Fusion: Collaborative Descent for Distributed Multitask Finetuningcited by 52
38 more
- 2024What Matters for Model Merging at Scale?cited by 52
- 2022Mechanistic Mode Connectivitycited by 50
- 2024Towards Scalable and Versatile Weight Space Learningcited by 40
- 2022Model Zoos: A Dataset of Diverse Populations of Neural Network Modelscited by 38
- 2023Merging by Matching Models in Task Parameter Subspacescited by 36
- 2022Symmetries, flat minima, and the conserved quantities of gradient flowcited by 35
- 2021Dangers of Bayesian Model Averaging under Covariate Shiftcited by 31
- 2025Symmetry in Neural Network Parameter Spacescited by 31
- 2022Building a Subspace of Policies for Scalable Continual Learningcited by 28
- 2022lo-fi: distributed fine-tuning without communicationcited by 19
- 2022Continual Learning Beyond a Single Modelcited by 18
- 2025Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusioncited by 16
- 2021Learning a subspace of policies for online adaptation in Reinforcement Learningcited by 14
- 2023Fuse to Forget: Bias Reduction and Selective Memorization through Model Fusioncited by 13
- 2024QT-DoG: Quantization-aware Training for Domain Generalizationcited by 13
- 2021Connecting Low-Loss Subspace for Personalized Federated Learningcited by 12
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learningcited by 12
- 2025TinyR1-32B-Preview: Boosting Accuracy with Branch-Merge Distillationcited by 12
- 2024FusionBench: A Unified Library and Comprehensive Benchmark for Deep Model Fusioncited by 10
- 2025Understanding Mode Connectivity via Parameter Space Symmetrycited by 10
- 2024Exploring Neural Network Landscapes: Star-Shaped and Geodesic Connectivitycited by 9
- 2024WASH: Train your Ensemble with Communication-Efficient Weight Shuffling, then Averagecited by 9
- 2023Disentangling Linear Mode-Connectivitycited by 8
- 2024SMILE: Zero-Shot Sparse Mixture of Low-Rank Experts Construction From Pre-Trained Foundation Modelscited by 8
- 2024Improving Model Fusion by Training-time Neuron Alignment with Fixed Neuron Anchorscited by 7
- 2023To Stay or Not to Stay in the Pre-train Basin: Insights on Ensembling in Transfer Learningcited by 6
- 2024Do Deep Neural Network Solutions Form a Star Domain?cited by 6
- 2025Paths and Ambient Spaces in Neural Loss Landscapescited by 6
- 2023Sparsified Model Zoo Twins: Investigating Populations of Sparsified Neural Network Modelscited by 4
- 2023On permutation symmetries in Bayesian neural network posteriors: a variational perspectivecited by 4
- 2024How Good is a Single Basin?cited by 4
- 2024Subgraph Aggregation for Out-of-Distribution Generalization on Graphscited by 4
- 2025A Systematic Study of In-the-Wild Model Merging for Large Language Modelscited by 4
- 2026Where Does Warm-Up Come From? Adaptive Scheduling for Norm-Constrained Optimizerscited by 4
- 2023From Charts to Atlas: Merging Latent Spaces into Onecited by 3
- 2024Federated Learning over Connected Modescited by 3
- 2025Estimating the Probability of Sampling a Trained Neural Network at Randomcited by 3
- 2025Connecting Independently Trained Modes via Layer-Wise Connectivitycited by 3
50 most cited of 75 All 75 (JSON)
- 2026Misalignment of Low-Loss Regions Causes Grokking30 Sep 2026
- 2026Understanding Machine Unlearning Through the Lens of Mode Connectivity27 Jul 2026
- 2026Fast Generalization after Interpolation via Critically Damped Momentum Optimization1 Jun 2026
- 2026Trade-offs in Ensembling, Merging and Routing Among Parameter-Efficient Experts3 Mar 2026
- 2026The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging24 Feb 2026
- 2026Where Does Warm-Up Come From? Adaptive Scheduling for Norm-Constrained Optimizers5 Feb 2026
- 2025Walking on the Fiber: A Simple Geometric Approximation for Bayesian Neural Networks1 Dec 2025
- 2025A Systematic Study of In-the-Wild Model Merging for Large Language Models26 Nov 2025
- 2025Linear Mode Connectivity under Data Shifts for Deep Ensembles of Image Classifiers6 Nov 2025
- 2025Closing the Oracle Gap: Increment Vector Transformation for Class Incremental Learning26 Sep 2025
- 2025SAM Encoder Breach by Adversarial Simplicial Complex Triggers Downstream Model Failures8 Aug 2025
- 2025Learning from Oblivion: Predicting Knowledge Overflowed Weights via Retrodiction of Forgetting7 Aug 2025
38 more
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
- 2025Understanding Mode Connectivity via Parameter Space Symmetry29 May 2025
- 2025Connecting Independently Trained Modes via Layer-Wise Connectivity5 May 2025
- 2025Understanding Machine Unlearning Through the Lens of Mode Connectivity8 Apr 2025
- 2025Aggregation on Learnable Manifolds for Asynchronous Federated Optimization18 Mar 2025
- 2025You Only Debias Once: Towards Flexible Accuracy-Fairness Trade-offs at Inference Time10 Mar 2025
- 2025TinyR1-32B-Preview: Boosting Accuracy with Branch-Merge Distillation6 Mar 2025
- 2025Paths and Ambient Spaces in Neural Loss Landscapes5 Mar 2025
- 2025Deep Learning is Not So Mysterious or Different3 Mar 2025
- 2025Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusion1 Feb 2025
- 2025Estimating the Probability of Sampling a Trained Neural Network at Random31 Jan 2025
- 2024Set-Valued Sensitivity Analysis of Deep Neural Networks15 Dec 2024
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learning27 Nov 2024
- 2024Ex Uno Pluria: Insights on Ensembling in Low Precision Number Systems22 Nov 2024
- 2024Subgraph Aggregation for Out-of-Distribution Generalization on Graphs29 Oct 2024
- 2024Deep Model Merging: The Sister of Neural Network Interpretability -- A Survey16 Oct 2024
- 2024QT-DoG: Quantization-aware Training for Domain Generalization8 Oct 2024
- 2024Hyper-Representations: Learning from Populations of Neural Networks7 Oct 2024
- 2024What Matters for Model Merging at Scale?4 Oct 2024
- 2024SMILE: Zero-Shot Sparse Mixture of Low-Rank Experts Construction From Pre-Trained Foundation Models19 Aug 2024
- 2024Mitigating the Stability-Plasticity Dilemma in Adaptive Train Scheduling with Curriculum-Driven Continual DQN Expansion19 Aug 2024
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunities14 Aug 2024
- 2024Towards Scalable and Versatile Weight Space Learning14 Jun 2024
- 2024FusionBench: A Unified Library and Comprehensive Benchmark for Deep Model Fusion5 Jun 2024
- 2024WASH: Train your Ensemble with Communication-Efficient Weight Shuffling, then Average27 May 2024
- 2024Manifold Metric: A Loss Landscape Approach for Predicting Model Performance24 May 2024
- 2024Visualizing, Rethinking, and Mining the Loss Landscape of Deep Neural Networks21 May 2024
- 2024Exploring Neural Network Landscapes: Star-Shaped and Geodesic Connectivity9 Apr 2024
- 2024Do Deep Neural Network Solutions Form a Star Domain?12 Mar 2024
- 2024Federated Learning over Connected Modes5 Mar 2024
- 2024How Good is a Single Basin?5 Feb 2024
- 2024Improving Model Fusion by Training-time Neuron Alignment with Fixed Neuron Anchors2 Feb 2024
- 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Experts1 Feb 2024
- 2023Disentangling Linear Mode-Connectivity15 Dec 2023
- 2023Merging by Matching Models in Task Parameter Subspaces7 Dec 2023
- 2023Fuse to Forget: Bias Reduction and Selective Memorization through Model Fusion13 Nov 2023
- 2023From Charts to Atlas: Merging Latent Spaces into One11 Nov 2023
- 2023On permutation symmetries in Bayesian neural network posteriors: a variational perspective16 Oct 2023
50 newest of 75 All 75 (JSON)
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2023Deep Model Fusion: A Surveycited by 89
- 2021Prior and Posterior Networks: A Survey on Evidential Deep Learning Methods For Uncertainty Estimationcited by 71
- 2023Knowledge is a Region in Weight Space for Fine-tuned Language Modelscited by 58
- 2021Which Shortcut Cues Will DNNs Choose? A Study from the Parameter-Space Perspectivecited by 56
- 2022Linear Connectivity Reveals Generalization Strategiescited by 55
- 2025Deep Learning is Not So Mysterious or Differentcited by 55
- 2022ColD Fusion: Collaborative Descent for Distributed Multitask Finetuningcited by 52
- 2024What Matters for Model Merging at Scale?cited by 52
12 most cited of 71 All 71 (JSON)
- 2026Misalignment of Low-Loss Regions Causes Grokking30 Sep 2026
- 2026Fast Generalization after Interpolation via Critically Damped Momentum Optimization1 Jun 2026
- 2026Trade-offs in Ensembling, Merging and Routing Among Parameter-Efficient Experts3 Mar 2026
- 2026The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging24 Feb 2026
- 2026Where Does Warm-Up Come From? Adaptive Scheduling for Norm-Constrained Optimizers5 Feb 2026
- 2025Walking on the Fiber: A Simple Geometric Approximation for Bayesian Neural Networks1 Dec 2025
- 2025A Systematic Study of In-the-Wild Model Merging for Large Language Models26 Nov 2025
- 2025Linear Mode Connectivity under Data Shifts for Deep Ensembles of Image Classifiers6 Nov 2025
- 2025Closing the Oracle Gap: Increment Vector Transformation for Class Incremental Learning26 Sep 2025
- 2025SAM Encoder Breach by Adversarial Simplicial Complex Triggers Downstream Model Failures8 Aug 2025
- 2025Learning from Oblivion: Predicting Knowledge Overflowed Weights via Retrodiction of Forgetting7 Aug 2025
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
12 newest of 71 All 71 (JSON)