中文

一种用于深度神经网络训练的两阶段子空间信赖域方法

计算机视觉与模式识别 2018-05-25 v1

摘要

本文中,我们提出了一种用于训练前馈神经网络的新型二阶方法。在每次迭代中,我们在低维子空间中对代价函数构造二次近似。我们通过两阶段过程在信赖域内最小化该近似:首先在嵌入的正曲率子空间内,随后进行一次梯度下降步。该方法带来了目标函数的快速衰减,防止收敛到鞍点,并减轻了手动调参的需要。我们在基准数据集上展示了所提算法的良好性能。

关键词

引用

@article{arxiv.1805.09430,
  title  = {A Two-Stage Subspace Trust Region Approach for Deep Neural Network Training},
  author = {Viacheslav Dudar and Giovanni Chierchia and Emilie Chouzenoux and Jean-Christophe Pesquet and Vladimir Semenov},
  journal= {arXiv preprint arXiv:1805.09430},
  year   = {2018}
}

备注

EUSIPCO 2017