提问者:小点点

用Cholesky分解求numpy中矩阵的求逆效率


我有一个对称的正定矩阵(例如协方差矩阵),我想计算它的逆。在数学中,我知道用Cholesky分解来求矩阵的逆是更有效率的,特别是如果你的矩阵很大。但我不确定“numpy.lianlg.inv()”是如何工作的。假设我有以下代码:

import numpy as np

X = np.arange(10000).reshape(100,100)
X = X + X.T - np.diag(X.diagonal()) #  symmetry 
X = np.dot(X,X.T) # positive-definite

# simple inversion:
inverse1 = np.linalg.inv(X) 

# Cholesky decomposition inversion:
c = np.linalg.inv(np.linalg.cholesky(X))
inverse2 = np.dot(c.T,c)

哪一个效率更高(inverse1还是inverse2)?如果第二个效率更高,为什么numpy.linalg.inv()不使用这个呢?


共1个答案

匿名用户

使用以下设置:

import numpy as np

N = 100
X = np.linspace(0, 1, N*N).reshape(N, N)
X = 0.5*(X + X.T) + np.eye(N) * N

我使用获得以下计时:

In [28]: %timeit np.linalg.inv(X)
255 µs ± 30.9 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

In [29]: %timeit c = np.linalg.inv(np.linalg.cholesky(X)); np.dot(c.T,c)
414 µs ± 15.4 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)