Python如何實現兩種稀疏矩陣的最小二乘法

今天小編給大家分享一下Python如何實現兩種稀疏矩陣的最小二乘法的相關知識點，內容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

最小二乘法

scipy.sparse.linalg實現了兩種稀疏矩陣最小二乘法lsqr和lsmr，前者是經典算法，后者來自斯坦福優化實驗室，據稱可以比lsqr更快收斂。

這兩個函數可以求解Ax=b，或arg min_x ∥Ax−b∥²，或arg min_x ∥Ax−b∥² +d²∥x−x0∥²，其中A必須是方陣或三角陣，可以有任意秩。

通過設置容忍度a_t ,b_t,可以控制算法精度，記r=b-A_x 為殘差向量，如果A_x=b是相容的，lsqr在∥r∥?a_t∗∥A∥⋅∥x∥+b_t∥b∥時終止；否則將在∥A^T_r∥?a_t∥A∥⋅∥r∥。

如果兩個容忍度都是10^−6 ，最終的∥r∥將有6位精度。

lsmr的參數如下

lsmr(A, b, damp=0.0, atol=1e-06, btol=1e-06, conlim=100000000.0, maxiter=None, show=False, x0=None)

參數解釋：

• A 可謂稀疏矩陣、數組以及線性算子

• b 為數組

• damp 阻尼系數，默認為0

• atol, btol 截止容忍度，是lsqr迭代的停止條件，即a_t ,b_t 。

• conlim 另一個截止條件，對于最小二乘問題，conlim應該小于10⁸，如果A_x=b是相容的，則conlim最大可以設到10¹²

• iter_limint 迭代次數

• show 如果為True，則打印運算過程

• calc_var 是否估計(A.T@A + damp**2*I)^{-1}的對角線

• x0 阻尼系數相關

lsqr和lsmr相比，沒有maxiter參數，但多了iter_lim, calc_va參數。

上述參數中，damp為阻尼系數，當其不為0時，記作δ，待解決的最小二乘問題變為

返回值

lsmr的返回值依次為：

• x 即A_x=b中的x

• istop 程序結束運行的原因

• itn 迭代次數

• normr ∥b−A_x∥

• normar ∥A^T (b−Ax)∥

• norma ∥A∥

• conda A的條件數

• normx ∥x∥

lsqr的返回值為

1. x 即A_x=b中的x

2. istop 程序結束運行的原因

3. itn 迭代次數

4. r1norm

5. anorm 估計的Frobenius范數Aˉ

6. acond Aˉ的條件數

7. arnorm ∥A^T_r−δ²(x−x₀)∥

8. xnorm ∥x∥

9. var (A^TA)^−1

二者的返回值較多，而且除了前四個之外，剩下的意義不同，調用時且須注意。

測試

下面對這兩種算法進行驗證，第一步就得先有一個稀疏矩陣

import numpy as np
from scipy.sparse import csr_array

np.random.seed(42)  # 設置隨機數狀態
mat = np.random.rand(500,500)
mat[mat<0.9] = 0
csr = csr_array(mat)

然后用這個稀疏矩陣乘以一個x，得到b

xs = np.arange(500)
b = mat @ xs

接下來對這兩個最小二乘函數進行測試

from scipy.sparse.linalg import lsmr, lsqr
import matplotlib.pyplot as plt
mx = lsmr(csr, b)[0]
qx = lsqr(csr, b)[0]
plt.plot(xs, lw=0.5)
plt.plot(mx, lw=0, marker='*', label="lsmr")
plt.plot(qx, lw=0, marker='.', label="lsqr")
plt.legend()
plt.show()

為了對比清晰，對圖像進行放大，可以說二者不分勝負

接下來比較二者的效率，500 × 500 500\times500500×500這個尺寸顯然已經不合適了，用2000×2000

from timeit import timeit

np.random.seed(42)  # 設置隨機數狀態
mat = np.random.rand(500,500)
mat[mat<0.9] = 0
csr = csr_array(mat)
timeit(lambda : lsmr(csr, b), number=10)
timeit(lambda : lsqr(csr, b), number=10)

測試結果如下

>>> timeit(lambda : lsqr(csr, b), number=10)
0.5240591000001587
>>> timeit(lambda : lsmr(csr, b), number=10)
0.6156221000019286

看來lsmr并沒有更快，看來斯坦福也不靠譜(滑稽)。

以上就是“Python如何實現兩種稀疏矩陣的最小二乘法”這篇文章的所有內容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業資訊頻道。