The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networks
About
In this paper, we conjecture that if the permutation invariance of neural networks is taken into account, SGD solutions will likely have no barrier in the linear interpolation between them. Although it is a bold conjecture, we show how extensive empirical attempts fall short of refuting it. We further provide a preliminary theoretical result to support our conjecture. Our conjecture has implications for lottery ticket hypothesis, distributed training, and ensemble methods.
Rahim Entezari, Hanie Sedghi, Olga Saukh, Behnam Neyshabur• 2021
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Cora | Accuracy89.71 | 609 | |
| Node Classification | Pubmed | Accuracy78.47 | 501 | |
| Node Classification | Amazon Photo | Accuracy81.12 | 327 | |
| Node Classification | arXiv | Accuracy70.56 | 325 | |
| Node Classification | Accuracy92.33 | 268 | ||
| Node Classification | Citeseer | Mean Accuracy79.15 | 238 | |
| Link Prediction | PubMed (test) | AUC91.78 | 120 | |
| Link Prediction | Cora (test) | AUC0.769 | 117 | |
| Node Classification | wikiCS | Accuracy (WikiCS)83.35 | 101 | |
| Node Classification | AmzComp | Accuracy85.42 | 39 |
Showing 10 of 13 rows