arXiv:2505.23681cs.LG2025-05ICML被引 18

用参数空间对称性解释神经网络最优解的连通性

Understanding Mode Connectivity via Parameter Space Symmetry

  • 通过参数空间对称性分析最优解的拓扑结构
  • 证明线性网络最优解连通分量数量可被对称性决定
  • 揭示跳跃连接降低连通性复杂度,适合研究模型结构者

神经网络的极小值点常可通过损失几乎不变的曲线相互连接,这一现象称为模式连通性。尽管该性质已用于模型合并与微调等应用,其理论机制仍不明确。本文提出基于参数空间对称性的新方法,将对称群的拓扑结构与极小值的连通性关联起来,推导出线性网络极小值的连通分量数量,并发现跳跃连接能减少该数量。进一步利用参数对称性分析模式连通性及线性模式连通性成立或失效的条件,揭示了对称性在极小值中占据显著部分。最后,给出由对称性诱导的极小值间连接曲线的显式表达式,并通过曲线曲率推导出线性模式连通性近似成立的条件。研究强调连续对称性在理解神经网络损失景观中的关键作用。

原文摘要 · Abstract (English)

Neural network minima are often connected by curves along which train and test loss remain nearly constant, a phenomenon known as mode connectivity. While this property has enabled applications such as model merging and fine-tuning, its theoretical explanation remains unclear. We propose a new approach to exploring the connectedness of minima using parameter space symmetry. By linking the topology of symmetry groups to that of the minima, we derive the number of connected components of the minima of linear networks and show that skip connections reduce this number. We then examine when mode connectivity and linear mode connectivity hold or fail, using parameter symmetries which account for a significant part of the minimum. Finally, we provide explicit expressions for connecting curves in the minima induced by symmetry. Using the curvature of these curves, we derive conditions under which linear mode connectivity approximately holds. Our findings highlight the role of continuous symmetries in understanding the neural network loss landscape.

神经网络损失景观对称性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。