Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

TaskDatasetResultRank
Node ClassificationCora
Accuracy89.71
609
Node ClassificationPubmed
Accuracy78.47
501
Node ClassificationAmazon Photo
Accuracy81.12
327
Node ClassificationarXiv
Accuracy70.56
325
Node ClassificationREDDIT
Accuracy92.33
268
Node ClassificationCiteseer
Mean Accuracy79.15
238
Link PredictionPubMed (test)
AUC91.78
120
Link PredictionCora (test)
AUC0.769
117
Node ClassificationwikiCS
Accuracy (WikiCS)83.35
101
Node ClassificationAmzComp
Accuracy85.42
39
Showing 10 of 13 rows

Other info

Follow for update