2022

Linear Connectivity Reveals Generalization Strategies

Juneja, Jeevesh, Bansal, Rachit, Cho, Kyunghyun et al.

Understand

It is widely accepted in the mode connectivity literature that when two neural networks are trained similarly on the same data, they are connected by a path through parameter space over which test set accuracy is maintained.

  • Under some circumstances, including transfer learning from pretrained models, these paths are presumed to be linear.
  • In contrast to existing results, we find that among text classifiers (trained on MNLI, QQP, and CoLA), some pairs of finetuned models have large barriers of increasing loss on the linear paths between them.
  • On each task, we find distinct clusters of models which are linearly connected on the test loss surface, but are disconnected from models outside the cluster -- models that occupy separate basins on the surface.

Reading the bibliography…