Nonlinear Least Squares for Large-Scale Machine Learning using Stochastic Jacobian Estimates

Johannes J. Brust
DOI: https://doi.org/10.48550/arXiv.2107.05598
2021-07-13
Abstract:For large nonlinear least squares loss functions in machine learning we exploit the property that the number of model parameters typically exceeds the data in one batch. This implies a low-rank structure in the Hessian of the loss, which enables effective means to compute search directions. Using this property, we develop two algorithms that estimate Jacobian matrices and perform well when compared to state-of-the-art methods.
Machine Learning,Numerical Analysis
What problem does this paper attempt to address?