Parallel Trust-Region Approaches in Neural Network Training: Beyond Traditional Methods
Trotti, Ken, Alegría, Samuel A. Cruz, Kopaničáková, Alena, Krause, Rolf
–arXiv.org Artificial Intelligence
We propose to train neural networks (NNs) using a novel variant of the ``Additively Preconditioned Trust-region Strategy'' (APTS). The proposed method is based on a parallelizable additive domain decomposition approach applied to the neural network's parameters. Built upon the TR framework, the APTS method ensures global convergence towards a minimizer. Moreover, it eliminates the need for computationally expensive hyper-parameter tuning, as the TR algorithm automatically determines the step size in each iteration. We demonstrate the capabilities, strengths, and limitations of the proposed APTS training method by performing a series of numerical experiments. The presented numerical study includes a comparison with widely used training methods such as SGD, Adam, LBFGS, and the standard TR method.
arXiv.org Artificial Intelligence
Dec-21-2023
- Country:
- Europe > Switzerland (0.04)
- North America
- United States > New York
- New York County > New York City (0.14)
- Canada > Ontario
- Toronto (0.14)
- United States > New York
- Genre:
- Research Report (0.64)
- Technology: