穩定雙共軛梯度法

在數值線性代數中，穩定雙共軛梯度法（Biconjugate gradient stabilized method，通常簡稱為BiCGSTAB）是一種由荷蘭數學家 H. A. van der Vorst 提出的用於數值求解非對稱線性方程組的迭代方法。它是雙共軛梯度法（BiCG）的一個變種，比雙共軛梯度法本身以及諸如共軛梯度平方法（CGS）等其他變種有更快速和更平滑的收斂性。它是一種 Krylov 子空間方法。

## 演算法步驟

### 無預處理穩定雙共軛梯度法

要求解線性方程組 $\boldsymbol{Ax}=\boldsymbol{b}$，穩定雙共軛梯度法從初始解 $\boldsymbol{x}_0$ 開始按以下步驟迭代：

- $$\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{Ax}$$
- 任意選擇向量 $\boldsymbol{\hat{r}}_0\;$ 使得 $(\boldsymbol{\hat{r}}_0,\boldsymbol{r}_0) \neq0\;$，例如，$\boldsymbol{\hat{r}}_0=\boldsymbol{r}_0\;$
- $$\rho_0=\alpha=\omega_0=1\;$$
- $$\boldsymbol{v}_0=\boldsymbol{p}_0=\boldsymbol{0}$$
- 對 $i=1,2,3,\ldots$
- $$\rho_i=(\boldsymbol{\hat{r}}_0,\boldsymbol{r}_{i-1})\;$$
- $$\beta=(\rho_i/\rho_{i-1})(\alpha/\omega_{i-1})\;$$
- $$\boldsymbol{p}_i=\boldsymbol{r}_{i-1}+\beta(\boldsymbol{p}_{i-1}-\omega_{i-1}\boldsymbol{v}_{i-1})$$
- $$\boldsymbol{v}_i=\boldsymbol{Ap}_i$$
- $$\alpha=\rho_i/(\boldsymbol{\hat{r}}_0,\boldsymbol{v}_i)\;$$
- $$\boldsymbol{s}=\boldsymbol{r}_{i-1}-\alpha\boldsymbol{v}_i$$
- $$\boldsymbol{t}=\boldsymbol{As}$$
- $$\omega_i=(\boldsymbol{t},\boldsymbol{s})/(\boldsymbol{t},\boldsymbol{t})$$
- $$\boldsymbol{x}_i=\boldsymbol{x}_{i-1}+\alpha\boldsymbol{p}_i+\omega_i\boldsymbol{s}$$
- 若 $\boldsymbol{x}_i$ 足夠精確則退出
- $$\boldsymbol{r}_i=\boldsymbol{s}-\omega_i\boldsymbol{t}$$

### 預處理穩定雙共軛梯度法

預處理通常被用來加速迭代方法的收斂。要使用預處理子 $\boldsymbol{K}=\boldsymbol{K}_1\boldsymbol{K}_2\approx\boldsymbol{A}$ 來求解線性方程組 $\boldsymbol{Ax}=\boldsymbol{b}$，預處理穩定雙共軛梯度法從初始解 $\boldsymbol{x}_0$ 開始按以下步驟迭代：

- $$\boldsymbol{r}_0=\boldsymbol{b}-\boldsymbol{Ax}$$
- 任意選擇向量 $\boldsymbol{\hat{r}}_0\;$ 使得 $(\boldsymbol{\hat{r}}_0,\boldsymbol{r}_0) \neq0\;$，例如，$\boldsymbol{\hat{r}}_0=\boldsymbol{r}_0\;$
- $$\rho_0=\alpha=\omega_0=1\;$$
- $$\boldsymbol{v}_0=\boldsymbol{p}_0=\boldsymbol{0}$$
- 對 $i=1,2,3,\ldots$
- $$\rho_i=(\boldsymbol{\hat{r}}_0,\boldsymbol{r}_{i-1})\;$$
- $$\beta=(\rho_i/\rho_{i-1})(\alpha/\omega_{i-1})\;$$
- $$\boldsymbol{p}_i=\boldsymbol{r}_{i-1}+\beta(\boldsymbol{p}_{i-1}-\omega_{i-1}\boldsymbol{v}_{i-1})$$
- $$\boldsymbol{y}=\boldsymbol{K}^{-1}\boldsymbol{p}_i$$
- $$\boldsymbol{v}_i=\boldsymbol{Ay}$$
- $$\alpha=\rho_i/(\boldsymbol{\hat{r}}_0,\boldsymbol{v}_i)\;$$
- $$\boldsymbol{s}=\boldsymbol{r}_i-\alpha\boldsymbol{v}_i$$
- $$\boldsymbol{z}=\boldsymbol{As}$$
- $$\boldsymbol{t}=\boldsymbol{K}^{-1}\boldsymbol{z}$$
- $$\omega_i=(\boldsymbol{K}_1^{-1}\boldsymbol{t},\boldsymbol{K}_1^{-1}\boldsymbol{s})/(\boldsymbol{K}_1^{-1}\boldsymbol{t},\boldsymbol{K}_1^{-1}\boldsymbol{t})$$
- $$\boldsymbol{x}_i=\boldsymbol{x}_{i-1}+\alpha\boldsymbol{y}+\omega_i\boldsymbol{z}$$
- 若 $\boldsymbol{x}_i$ 足夠精確則退出
- $$\boldsymbol{r}_i=\boldsymbol{s}-\omega_i\boldsymbol{t}$$

