mode connectivity
In plain wordsThe discovery that two separately trained versions of a network, though their settings look very different, can usually be joined by a path along which the model keeps working well.
The empirical finding that solutions found by training the same architecture several times (for example, from different random seeds) are usually not isolated basins walled off by high loss: low-loss paths link them, often as simple as a quadratic Bézier curve or a two-segment polygonal line, and training and test performance barely change along the way. Linear mode connectivity is the stronger property that the straight line between two solutions has no loss barrier; it generally fails between independently trained networks but holds between runs that share an early segment of training, and can be recovered for independent runs by permuting one network's hidden units to align with the other's, exploiting permutation symmetry (as in Git Re-Basin). Mode connectivity helps explain why weight averaging works (see exponential moving average of weightsexponential moving average of weights, n.: A smoothed copy of a model kept during training, which each update nudges only slightly toward the latest weights. The calmer copy often performs better than the jittery original.), underlies model mergingmodel merging, n.: Combining two or more trained AI models into one by blending their internal numbers directly, so the result shares their skills without any further training. and task arithmetictask arithmetic, n.: Editing an AI model by adding or subtracting the changes that fine-tuning made to it, as if skills were arrows that can be added together or reversed. for fine-tuned models that share a pretrained starting point, and informs the lottery ticket hypothesislottery ticket hypothesis, n.: The idea that a large, randomly started neural network contains a much smaller network inside it that, trained alone from the same starting values, would learn just as well — a winning lottery ticket., whose tickets emerge once training becomes stable in this linear sense.
Origin
Discovered independently in 2018 by Timur Garipov, Pavel Izmailov and colleagues (NeurIPS 2018) and by Felix Draxler and colleagues (ICML 2018). Jonathan Frankle and colleagues defined linear mode connectivity (2020), and Samuel Ainsworth, Jonathan Hayase and Siddhartha Srinivasa's Git Re-Basin (2022) used permutation alignment to connect independently trained networks.
Related
- loss landscapeloss landscape, n.: A way of picturing training as a journey across hilly terrain, where position stands for the model's settings and height for its error. Training searches for low valleys.
- model mergingmodel merging, n.: Combining two or more trained AI models into one by blending their internal numbers directly, so the result shares their skills without any further training.
- task arithmetictask arithmetic, n.: Editing an AI model by adding or subtracting the changes that fine-tuning made to it, as if skills were arrows that can be added together or reversed.
- lottery ticket hypothesislottery ticket hypothesis, n.: The idea that a large, randomly started neural network contains a much smaller network inside it that, trained alone from the same starting values, would learn just as well — a winning lottery ticket.
- exponential moving average of weightsexponential moving average of weights, n.: A smoothed copy of a model kept during training, which each update nudges only slightly toward the latest weights. The calmer copy often performs better than the jittery original.
- ensemble learningensemble learning, n.: Combining the predictions of several models — by voting or averaging — so their individual mistakes partly cancel out, giving more accurate and stable results than any one model.
Sources
- Garipov, T., Izmailov, P., Podoprikhin, D., Vetrov, D. & Wilson, A. G. (2018). Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNs. NeurIPS 2018.
- Draxler, F., Veschgini, K., Salmhofer, M. & Hamprecht, F. A. (2018). Essentially No Barriers in Neural Network Energy Landscape. ICML 2018.
- Frankle, J., Dziugaite, G. K., Roy, D. M. & Carbin, M. (2020). Linear Mode Connectivity and the Lottery Ticket Hypothesis. ICML 2020.
Mentioned in
loss landscapeloss landscape, n.: A way of picturing training as a journey across hilly terrain, where position stands for the model's settings and height for its error. Training searches for low valleys., lottery ticket hypothesislottery ticket hypothesis, n.: The idea that a large, randomly started neural network contains a much smaller network inside it that, trained alone from the same starting values, would learn just as well — a winning lottery ticket.