数值计算之 最小二乘法(3)最小二乘的矩阵解法

2023-11-15

前言

之前将最小二乘法与线性方程组求解关联,得到了线性最小二乘的矩阵形式,以及线性最小二乘的几何意义。

本篇将介绍线性最小二乘的矩阵解法。

回顾最小二乘的线性解

对于线性超定方程组 A x = b , A ∈ R m × n , m > n Ax=b,A\in R^{m\times n},m>n Ax=b,ARm×n,m>n,其最小二乘解可表示为:
arg min ⁡ x ∣ ∣ A x − b ∣ ∣ 2 2 = arg min ⁡ x ( A x − b ) T ( A x − b ) → A T A x = A T b i f r a n k ( A ) = n , x = ( A T A ) − 1 A T b \argmin_x ||Ax-b||^2_2=\argmin_x (Ax-b)^T(Ax-b) \\ \to A^TAx=A^Tb \\ \quad \\ if \quad rank(A)=n, x=(A^TA)^{-1}A^Tb xargminAxb22=xargmin(Axb)T(Axb)ATAx=ATbifrank(A)=n,x=(ATA)1ATb

以上解法具有两个问题:① A T A A^TA ATA求逆运算的效率;② r a n k ( A ) < n rank(A)<n rank(A)<n时的最小二乘解。

首先讨论问题①。

列满秩矩阵的最小二乘解法

对于线性方程组 A x = b , A ∈ R m × n , m > n , r a n k ( A ) = n Ax=b,A\in R^{m\times n},m>n,rank(A)=n Ax=b,ARm×n,m>n,rank(A)=n,有 A T A A^TA ATA对称且可逆。回顾矩阵分解的知识,可知 A T A A^TA ATA矩阵能够进行Cholesky分解和QR分解。

Cholesky分解求线性最小二乘解

通过将 A T A A^TA ATA分解为下三角矩阵 L L L的乘积 L L T LL^T LLT
A T A = L L T A^TA=LL^T ATA=LLT
则可以将复杂的线性方程组求解:
A T A x = A T b A^TAx=A^Tb ATAx=ATb
转化为两个简单的三角线性方程组求解:
L L T x = A T b → L T x = y , L y = A T b LL^Tx=A^Tb \to L^Tx=y, Ly=A^Tb LLTx=ATbLTx=y,Ly=ATb

Cholesky分解速度很快,但精度一般,稳定性差。适合在限定时间内的大规模超定线性方程计算求解。

QR分解求线性最小二乘解

对于列满秩矩阵 A A A而言,可以唯一分解为正交矩阵与对角元为正的上三角矩阵的乘积:
A = Q R = Q [ R 0 ] A=QR=Q\begin{bmatrix} R \\ 0 \end{bmatrix} A=QR=Q[R0]
现在考虑最小二乘法的QR分解法:
arg min ⁡ x ∣ ∣ A x − b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ Q [ R 0 ] x − b ∣ ∣ = arg min ⁡ x ∣ ∣ [ R 0 ] x − Q T b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ [ R x 0 ] − [ β 1 β 2 ] ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ R x − β 1 ∣ ∣ 2 2 + ∣ ∣ β 2 ∣ ∣ 2 2    ⟺    arg min ⁡ x ∣ ∣ R x − β 1 ∣ ∣ 2 2 → R x = β 1 , x = R − 1 β 1 [ β 1 β 2 ] = Q T b \argmin_x ||Ax-b||^2_2=\argmin_x ||Q\begin{bmatrix} R \\ 0 \end{bmatrix}x-b|| \\ = \argmin_x ||\begin{bmatrix} R \\ 0 \end{bmatrix}x-Q^Tb||^2_2 \\ = \argmin_x ||\begin{bmatrix} Rx \\ 0 \end{bmatrix} - \begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix}||^2_2 \\ = \argmin_x ||Rx-\beta_1||^2_2 + ||\beta_2||^2_2 \\ \iff \argmin_x ||Rx-\beta_1||^2_2 \\ \to Rx=\beta_1,x=R^{-1}\beta_1 \\ \quad \\ \begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix} = Q^Tb xargminAxb22=xargminQ[R0]xb=xargmin[R0]xQTb22=xargmin[Rx0][β1β2]22=xargminRxβ122+β222xargminRxβ122Rx=β1,x=R1β1[β1β2]=QTb

QR分解的速度较快,精度一般。不过由于存在高精度的QR分解方式,因此适合需要高精度解的小规模超定方程组计算。

亏秩矩阵的最小二乘解法

对于线性方程组 A x = b , A ∈ R m × n , m > n , r a n k ( A ) < n Ax=b,A\in R^{m\times n},m>n,rank(A)<n Ax=b,ARm×n,m>n,rank(A)<n,称为亏秩超定方程组。此时 A T A A^TA ATA不是正定矩阵,QR分解也不是唯一的。通常使用SVD来解决亏秩问题。

SVD分解求亏秩最小二乘解

对于矩阵 A m × n , m > n , r a n k ( A ) = r < n A_{m\times n},m>n,rank(A)=r<n Am×n,m>n,rank(A)=r<n,可分解为正交矩阵与奇异值矩阵的乘积:
A = U m × m [ Σ r × r 0 r × ( n − r ) 0 ( m − r ) × r 0 ( m − r ) × ( n − r ) ] V n × n T U T U = E , V T V = E A=U_{m\times m}\begin{bmatrix}\Sigma_{r\times r} & 0_{r\times (n-r)} \\ 0_{(m-r)\times r} & 0_{(m-r)\times (n-r)}\end{bmatrix} V_{n\times n}^T \\ \quad \\ U^TU=E,V^TV=E A=Um×m[Σr×r0(mr)×r0r×(nr)0(mr)×(nr)]Vn×nTUTU=E,VTV=E
现在考虑最小二乘法的SVD解法:
arg min ⁡ x ∣ ∣ A x − b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ U T A x − U T b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ U T U [ Σ 0 0 0 ] V T x − U T b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ [ Σ 0 0 0 ] V T x − U T b ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ [ Σ 0 0 0 ] [ α 1 α 2 ] − [ β 1 β 2 ] ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ [ Σ α 1 0 ] − [ β 1 β 2 ] ∣ ∣ 2 2 = arg min ⁡ x ∣ ∣ Σ α 1 − β 1 ∣ ∣ 2 2 + ∣ ∣ β 2 ∣ ∣ 2 2 → Σ α 1 = β 1 [ α 1 α 2 ] = V T x , [ β 1 β 2 ] = U T b α 1 = Σ − 1 β 1 , x = V [ α 1 α 2 ] \argmin_x ||Ax-b||^2_2 = \argmin_x ||U^TAx-U^Tb||^2_2 \\ = \argmin_x ||U^TU\begin{bmatrix}\Sigma & 0 \\ 0 & 0 \end{bmatrix}V^Tx-U^Tb||^2_2 \\ = \argmin_x ||\begin{bmatrix}\Sigma & 0 \\ 0 & 0 \end{bmatrix}V^Tx-U^Tb||^2_2 \\ = \argmin_x ||\begin{bmatrix}\Sigma & 0 \\ 0 & 0 \end{bmatrix}\begin{bmatrix} \alpha_1 \\ \alpha_2 \end{bmatrix}-\begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix}||^2_2 \\ = \argmin_x ||\begin{bmatrix}\Sigma\alpha_1 \\ 0 \end{bmatrix}-\begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix}||^2_2 \\ = \argmin_x ||\Sigma\alpha_1-\beta_1||^2_2 +||\beta_2||^2_2 \\ \to \Sigma\alpha_1=\beta_1 \\ \quad \\ \begin{bmatrix} \alpha_1 \\ \alpha_2 \end{bmatrix} = V^Tx,\begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix}=U^Tb \\ \quad \\ \alpha_1=\Sigma^{-1}\beta_1,x =V\begin{bmatrix} \alpha_1 \\ \alpha_2 \end{bmatrix} \\ \quad \\ xargminAxb22=xargminUTAxUTb22=xargminUTU[Σ000]VTxUTb22=xargmin[Σ000]VTxUTb22=xargmin[Σ000][α1α2][β1β2]22=xargmin[Σα10][β1β2]22=xargminΣα1β122+β222Σα1=β1[α1α2]=VTx,[β1β2]=UTbα1=Σ1β1,x=V[α1α2]
在上面的解法中,由于 α 2 \alpha_2 α2是一个任意向量,因此求出的 x x x不是唯一的。即亏秩方程的最小二乘解不唯一。可以选择范数最小的最小二乘解作为亏秩方程的解,即 α 2 = 0 ⃗ \alpha_2=\vec 0 α2=0

可以通过正交矩阵分块,将SVD最小二乘解进一步化简:
U = [ U 1 , U 2 ] , V = [ V 1 , V 2 ] U 1 ∈ R m × r , V 1 ∈ R n × r α 1 = Σ − 1 β 1 [ α 1 α 2 ] = [ V 1 T V 2 T ] x [ β 1 β 2 ] = [ U 1 T U 2 T ] b V 1 T x = Σ − 1 U 1 T b x = V 1 Σ − 1 U 1 T b U=[U_1,U_2],V=[V_1,V_2] \\ U_1\in R^{m\times r},V_1\in R^{n\times r} \\ \quad \\ \alpha_1=\Sigma^{-1}\beta_1 \\ \quad \\ \begin{bmatrix} \alpha_1 \\ \alpha_2 \end{bmatrix} = \begin{bmatrix} V_1^T \\ V_2^T \end{bmatrix} x \\ \quad \\ \begin{bmatrix} \beta_1 \\ \beta_2 \end{bmatrix} = \begin{bmatrix} U_1^T \\ U_2^T \end{bmatrix} b \\ \quad \\ V_1^Tx=\Sigma^{-1} U_1^Tb \\ \quad \\ x=V_1\Sigma^{-1} U_1^Tb \\ U=[U1,U2],V=[V1,V2]U1Rm×r,V1Rn×rα1=Σ1β1[α1α2]=[V1TV2T]x[β1β2]=[U1TU2T]bV1Tx=Σ1U1Tbx=V1Σ1U1Tb

需要说明的是,SVD也适用于列满秩矩阵的最小二乘求解实际上SVD分解是最常用的线性最小二乘解法之一

补充1:超定齐次方程组的线性最小二乘解法

之前的最小二乘法都在考虑 A x = b Ax=b Ax=b的最小非齐次方程组问题。现在补充齐次方程组的最小二乘解法。

对于齐次方程组 A m × n x = 0 , m > n A_{m\times n}x=0,m>n Am×nx=0,m>n而言,其最小二乘解就是 A A A的SVD分解后的 V V V的最后一个列向量

证明:
A = U [ Σ 0 ] V T A x = U [ Σ 0 ] V T x = 0 y = V T x , Σ y = 0 也 就 是 说 , 求 A x = 0 , 转 换 为 求 Σ y = 0 的 最 小 二 乘 解 arg min ⁡ y ∣ ∣ Σ y ∣ ∣ 2 2 = arg min ⁡ y y T Σ T Σ y = arg min ⁡ y ∑ i = 1 n σ i 2 y i 2 s . t . ∣ ∣ y ∣ ∣ > 0 Σ 对 角 线 元 素 由 大 到 小 排 列 , 则 满 足 y = [ 0 , 0 , … , 0 , y n ] T , y n ≠ 0 时 , 获 得 Σ y 的 最 小 二 乘 解 V T x = y , x = V y = y n v n i f ∣ ∣ y ∣ ∣ 2 = 1 t h e n x = v n A=U\begin{bmatrix} \Sigma \\ 0 \end{bmatrix} V^T \\ \quad \\ Ax= U\begin{bmatrix} \Sigma \\ 0 \end{bmatrix} V^Tx=0 \\ \quad \\ y=V^Tx,\Sigma y=0 \\ \quad \\ 也就是说,求Ax=0,转换为求\Sigma y=0的最小二乘解 \\ \quad \\ \argmin_y ||\Sigma y||^2_2 \\ =\argmin_y y^T\Sigma^T\Sigma y \\ =\argmin_y \sum_{i=1}^n \sigma_i^2 y_i^2 \\ s.t. \quad ||y||>0 \\ \quad \\ \Sigma 对角线元素由大到小排列,则满足 \\ \quad \\ y=[0,0,\dots,0,y_n]^T,y_n\ne 0 \\ \quad \\ 时,获得\Sigma y的最小二乘解 \\ \quad \\ V^Tx=y,x=Vy=y_nv_n \\ \quad \\ if \quad ||y||_2=1 \\ then \quad x=v_n \\ A=U[Σ0]VTAx=U[Σ0]VTx=0y=VTx,Σy=0Ax=0Σy=0yargminΣy22=yargminyTΣTΣy=yargmini=1nσi2yi2s.t.y>0Σ线y=[0,0,,0,yn]T,yn=0ΣyVTx=y,x=Vy=ynvnify2=1thenx=vn
进一步,超定齐次线性方程组 A x = 0 Ax=0 Ax=0的最小二乘解还等于 A T A A^TA ATA的最小特征值对应的特征向量:

