An Optimal Scheduled Learning Rate for a Randomized Kaczmarz Algorithm

Nicholas F. Marshall,Oscar Mickelin
DOI: https://doi.org/10.1137/22M148803X
IF: 1.908
2023-03-17
SIAM Journal on Matrix Analysis and Applications
Abstract:SIAM Journal on Matrix Analysis and Applications, Volume 44, Issue 1, Page 312-330, March 2023. We study how the learning rate affects the performance of a relaxed randomized Kaczmarz algorithm for solving [math], where [math] is a consistent linear system and [math] has independent mean zero random entries. We derive a learning rate schedule which optimizes a bound on the expected error that is sharp in certain cases; in contrast to the exponential convergence of the standard randomized Kaczmarz algorithm, our optimized bound involves the reciprocal of the Lambert-[math] function of an exponential.
mathematics, applied
What problem does this paper attempt to address?