arXiv reader

Mixed precision sketching for least-squares problems and its application in GMRES-based iterative refinement

Erin Carson and Ieva Daužickaitė

arXiv
2410.06319v2 · PDF · 11 June 2025
Categories
math.NA, cs.NA
Licence
CC BY 4.0 The site shows the text.

Abstract

Sketching-based preconditioners have been shown to accelerate the solution of dense least-squares problems with coefficient matrices having substantially more rows than columns. The cost of generating these preconditioners can be reduced by employing low precision floating-point formats for all or part of the computations. We perform finite precision analysis of a mixed precision algorithm that computes the R-factor of a QR factorization of the sketched coefficient matrix. Two precisions can be chosen and the analysis allows understanding how to set these precisions to exploit the potential benefits of low precision formats and still guarantee an effective preconditioner. If the nature of the least-squares problem requires a solution with a small forward error, then mixed precision iterative refinement (IR) may be needed. For ill-conditioned problems the GMRES-based IR approach can be used, but good preconditioner is crucial to ensure convergence. We theoretically show when the sketching-based preconditioner can guarantee that the GMRES-based IR reduces the relative forward error of the least-squares solution and the residual to the level of the working precision unit roundoff. Small numerical examples illustrate the analysis.

Items 5

Definition

Definition 1

VD34

Consider a subspace \mathbb{F} \subseteq \mathbb{R}^m, a linear operator \Omega : \mathbb{R}^{m} \to \mathbb{R}^{s}, and a distortion parameter \epsilon \in (0,1). \Omega is called an l_2 \epsilon-subspace embedding if for every x \in \mathbb{F} we have

\sqrt{1 - \epsilon} \, \Vert x \Vert_2 \leq \Vert \Omega x \Vert_2 \leq \sqrt{1 + \epsilon} \, \Vert x \Vert_2.
Definition

Definition 2

ZRZS

A linear operator \Omega : \mathbb{R}^{m} \to \mathbb{R}^{s} is called an oblivious (\epsilon,\delta,n)-subspace embedding if it is a subspace embedding for any n-dimensional \mathbb{F} \subseteq \mathbb{R}^m with probability at least 1-\delta.

Theorem

Theorem 3

EFNG

r@th:finite_recision_error_R Label ‘th:finite_recision_error_R’ multiply defined

SIAM Label Error: There were multiply-defined labels Correct multiply-defined labels. r@th:finite_recision_error_Rsection.equation Consider \widehat{Y} = \Omega A + \Delta_s computed in step ? of Algorithm ?, where \Delta_s accounts for the errors in casting A to u_s and computing the sketch. If \Omega, u_s, and u_{QR} are set so that

\begin{gathered} (2 \log_2n + 4) n^{1/2} \widetilde{\gamma}_{sn}^{(QR)} \kappa_2(\widehat{Y}) < 1 \quad \text{ and } \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:assumption_u_qr} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:assumption_u_qr' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:assumption_u_qr}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \\ (2 \log_2n + 4) n^{1/2} \Vert (\Omega A)^{\dagger} \Vert_2 \Vert \Delta_s \Vert_2 <1 \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:assumption_u_s} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:assumption_u_s' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:assumption_u_s}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \end{gathered}

and we denote

\begin{aligned} \beta = & \, \left( 1 + 18 (2 \log_2n + 4) n^{1/2} \widetilde{\gamma}_{sn}^{(QR)} \kappa_2(\Omega A)\right)\\ & \, \times \left(1 + 2(2 \log_2n + 4) n^{1/2} \Vert (\Omega A)^{\dagger} \Vert_2 \Vert \Delta_s \Vert_2 \right), \end{aligned}

then \widehat{R} satisfies the following:

\begin{aligned} \Vert \widehat{R} \Vert_2 & \leq \beta \Vert \Omega A \Vert_2, \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:Rhat_norm_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:Rhat_norm_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:Rhat_norm_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack\\ \Vert \widehat{R}^{-1} \Vert_2 & \leq \beta \Vert (\Omega A)^{\dagger} \Vert_2, \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:Rhat_inv_norm_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:Rhat_inv_norm_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:Rhat_inv_norm_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack\\ \kappa_2(\widehat{R}) & \leq \beta^2 \kappa_2(\Omega A), \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:Rhat_cond_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:Rhat_cond_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:Rhat_cond_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \\ \Vert A \widehat{R}^{-1} \Vert_2 & \leq \beta \Vert AR^{-1} \Vert_2, \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:ARhat_inv_norm_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:ARhat_inv_norm_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:ARhat_inv_norm_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack\\ \Vert (A \widehat{R}^{-1})^{\dagger} \Vert_2 & \leq \beta \Vert (AR^{-1})^{\dagger} \Vert_2, \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:ARhat_inv_pinv_norm_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:ARhat_inv_pinv_norm_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:ARhat_inv_pinv_norm_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack\\ \kappa_2(A \widehat{R}^{-1}) & \leq \beta^2 \kappa_2( A R^{-1}). \@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:ARhat_inv_cond_bound} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:ARhat_inv_cond_bound' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:ARhat_inv_cond_bound}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \end{aligned}
Corollary