這個形式等價於將無預處理的穩定雙共軛梯度法應用於顯式預處理後的方程組

$\boldsymbol{\tilde{A}\tilde{x}}=\boldsymbol{\tilde{b}}$。

其中 $\boldsymbol{\tilde{A}}=\boldsymbol{K}_1^{-1}\boldsymbol{AK}_2^{-1}$，$\boldsymbol{\tilde{x}}=\boldsymbol{K}_2\boldsymbol{x}$，$\boldsymbol{\tilde{b}}=\boldsymbol{K}_1^{-1}\boldsymbol{b}$。換句話說，左預處理和右預處理都可以通過這個形式實施。

## 推導

### 雙共軛梯度法的多項式形式

在雙共軛梯度法中，搜索方向 $\boldsymbol{p}_i$ 和 $\boldsymbol{\hat{p}}_i$ 以及殘量 $\boldsymbol{r}_i$ 和 $\boldsymbol{\hat{r}}_i$ 通過以下遞推關係更新：

$$\boldsymbol{p}_i=\boldsymbol{r}_i+\beta_i\boldsymbol{p}_{i-1}\text{,}$$

$$\boldsymbol{\hat{p}}_i=\boldsymbol{\hat{r}}_i+\beta_i\boldsymbol{p}_{i-1}\text{,}$$

$$\boldsymbol{r}_i=\boldsymbol{r}_{i-1}-\alpha_i\boldsymbol{Ap}_{i}\text{,}$$

$$\boldsymbol{\hat{r}}_i=\boldsymbol{\hat{r}}_{i-1}-\alpha\boldsymbol{A}^{\mathrm{T}}\boldsymbol{\hat{p}}_i\text{.}$$

常數 $\alpha_i\;$ 和 $\beta_i\;$ 取值為

$$\alpha_i=\rho_i/(\boldsymbol{\hat{p}}_i,\boldsymbol{Ap}_i)\text{,}$$

$$\beta_i=\rho_i/\rho_{i-1}\text{,}\;$$

其中 $\rho_i=(\boldsymbol{\hat{r}}_{i-1},\boldsymbol{r}_{i-1})$，使得殘量和搜索方向分別滿足雙正交性和雙共軛性，也就是對於 $i\neq j$。

$$(\boldsymbol{\hat{r}}_i,\boldsymbol{r}_j)=0\text{,}$$

$$(\boldsymbol{\hat{p}}_i,\boldsymbol{Ap}_j)=0\text{.}$$

不難證明。

$$\boldsymbol{r}_i=P_i(\boldsymbol{A})\boldsymbol{r}_0\text{,}$$

