中文

缓解宽神经网络中病态尖锐性的归一化方法

机器学习 2019-10-29 v2 无序系统与神经网络 机器学习

摘要

归一化方法在提升深度学习性能方面发挥着重要作用,而其理论理解却很有限。为从理论上阐明归一化的有效性,我们量化了由Fisher信息矩阵(FIM)确定的参数空间几何,这在某些条件下也对应于损失景观的局部形状。我们分析了随机初始化的深度神经网络,已知当网络足够宽时会遭受景观的病态尖锐形状。我们揭示,若宽度与样本数满足特定条件,最后一层的批归一化可显著减小此类病态尖锐性。相反,在许多设定下,中间隐藏层的批归一化难以缓解病态尖锐性。我们还发现层归一化也无法缓解病态尖锐性。因此,我们可以得出结论:最后一层的批归一化对降低由FIM引起的尖锐性有显著贡献。

关键词

引用

@article{arxiv.1906.02926,
  title  = {The Normalization Method for Alleviating Pathological Sharpness in Wide Neural Networks},
  author = {Ryo Karakida and Shotaro Akaho and Shun-ichi Amari},
  journal= {arXiv preprint arXiv:1906.02926},
  year   = {2019}
}

备注

To appear in NeurIPS 2019