中文

神经网络训练中的并行信赖域方法:超越传统方法

数值分析 2023-12-22 v1 机器学习 数值分析

摘要

我们提出使用“加法预条件信赖域策略”(APTS)的一种新颖变体来训练神经网络(NNs)。所提出的方法基于应用于神经网络参数的可并行加法区域分解方法。建立在信赖域(TR)框架之上,APTS 方法确保了向最小化器的全局收敛。此外,它消除了对计算昂贵的超参数调整的需求,因为 TR 算法会自动确定每次迭代的步长。我们通过一系列数值实验展示了所提出的 APTS 训练方法的能力、优势和局限性。所展示的数值研究包括与广泛使用的训练方法(如 SGD、Adam、LBFGS 和标准 TR 方法)的比较。

关键词

引用

@article{arxiv.2312.13677,
  title  = {Parallel Trust-Region Approaches in Neural Network Training: Beyond Traditional Methods},
  author = {Ken Trotti and Samuel A. Cruz Alegría and Alena Kopaničáková and Rolf Krause},
  journal= {arXiv preprint arXiv:2312.13677},
  year   = {2023}
}