研究人员开发了一个新的数学标准,用于选择深度神经网络(DNN)中权重初始化的最优方差。通过分析随机梯度下降(SGD)在连续时间 Fokker-Planck 方程中的行为,他们通过 Kullback-Leibler 散度建立了初始分布与期望成本函数之间的联系。该方法提供了一种具体的、非启发式的方法来确定初始化尺度,与传统的 He-normal 初始化相比,在 MNIST 和 Fashion-MNIST 数据集上的验证结果显示,该方法能带来更低的训练损失和更高的测试准确率。 AI
影响 提供了一种基于数学的方法来提高 DNN 的训练效率和性能。
排序理由 学术论文,详细介绍了用于优化 DNN 初始化的一项新理论标准。[lever_c_demoted from research: ic=1 ai=1.0]
- deep neural network
- Fashion-MNIST
- He-normal initialization
- Hiroshi Horii
- Kullback--Leibler divergence
- MNIST database
- stochastic gradient descent
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →