Skip to content

算法Algorithm

预条件共轭梯度法

Preconditioned conjugate gradient method · PCG · 预条件共轭梯度

从固定正定预条件器的对称坐标变换推导 PCG,以加权残差生成共轭方向,并用三步算例区分能量收敛、原残差与总成本。

形式陈述 ​

预条件共轭梯度法(PCG)求解 Ax=b,其中 A=A∗≻0。除矩阵—向量乘 v↦Av 外,算法需要一个固定的线性接口

apply(v)=M−1v,M=M∗≻0.

这里的Hermitian 正定条件同时约束原矩阵与预条件器。实现可以只提供 apply,不显式存储 M 或其逆;所提供的映射必须在所有迭代中保持同一个线性、自伴、正定算子。M 近似 A 是为了改善收敛,M−1v 易于计算则是获得实际加速的前提。

给定 b,x0,先计算真实残差 r0=b−Ax0。若它已满足停止准则,就直接返回 x0;否则初始化

z0=M−1r0,ρ0=r0∗z0,p0=z0.

第 k 步先计算

vk=Apk,αk=ρkpk∗vk,

再更新

xk+1=xk+αkpk,rk+1=rk−αkvk.

若新近似通过真实残差检查,则结束;若继续迭代,则计算

zk+1=M−1rk+1,ρk+1=rk+1∗zk+1,βk=ρk+1ρk,pk+1=zk+1+βkpk.

精确算术中,只要尚未得到解,ρk>0、pk∗Apk>0,步长和继续迭代时的 βk 都是正实数;复数数据必须使用共轭转置。M=I 时,这些公式逐项退化为普通共轭梯度法。两处标量递推都使用 ρk=rk∗M−1rk,不能只把方向中的 rk 换成 zk,却保留普通 CG 的 rk∗rk。

接口还应接受非负的绝对、相对容差和工作预算,并返回原变量 x、真实残差、迭代次数、A 与 apply 的调用次数及退出状态。本文采用的成功准则是

‖b−Ax‖2≤atol+rtol‖b‖2.

这一定义对 b=0 仍有意义;若此时也取 atol=0,要求的便是零残差。内部的 ρk 不是这个停止量。

直觉

预条件改变 CG 观察问题的坐标。取分解 M=CC∗,令

B=C−1AC−∗,c=C−1b,y=C∗x,C−∗=(C∗)−1.

C 非奇异,且对非零 u 有

u∗Bu=(C−∗u)∗A(C−∗u)>0,

所以 B 仍是 Hermitian 正定矩阵,可以对 By=c 应用普通 CG。以下推导使用普通 CG 已有的正交性与最小化定理,证明它如何成为上述 PCG;并不另行重证普通 CG 定理。

记变换坐标中的残差和搜索方向为 sk,qk,映回原变量时定义

sk=c−Byk=C−1rk,pk=C−∗qk.

普通 CG 的初始方向 q0=s0 因而给出

p0=C−∗C−1r0=M−1r0=z0.

它的步长分子和分母分别变成

sk∗sk=rk∗C−∗C−1rk=rk∗zk=ρk,qk∗Bqk=pk∗Apk.

因此 αk=(sk∗sk)/(qk∗Bqk) 正好给出 PCG 的步长。对普通更新 yk+1=yk+αkqk 左乘 C−∗,便得到原变量的 x 更新;对 sk+1=sk−αkBqk 左乘 C,便得到 rk+1=rk−αkApk。剩下两条普通 CG 递推为

βk=sk+1∗sk+1sk∗sk=ρk+1ρk,qk+1=sk+1+βkqk.

第二式左乘 C−∗ 后成为 pk+1=M−1rk+1+βkpk,完整解释了 apply 和方向更新的来源。C 只用于证明,执行算法不必构造或调用它。

同一映射也确定了正确的内积几何。普通 CG 的 si∗sj=0 与 qi∗Bqj=0 分别给出

ri∗M−1rj=0,pi∗Apj=0(i≠j).

原残差在 M−1 内积中正交,搜索方向仍在原来的 A 内积中共轭。通常并没有 ri∗rj=0;把普通 CG 的 Euclidean 残差正交性直接搬到原变量,会遗漏整个坐标变化。

令 T=M−1A。它通常不是 Euclidean Hermitian 矩阵,因为 T∗=AM−1;但它在 M 内积中满足

⟨u,Tv⟩M=u∗Av=⟨Tu,v⟩M,⟨u,Tu⟩M=u∗Au>0.

此外 C∗TC−∗=B,故 T 与正定矩阵 B 相似,具有相同的正特征值。PCG 的几何来自这种带权自伴性,不能由“左乘以后看起来更好解”替代。

例子与边界

取一个只需二阶块求解和一次缩放的预条件器:

C=(100120004),M=CCT=(1101500016).

令

B=(210121012),A=CBCT=(24041480832),b=(110),x0=0.

B 的特征值为 2−2,2,2+2,所以 A,M 均正定。这里 M≠A,且 (AM)12=22、(MA)12=18,两者并不交换;因而此例保留了左预条件算子不对称的真实情形。对向量 r=(r(1),r(2),r(3))T,apply 明确为

M−1r=((5r(1)−r(2))/4(−r(1)+r(2))/4r(3)/16).

以下均为精确分数,表中三元组表示列向量。先列出每个时刻的解、残差和预条件残差:

k xk rk zk
0 (0,0,0) (1,1,0) (1,0,0)
1 (1/2,0,0) (0,−1,0) (1/4,−1/4,0)
2 (5/6,−1/6,0) (0,0,4/3) (0,0,1/12)
3 (1,−1/4,1/16) (0,0,0) 无须计算

