🔥 今日值得一读 4百万参数网络泛化界不再爆表!新方法彻底摆脱维度依赖
Dimension-Free PAC-Bayes Generalization for Deep Neural Networks via Dirichlet Processes
arXiv ML (stat.ML) 🔥 重点 #泛化理论#PAC-Bayes#深度网络 🕐 今天 12:00
👨‍💼 主理人解读 · 为什么值得关注
为过参数化网络提供不随参数量爆炸的泛化界,帮助理解深度模型为何能泛化。

📖 AI 总结

本文针对经典PAC-Bayes泛化理论在深度神经网络中的失效问题提出新解法。传统方法将先验置于欧几里得参数空间,导致相对熵随网络维度D增长,在过参数化场景(D远大于样本量n)下产生超过1000%的无意义泛化界。作者转而通过Dirichlet过程将先验定义在数据分布的非参数空间上,借助数据处理不等式证明参数空间相对熵被数据空间散度严格约束,并利用Digamma函数给出精确闭式解。所得泛化证书与网络参数量、宽度和深度完全无关,在本征维度子流形上达到极小极大非参数收敛速率。在MNIST上对参数量从7850到420万的多种视觉架构实验中,经典权重空间界超过940%而失效,该数据空间证书稳定保持在8.8%至9.4%的非平凡水平,为深度网络泛化理论提供了维度无关的新框架。

🔑 关键词速览

PAC-贝叶斯泛化理论一种基于概率近似正确(PAC)框架的泛化理论,通过先验分布和后验分布之间的相对熵来界定学习算法的泛化误差。
Dirichlet 过程 (DP)一种用于非参数贝叶斯建模的随机过程,其样本路径是概率分布,常用于定义数据分布空间上的先验。
相对熵也称为 Kullback-Leibler 散度,衡量两个概率分布之间的差异,在 PAC-贝叶斯界中用于控制泛化误差。
数据处理不等式信息论中的一个不等式,表明对数据进行确定性处理后,互信息或相对熵不会增加,用于约束参数空间与数据空间散度的关系。
极小极大非参数速率在非参数统计中,衡量估计量收敛速度的最优速率,通常表示为样本量 $n$ 和内在维度 $d^*$ 的函数。
infoAI 公众号二维码

📱 每天一份 AI 前沿日报

关注公众号,每天 09:00 推送 · 不错过任何重磅