Git Re-Basin: Merging Models modulo Permutation Symmetries

AI-generated keywords: Deep Learning Non-Convex Optimization Stochastic Gradient Descent Loss Landscapes Mode Connectivity

AI-generated Key Points

  • Deep learning success attributed to solving complex non-convex optimization problems easily
  • Simple algorithms like stochastic gradient descent effective in fitting large neural networks
  • Neural network loss landscapes typically contain a single basin, considering permutation symmetries of hidden units
  • Introduction of three algorithms to permute units and align them with reference model
  • Transformation results in functionally equivalent weights in approximately convex basin near reference model
  • Experimental demonstration of single basin phenomenon across various model architectures and datasets
  • Discovery of zero-barrier linear mode connectivity between independently trained ResNet models on CIFAR-10 and CIFAR-100 datasets
  • Independently trained networks can have meaningful differences in learned features under certain scenarios
  • Model width and training time affect mode connectivity across different models and datasets
  • Shortcomings of single basin theory discussed, including counterexample to linear mode connectivity hypothesis
  • Study provides insights into optimization landscape of neural networks and understanding convergence towards similar solutions
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Authors: Samuel K. Ainsworth, Jonathan Hayase, Siddhartha Srinivasa

License: CC BY 4.0

Abstract: The success of deep learning is thanks to our ability to solve certain massive non-convex optimization problems with relative ease. Despite non-convex optimization being NP-hard, simple algorithms -- often variants of stochastic gradient descent -- exhibit surprising effectiveness in fitting large neural networks in practice. We argue that neural network loss landscapes contain (nearly) a single basin, after accounting for all possible permutation symmetries of hidden units. We introduce three algorithms to permute the units of one model to bring them into alignment with units of a reference model. This transformation produces a functionally equivalent set of weights that lie in an approximately convex basin near the reference model. Experimentally, we demonstrate the single basin phenomenon across a variety of model architectures and datasets, including the first (to our knowledge) demonstration of zero-barrier linear mode connectivity between independently trained ResNet models on CIFAR-10 and CIFAR-100. Additionally, we identify intriguing phenomena relating model width and training time to mode connectivity across a variety of models and datasets. Finally, we discuss shortcomings of a single basin theory, including a counterexample to the linear mode connectivity hypothesis.

Submitted to arXiv on 11 Sep. 2022

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2209.04836v1

The success of deep learning can be attributed to our ability to solve complex non-convex optimization problems with relative ease. Despite the NP-hard nature of non-convex optimization, simple algorithms like stochastic gradient descent have proven to be surprisingly effective in fitting large neural networks in practice. In this paper, the authors argue that neural network loss landscapes typically contain a single basin, taking into account all possible permutation symmetries of hidden units. To support their argument, the authors introduce three algorithms that permute the units of one model to align them with units of a reference model. This transformation results in a functionally equivalent set of weights that lie in an approximately convex basin near the reference model. The authors experimentally demonstrate this single basin phenomenon across various model architectures and datasets. One notable finding is the discovery of zero-barrier linear mode connectivity between independently trained ResNet models on CIFAR-10 and CIFAR-100 datasets. This suggests that independently trained networks can exhibit meaningful differences in the features they learn under certain scenarios. Additionally, the authors identify intriguing phenomena related to model width and training time in relation to mode connectivity across different models and datasets. They also discuss shortcomings of the single basin theory, including a counterexample to the linear mode connectivity hypothesis. Overall, this study provides valuable insights into the optimization landscape of neural networks and sheds light on how different models can converge towards similar solutions despite their initial differences. The findings have implications for improving training algorithms and understanding the behavior of deep learning models more effectively.
Created on 01 Jan. 2024

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Look for similar papers (in beta version)

By clicking on the button above, our algorithm will scan all papers in our database to find the closest based on the contents of the full papers and not just on metadata. Please note that it only works for papers that we have generated summaries for and you can rerun it from time to time to get a more accurate result while our database grows.

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.