$$\boldsymbol{\hat{r}}_i=P_i(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0\text{,}$$

$$\boldsymbol{p}_{i+1}=T_i(\boldsymbol{A})\boldsymbol{r}_0\text{,}$$

$$\boldsymbol{\hat{p}}_{i+1}=T_i(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0\text{,}$$

其中 $P_i(\boldsymbol{A})$ 和 $T_i(\boldsymbol{A})$ 是關於 $\boldsymbol{A}$ 的 $i\;$ 次多項式。這些多項式滿足以下遞推關係：

$$P_i(\boldsymbol{A})=P_{i-1}(\boldsymbol{A})-\alpha_i\boldsymbol{A}T_{i-1}(\boldsymbol{A})\text{,}$$

$$T_i(\boldsymbol{A})=P_i(\boldsymbol{A})-\beta_{i+1}T_{i-1}(\boldsymbol{A})\text{.}$$

### 從雙共軛梯度法導出穩定雙共軛梯度 法

雙共軛梯度法的殘量和搜索方向不是必須顯式跟蹤的。換句話說，雙共軛梯度法的迭代是可以隱式進行的。穩定雙共軛梯度法中希望得到

$$\boldsymbol{\tilde{r}}_i=Q_i(\boldsymbol{A})P_i(\boldsymbol{A})\boldsymbol{r}_0$$

遞推關係，其中 $Q_i(\boldsymbol{A})=(\boldsymbol{I}-\omega_1\boldsymbol{A})(\boldsymbol{I}-\omega_1\boldsymbol{A})\cdots(\boldsymbol{I}-\omega_i\boldsymbol{A})$，$\omega_j\;$ 為適當選取的常數。以此代替 $\boldsymbol{r}_i=P_i(\boldsymbol{A})$ 的目的是希望 $Q_i(\boldsymbol{A})$ 可以使 $\boldsymbol{\tilde{r}}_i$ 有比 $\boldsymbol{r}_i$ 更快速和更平滑的收斂性。

根據 $P_i(\boldsymbol{A})$ 和 $T_i(\boldsymbol{A})$ 的遞推關係以及 $Q_i(\boldsymbol{A})$ 的定義。

$$Q_i(\boldsymbol{A})P_i(\boldsymbol{A})\boldsymbol{r}_0=(\boldsymbol{I}-\omega_i\boldsymbol{A})\bigl(Q_{i-1}(\boldsymbol{A})P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0-\alpha_i\boldsymbol{A}Q_{i-1}(\boldsymbol{A})P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)\text{.}$$

於是還需要一條關於 $Q_i(\boldsymbol{A})T_i(\boldsymbol{A})\boldsymbol{r}_0$ 的遞推關係。這同樣可以從雙共軛梯度法的遞推關係中導出：

$$Q_i(\boldsymbol{A})T_i(\boldsymbol{A})\boldsymbol{r}_0=Q_i(\boldsymbol{A})P_i(\boldsymbol{A})\boldsymbol{r}_0-\beta_{i+1}(\boldsymbol{I}-\omega_i\boldsymbol{A})Q_{i-1}(\boldsymbol{A})T_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\text{.}$$

類似於 $\boldsymbol{\tilde{r}}_i$，穩定雙共軛梯度法定義

$$\boldsymbol{\tilde{p}}_{i+1}=Q_i(\boldsymbol{A})T_i(\boldsymbol{A})\boldsymbol{r}_0\text{.}$$

寫成向量形式，$\boldsymbol{\tilde{p}}_i$ 和 $\boldsymbol{\tilde{r}}_i$ 的遞推關係就是

$$\boldsymbol{\tilde{p}}_i=\boldsymbol{\tilde{r}}_{i-1}+\beta_i(\boldsymbol{I}-\omega_{i-1}\boldsymbol{A})\boldsymbol{\tilde{p}}_{i-1}\text{,}$$

$$\boldsymbol{\tilde{r}}_i=(\boldsymbol{I}-\omega_i\boldsymbol{A})(\boldsymbol{\tilde{r}}_{i-1}-\alpha_i\boldsymbol{A\tilde{p}}_i)\text{.}$$

為了導出 $\boldsymbol{x}_i$ 的遞推關係，定義

$$\boldsymbol{s}_i=\boldsymbol{\tilde{r}}_{i-1}-\alpha_i\boldsymbol{A\tilde{p}}_i\text{.}$$

於是 $\boldsymbol{\tilde{r}}_i$ 的遞推關係就可以寫成

$$\boldsymbol{\tilde{r}}_i=\boldsymbol{\tilde{r}}_{i-1}-\alpha_i\boldsymbol{A\tilde{p}}_i-\omega_i\boldsymbol{As}_i\text{,}$$

這對應於

$$\boldsymbol{x}_i=\boldsymbol{x}_{i-1}+\alpha_i\boldsymbol{\tilde{p}}_i+\omega_i\boldsymbol{s}_i\text{.}$$

### 確定穩定雙共軛梯度法的常數

現在只需確定雙共軛梯度法的常數 $\alpha_i\;$ 和 $\beta_i\;$ 以及選擇一個合適的 $\omega_i\;$。

在雙共軛梯度法中，$\beta_i=\rho_i/\rho_{i-1}\;$，其中

$$\rho_i=(\boldsymbol{\hat{r}}_{i-1},\boldsymbol{r}_{i-1})=\bigl(P_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)\text{.}$$

由於穩定雙共軛梯度法不顯式跟蹤 $\boldsymbol{\hat{r}}_i$ 或 $\boldsymbol{r}_i$，$\rho_i\;$ 不能立即用這條公式計算出來。但是，它可以和標量

$$\tilde{\rho}_i=\bigl(Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)=\bigl(\boldsymbol{\hat{r}}_0,Q_{i-1}(\boldsymbol{A})P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)=(\boldsymbol{\hat{r}}_0,\boldsymbol{r}_{i-1})$$

關聯起來。由於雙正交性，$\boldsymbol{r}_{i-1}=P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0$ 正交於 $U_{i-2}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0$，其中 $U_{i-2}(\boldsymbol{A}^{\mathrm{T}})$ 是關於 $\boldsymbol{A}^{\mathrm{T}}$ 的任意 $i-2\;$ 次多項式。因此在點積 $\bigl(P_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)$ 和 $\bigl(Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)$ 中只需考慮 $P_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 和 $Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 的最高次項。$P_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 和 $Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 的最高次項係數分別是 $(-1)^{i-1}\alpha_1\alpha_2\cdots\alpha_{i-1}$ 和 $(-1)^{i-1}\omega_1\omega_2\cdots\omega_{i-1}$。因此

$$\rho_i=(\alpha_1/\omega_1)(\alpha_2/\omega_2)\cdots(\alpha_{i-1}/\omega_{i-1})\tilde{\rho}_i\text{,}$$

於是

$$\beta_i=\rho_i/\rho_{i-1}=(\tilde{\rho}_i/\tilde{\rho}_{i-1})(\alpha_{i-1}/\omega_{i-1})\text{.}$$

關於 $\alpha_i\;$ 的簡單公式可以類似地導出。在雙共軛梯度法中。

$$\alpha_i=\rho_i/(\boldsymbol{\hat{p}},\boldsymbol{Ap}_i)=\bigl(P_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)\big/\bigl(T_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,\boldsymbol{A}T_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)\text{.}$$

類似於上面的情況，由於雙正交性和雙共軛性，在點積中只需考慮 $P_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 和 $T_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 的最高次項。$P_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 和 $T_{i-1}(\boldsymbol{A}^{\mathrm{T}})$ 的最高次項係數恰巧是相同的。因此，它們可以在公式中被同時替換為 $Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})$，於是

