该研究探讨了深度神经网络权重矩阵的谱分布与模型质量之间的关系。此前已有工作发现,训练后权重矩阵中呈现重尾谱分布的特征能够有效预测模型质量,但缺乏严格的统计理论支撑。本文采用概率框架,仅利用网络最后一层权重推导出深度神经网络的边际似然,并在大规模卷积网络和大型语言模型上验证了该边际似然确实能够预测模型质量。为进一步理解重尾谱的作用,作者推导了服从高温Marchenko-Pastur分布的谱所对应的极限对数边际似然,证明该极限量关于重尾参数具有凸性,并据此得到了一个最优重尾程度,实验表明该值可有效预测大规模深度网络的过拟合现象。这项工作为基于权重谱的模型质量评估指标提供了统计学理论依据,并揭示了重尾程度与过拟合之间的定量联系。
| 重尾谱分布 | 指权重矩阵的特征值分布具有重尾特性,即极端大特征值出现的概率比高斯分布更高。 |
| 边际似然 | 在概率模型中,对隐变量积分后得到的观测数据似然,用于模型比较和选择。 |
| 高温Marchenko-Pastur分布 | 一种用于描述大维随机矩阵特征值分布的统计分布,通过温度参数控制尾部厚度。 |
| 过拟合 | 模型在训练数据上表现过好,但在新数据上泛化能力差的现象。 |
| DNN权重矩阵 | 深度神经网络中连接各层的参数矩阵,其谱特性与模型质量相关。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