产生这些状态的三个搜索方向及其矩阵乘积为

p0=(1,0,0)T,Ap0=(2,4,0)T,p1=(1/2,−1/4,0)T,Ap1=(0,−3/2,−2)T,p2=(2/9,−1/9,1/12)T,Ap2=(0,0,16/9)T.

标量更新与误差检查单独列出。最后一行的 ρ3=0 是由零残差得出的值,不需要再执行 apply。

k ρk αk βk ‖rk‖22 ‖x∗−xk‖A2
0 1 1/2 1/4 2 3/4
1 1/4 2/3 4/9 1 1/4
2 1/9 3/4 停止 16/9 1/12
3 0 — — 0 0

例如第二步的分母为 p1TAp1=3/8,所以 α1=(1/4)/(3/8)=2/3;它把 (0,−1,0)T 更新为 (0,0,4/3)T。第三步的分母为 4/27,所以 α2=(1/9)/(4/27)=3/4,直接得到 Ax3=b。这完整执行了三步递推,而非仅展示初始与最终状态。

这个历史同时区分了三种量。首先 r0Tr1=−1,但 r0TM−1r1=0,正交性确实依赖权重。其次,第二步的原残差范数由 1 增到 4/3,能量误差平方却由 1/4 降到 1/12。最后,ρk 与能量误差平方的数值也不相同;本例中二者恰好都下降,不能据此把预条件残差当作能量误差。

谱改善也可以精确检查:

κ2(B)=2+22−2=3+22<6.

原矩阵的坐标 Rayleigh 商给出 λmax(A)≥32、λmin(A)≤2,从而 κ2(A)≥16。廉价块预条件确实缩小了谱跨度;它没有把 B 变成单位阵,也没有消除所有耦合。

固定线性正定的 apply 是上述推导的边界。若每一步改变内层求解容差,或使用依赖输入的非线性近似过程,就不再有同一个 C 和同一个 B;普通 ρk+1/ρk 公式因而失去这里给出的依据。即使每次返回的向量都看似合理,也需要适用于该变化方式的算法分析。

推论与应用

令 ek=x∗−xk。坐标映射保持能量误差,因为

(C∗ek)∗B(C∗ek)=ek∗Aek.

又因 C−∗B=TC−∗ 且 C−∗s0=z0,有

C−∗Kk(B,s0)=Kk(T,z0).

将普通 CG 的最小化定理沿这两个等式映回,就得到 PCG 在原变量中的精确结论:

xk∈x0+Kk(M−1A,z0),‖x∗−xk‖A=minx∈x0+Kk(M−1A,z0)‖x∗−x‖A.

因此能量误差随嵌套的Krylov 子空间单调不增。若 s0 关于 B 的最小多项式次数为 d,精确算术中至多 d≤n 步终止;浮点运算中不保证第 n 步得到零残差。

普通 CG 的谱界同样映回为

‖ek‖A≤2(κ−1κ+1)k‖e0‖A,κ=κ2(B)=λmax(B)λmin(B).

这里使用对称正定 B 的条件数。虽然 T=M−1A 与 B 的特征值相同,非酉相似一般不保持奇异值,因此不能把上式中的 κ 换成 Euclidean 奇异值条件数 κ2(T)。该界只使用谱端点;特征值聚集与初始误差的分量还会影响实际迭代数。

对大型稀疏 SPD 系统,标准 PCG 每个继续迭代的周期需要一次 A 乘法、一次 apply、两个主要内积和 O(n) 向量更新;除算子及预条件结构外,只需常数个长向量。若完成 k≥1 次更新并在最后一次更新后直接停止,apply 的调用数是初始一次加中间 k−1 次,共 k 次。初始残差计算和另行重算真实残差所需的 A 乘法也必须记录;已知 x0=0 时可省略初始的零向量乘法。

把设置、应用和检查分开,总成本可写为

Csetup+NACA+NMCM+O(kn)+Cchecks,

其中 NA 包含初始及验证残差的矩阵乘法,NM 是实际 apply 次数,Cchecks 计入额外范数与通信归约。预条件器的因子、填充或层级存储也属于成本。比较方法时,应同时报告设置时间与总求解时间;多个右端能摊销设置,而一次短求解未必能回本。

有限精度会使递推残差偏离 b−Axk,并使精确正交性逐渐丢失。可以用递推残差筛选候选停止点,再以重算的真实残差决定成功;长迭代也可定期检查漂移。如果把重算残差替换进迭代并继续,应重新设置 z=M−1r、p=z、ρ=r∗z,作为一次重启,而不是保留旧方向并继续声称完整共轭性。

非零残差下的 ρ≤0、p∗Ap≤0 或非有限输出,与精确固定 SPD 合约不相容,实际实现应停止并报告数值故障或接口失败;达到预算上限则返回未收敛状态及真实残差。初始零残差是正常成功路径,须在任何 apply 或步长除法之前处理。这样,理论中的有限步终止、用户要求的原系统精度和机器上的实际退出原因才有各自明确的含义。

参考资料
关系图谱11 个相邻概念 · 2 类关系

拖动节点调整位置。

显示关系

显示:依赖

  1. 前置三跳
  2. 前置二跳
  3. 前置一跳
  4. 当前条目
  5. 后续一跳
  6. 后续二跳
  7. 后续三跳
文字版关系按与当前条目的最短距离分组
类型化关系