证明:
A = U [ Σ 0 ] V T A T A = V [ Σ 0 ] [ Σ 0 ] V T = V Σ 2 V T A T A x = V Σ 2 V T x = 0 y = V T x , Λ = Σ 2 , Λ y = 0 A=U\begin{bmatrix} \Sigma \\ 0 \end{bmatrix} V^T \\ \quad \\ A^TA = V \begin{bmatrix} \Sigma & 0 \end{bmatrix}\begin{bmatrix} \Sigma \\ 0 \end{bmatrix}V^T=V\Sigma^2V^T \\ \quad \\ A^TAx = V\Sigma^2V^Tx=0 \\ y=V^Tx,\Lambda=\Sigma^2,\Lambda y=0 \\ \quad \\ A=U[Σ0]VTATA=V[Σ0][Σ0]VT=VΣ2VTATAx=VΣ2VTx=0y=VTx,Λ=Σ2,Λy=0
后面的证明就与上面一模一样了。

补充2:欠定行满秩方程组的线性最小二乘解法

还需要补充的是欠定方程组的最小二乘解法。虽然用的少,但却实实在在的碰到了这个问题:对极几何中本质矩阵的求解。

首先需要确定,线性齐次欠定方程组必然存在无穷组解析解。

对于非齐次欠定方程组 A m × n x = b , m < n , r a n k ( A ) = r A_{m\times n}x=b,m<n,rank(A)=r Am×nx=b,m<n,rank(A)=r,其最小二乘也是通过SVD求解:
A = [ U 1 m × r , U 2 m × ( m − r ) ] [ Σ r × r 0 0 0 ( m − r ) × ( n − r ) ] [ V 1 n × r , V 2 n × ( n − r ) ] T A x = U 1 Σ V 1 T x = b y = V 1 T x , U 1 Σ y = b 特 别 的 , 如 果 r a n k ( A ) = m , 有 U Σ y = b , y = Σ − 1 U T b , x = V Σ − 1 U T b A=[U_1^{m\times r},U_2^{m\times (m-r)}]\begin{bmatrix} \Sigma^{r\times r} & 0 \\ 0 & 0^{(m-r)\times (n-r)} \end{bmatrix} [V_1^{n\times r},V_2^{n\times (n-r)}]^T \\ \quad \\ Ax = U_1\Sigma V_1^Tx=b \\ y=V_1^Tx,U_1\Sigma y=b \\ \quad \\ 特别的,如果rank(A)=m,有 \\ \quad \\ U\Sigma y=b,y=\Sigma^{-1}U^Tb,x=V\Sigma^{-1}U^Tb A=[U1m×r,U2m×(mr)][Σr×r000(mr)×(nr)][V1n×r,V2n×(nr)]TAx=U1ΣV1Tx=by=V1Tx,U1Σy=brank(A)=mUΣy=b,y=Σ1UTb,x=VΣ1UTb

后记

本篇介绍了最小二乘的矩阵解法,包括列满秩超定方程组的Cholesky分解法和QR分解法,以及亏秩超定方程组的SVD解法。

后续将继续学习非线性方程组的最小二乘问题。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

数值计算之 最小二乘法(3)最小二乘的矩阵解法 的相关文章

随机推荐