Corollary 4

E1VM

Let A \in \mathbb{R}^{m \times n} be full rank and A_s denote A cast to a precision u_s, such that

\begin{gathered} A_s = A + E, \quad \textrm{where } \Vert E \Vert_2 \leq \sqrt{n} u_s \Vert A \Vert_2, \quad \textrm{and} \\ \Omega A_s = Q_s R_s \end{gathered}

is the economic QR decomposition. We assume that A_s is full-rank and assume that the assumptions of Theorem 3 hold with A replaced by A_s, and denote the resulting \beta as \beta_s. Then (3) - (3) can be written as

\begin{aligned} \Vert \widehat{R} \Vert_2 & \leq \beta_s \Vert \Omega A_s \Vert_2, \\ \Vert \widehat{R}^{-1} \Vert_2 & \leq \beta_s \Vert (\Omega A_s)^{\dagger} \Vert_2, \\ \kappa_2(\widehat{R}) & \leq \beta_s^2 \kappa_2(\Omega A_s),\\ \Vert A \widehat{R}^{-1} \Vert_2 & \leq \beta_s \Vert AR_s^{-1} \Vert_2, \\ \Vert (A \widehat{R}^{-1})^{\dagger} \Vert_2 & \leq \beta_s \Vert (AR_s^{-1})^{\dagger} \Vert_2,\\ \kappa_2(A \widehat{R}^{-1}) & \leq \beta_s^2 \kappa_2( A R_s^{-1}). \end{aligned}
Theorem

Theorem 5

0VUI

Assume that the least-squares problem (?) is solved via Algorithm ?, the assumptions of Theorem 3 are satisfied, and the precisions in FGMRES are set so that (?) holds and \psi(\beta,\Omega A) is as in (?). Then the relative forward error of the augmented system reaches the limiting value of 4pu_r\textrm{cond}(\widetilde{A},d)+ u if

\@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:fe_converg_bound1} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:fe_converg_bound1' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:fe_converg_bound1}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \beta \max \left\{1, \left(\sqrt{1 + 4 \beta^{-1} \Vert ( AR^{-1})^{\dagger}\Vert_2^{-2}} - 1 \right)^{-1} \right\} \Vert AR^{-1}\Vert_2 \psi(\beta,\Omega A) < u^{-1}.

If the optimal scaling \alpha = 2^{-1/2} \sigma_{min}(A \widehat{R}^{-1}) is applied in step ? of Algorithm ?, then the forward error convergence condition can be replaced by

\@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:fe_converg_bound2} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:fe_converg_bound2' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:fe_converg_bound2}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \beta^2 \kappa_2(A R^{-1}) \psi(\beta,\Omega A) < u^{-1}.

The normwise relative backward error of the augmented system reaches the limiting value pu if

\@bsphack \protected@write\@auxout{} {\string \@ifundefined{r@eq:be_IR_cond_interim} \relax {\gdef \relax { \@latex@warning@no@line{Label `eq:be_IR_cond_interim' multiply defined}} \GenericError{ \space\space\space\@spaces\@spaces\@spaces }{ SIAM Label Error: There were multiply-defined labels }{ Correct multiply-defined labels. }{} } \global\@namedef{r@eq:be_IR_cond_interim}{{\p@equation\Alph{section}.\arabic{equation}}{\thepage}}} \@esphack \psi(\beta,\Omega A) \kappa(\widetilde{A}) < u^{-1}.

Source and licence

The text on this page is from arXiv:2410.06319v2 by Erin Carson and Ieva Daužickaitė, under the licence CC BY 4.0. The corpus made these changes: it read the items out of the LaTeX source, it expanded the macros of the paper, it put each proof after its statement, and it converted the text to HTML. The formulas are the TeX of the paper, and KaTeX renders them. A formula or a reference can look different from the PDF. The page is published under CC BY 4.0.

The record of each item is in analysis/objects/2410/2410.06319.jsonl.