Randomized flexible Krylov methods for $\ell_p$ regularization
Malena Sabaté Landman, Yuji Nakatsukasa
TL;DR
This work tackles large-scale ill-posed inverse problems with sparsity via \\ell_p regularization, primarily \\ell_1, by developing randomized flexible Krylov methods to accelerate iteratively reweighted norm schemes. It introduces sketch-based preconditioning and sketch-to-precondition strategies, and two families of randomized flexible solvers (with convergence guarantees) that reduce per-iteration cost while maintaining accuracy. The methods are analyzed for monotonic descent and implemented in sketch-and-solve and sketch-to-precondition variants, applicable to tall-skinny and near-square systems. Numerical experiments on subset selection, deblurring, and tomography show speedups and competitive accuracy, validating the practical impact of combining randomized linear algebra with flexible Krylov subspaces for sparse inverse problems.
Abstract
The computation of sparse solutions of large-scale linear discrete ill-posed problems remains a computationally demanding task. A powerful framework in this context is the use of iteratively reweighted schemes, which are based on constructing a sequence of quadratic tangent majorants of the $\ell_2$-$\ell_1$ regularization functional (with additional smoothing to ensure differentiability at the origin), and solving them successively. Recently, flexible Krylov-Tikhonov methods have been used to partially solve each problem in the sequence efficiently. However, in order to guarantee convergence, the complexity of the algorithm at each iteration increases with respect to more traditional methods. We propose a randomized flexible Krylov method to alleviate the increase of complexity, which leverages the adaptability of the flexible Krylov subspaces with the efficiency of `sketch-and-solve' methods. A possible caveat of the mentioned methods is their memory requirements. In this case, one needs to rely instead on inner-outer schemes. In these scenarios, we propose a `sketch-to-precondition' method to speed up the convergence of each of the subproblems in the sequence. The performance of these algorithms is shown through a variety of numerical examples.
