arXiv:2501.18812cs.LG2025-01被引 3

提出快速估算神经网络参数空间局部体积的方法,揭示复杂度与泛化能力的关系。

Estimating the Probability of Sampling a Trained Neural Network at Random

  • 利用梯度信息改进已有估计方法,实现快速局部体积估算
  • 训练中网络复杂度随体积负对数上升,过拟合模型局部体积更小
  • 支持'体积假说':简单函数在参数空间占更大体积,利于泛化

我们提出并分析了一种算法,用于估算在高斯或均匀分布下,神经网络参数空间中与特定'锚点'行为相似的局部邻域大小,称为'局部体积'。该方法改进了现有快速但仅提供下界的盆地体积估计器,通过利用主流优化器已提供的梯度信息,采用重要性采样提升估计精度。局部体积的负对数可解释为锚点网络的信息含量。如预期,该量在语言模型训练过程中增加。我们发现过拟合、泛化差的区域体积更小,表明其学习行为更复杂。从最小描述长度(MDL)角度看,这种小体积意味着次优压缩。结果支持'体积假说':神经网络训练产生良好泛化,主要因为架构使简单函数在参数空间中占据更大体积,而优化器以体积敏感方式采样低损失流形。我们认为,快速局部体积估计算法是衡量网络复杂度和结构归纳偏置的有前景实用指标,适用于可解释性研究。

原文摘要 · Abstract (English)

We present and analyze an algorithm for estimating the size, under a Gaussian or uniform measure, of a localized neighborhood in neural network parameter space with behavior similar to an ``anchor'' point. We refer to this as the "local volume" of the anchor. We adapt an existing basin-volume estimator, which is very fast but in many cases only provides a lower bound. We show that this lower bound can be improved with an importance-sampling method using gradient information that is already provided by popular optimizers. The negative logarithm of local volume can also be interpreted as a measure of the anchor network's information content. As expected for a measure of complexity, this quantity increases during language model training. We find that overfit, badly-generalizing neighborhoods are smaller, indicating a more complex learned behavior. This smaller volume can also be interpreted in an MDL sense as suboptimal compression. Our results are consistent with a picture of generalization we call the "volume hypothesis": that neural net training produces good generalization primarily because the architecture gives simple functions more volume in parameter space, and the optimizer samples from the low-loss manifold in a volume-sensitive way. We believe that fast local-volume estimators are a promising practical metric of network complexity and architectural inductive bias for interpretability purposes.

神经网络泛化理论体积估计可解释性

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