$$\alpha_i=\bigl(Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,P_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)\big/\bigl(Q_{i-1}(\boldsymbol{A}^{\mathrm{T}})\boldsymbol{\hat{r}}_0,\boldsymbol{A}T_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)=\tilde{\rho}_i\big/\bigl(\boldsymbol{\hat{r}}_0,\boldsymbol{A}Q_{i-1}(\boldsymbol{A})T_{i-1}(\boldsymbol{A})\boldsymbol{r}_0\bigr)=\tilde{\rho}_i/(\boldsymbol{\hat{r}}_0,\boldsymbol{A\tilde{p}}_i)\text{.}$$

最後，穩定雙共軛梯度法選擇 $\omega_i\;$ 使得 $\boldsymbol{\tilde{r}}_i=(\boldsymbol{I}-\omega_i\boldsymbol{A})\boldsymbol{s}_i$ 的 2-範數作為 $\omega_i\;$ 的函數被最小化。這在

$$\bigl((\boldsymbol{I}-\omega_i\boldsymbol{A})\boldsymbol{s}_i,\boldsymbol{As}_i\bigr)=0$$

時達到，因此 $\omega_i\;$ 的最優值是

$$\omega_i=(\boldsymbol{As}_i,\boldsymbol{s}_i)/(\boldsymbol{As}_i,\boldsymbol{As}_i)\text{.}$$