首页    期刊浏览 2024年10月05日 星期六
登录注册

文章基本信息

  • 标题:An Adaptive Memory Multi-Batch L-BFGS Algorithm for Neural Network Training
  • 本地全文:下载
  • 作者:Federico Zocco ; Seán McLoone
  • 期刊名称:IFAC PapersOnLine
  • 印刷版ISSN:2405-8963
  • 出版年度:2020
  • 卷号:53
  • 期号:2
  • 页码:8199-8204
  • DOI:10.1016/j.ifacol.2020.12.1996
  • 语种:English
  • 出版社:Elsevier
  • 摘要:AbstractMotivated by the potential for parallel implementation of batch-based algorithms and the accelerated convergence achievable with approximated second order information a limited memory version of the BFGS algorithm has been receiving increasing attention in recent years for large neural network training problems. As the shape of the cost function is generally not quadratic and only becomes approximately quadratic in the vicinity of a minimum, the use of second order information by L-BFGS can be unreliable during the initial phase of training, i.e. when far from a minimum. Therefore, to control the influence of second order information as training progresses, we propose a multi-batch L-BFGS algorithm, namely MB-AM, that gradually increases its trust in the curvature information by implementing a progressive storage and use of curvature data through a development-based increase(dev-increase)scheme. Using six discriminative modelling benchmark problems we show empirically that MB-AM has slightly faster convergence and, on average, achieves better solutions than the standard multi-batch L-BFGS algorithm when training MLP and CNN models.
  • 关键词:KeywordsDeep learningL-BFGSvariable memoryquasi-Newton methodsneural networks
国家哲学社会科学文献中心版权所有