[英]Most efficient way to calculate pairwise partial correlations in base R?
問題標題說明了這一點,計算每個其他變量控制矩陣的每列之間的成對偏相關的最有效方法是什么?
基本上,類似於下面的cor
函數,但導致部分相關而不是簡單的相關。
#> cor(iris[,-5])
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 -0.1175698 0.8717538 0.8179411
#Sepal.Width -0.1175698 1.0000000 -0.4284401 -0.3661259
#Petal.Length 0.8717538 -0.4284401 1.0000000 0.9628654
#Petal.Width 0.8179411 -0.3661259 0.9628654 1.0000000
結果應與我們使用ppcor
庫獲得的結果相匹配:
#> ppcor::pcor(iris[,-5])$estimate
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 0.6285707 0.7190656 -0.3396174
#Sepal.Width 0.6285707 1.0000000 -0.6152919 0.3526260
#Petal.Length 0.7190656 -0.6152919 1.0000000 0.8707698
#Petal.Width -0.3396174 0.3526260 0.8707698 1.0000000
我們知道控制每個其他變量的成對部分相關可以通過O(n ^ 3)時間內的相關或協方差矩陣(見這里 )的反演來獲得。 所以一個可能的解決方案就是:
pcor.solve = function(x){
res = solve(cov(x))
res = -res/sqrt(diag(res) %o% diag(res))
diag(res) = 1
return(res)
}
這基本上是ppcor::pcor
的精簡版本。 結果是:
pcor.solve(iris[,-5])
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 0.6285707 0.7190656 -0.3396174
#Sepal.Width 0.6285707 1.0000000 -0.6152919 0.3526260
#Petal.Length 0.7190656 -0.6152919 1.0000000 0.8707698
#Petal.Width -0.3396174 0.3526260 0.8707698 1.0000000
但是請注意,協方差矩陣(或相關矩陣,結果相同)必須是正定的。
由於這主要歸結為有效的反轉操作,我在stats.SE中查看了這個線程。 qr.solve
和chol2inv
可以在協方差矩陣中用於相同的效果。
pcor.qr = function(x){
res = qr.solve(cov(x))
res = -res/sqrt(diag(res) %o% diag(res))
diag(res) = 1
dimnames(res)[[1]] = dimnames(res)[[2]] = colnames(x)
return(res)
}
pcor.qr(iris[,-5])
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 0.6285707 0.7190656 -0.3396174
#Sepal.Width 0.6285707 1.0000000 -0.6152919 0.3526260
#Petal.Length 0.7190656 -0.6152919 1.0000000 0.8707698
#Petal.Width -0.3396174 0.3526260 0.8707698 1.0000000
pcor.chol = function(x){
res = chol2inv(chol(cov(x)))
res = -res/sqrt(diag(res) %o% diag(res))
diag(res) = 1
dimnames(res)[[1]] = dimnames(res)[[2]] = colnames(x)
return(res)
}
pcor.chol(iris[,-5])
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 0.6285707 0.7190656 -0.3396174
#Sepal.Width 0.6285707 1.0000000 -0.6152919 0.3526260
#Petal.Length 0.7190656 -0.6152919 1.0000000 0.8707698
#Petal.Width -0.3396174 0.3526260 0.8707698 1.0000000
SVD也可以用來解決。 如果我們有正定方矩陣,它的SVD分解是A = UDU ^ T,它的逆是簡單的A ^ -1 = UD ^ -1U ^ T.
pcor.svd = function(x){
res = svd(cov(x))
res = res$v %*% diag(1/res$d) %*% t(res$v)
res = -res/sqrt(diag(res) %o% diag(res))
diag(res) = 1
dimnames(res)[[1]] = dimnames(res)[[2]] = colnames(x)
return(res)
}
pcor.svd(iris[,-5])
# Sepal.Length Sepal.Width Petal.Length Petal.Width
#Sepal.Length 1.0000000 0.6285707 0.7190656 -0.3396174
#Sepal.Width 0.6285707 1.0000000 -0.6152919 0.3526260
#Petal.Length 0.7190656 -0.6152919 1.0000000 0.8707698
#Petal.Width -0.3396174 0.3526260 0.8707698 1.0000000
具有10000次重復的微microbenchmark
:
library(microbenchmark)
#iris
dt1 = iris[,-5]
microbenchmark(
ppcor = ppcor::pcor(dt1)$estimate,
solve = pcor.solve(dt1),
qr = pcor.qr(dt1),
chol = pcor.chol(dt1),
svd = pcor.svd(dt1),
times = 10000L)
#Unit: microseconds
# expr min lq mean median uq max neval cld
# ppcor 247.728 267.790 314.8356 280.853 296.248 196962.601 10000 c
# solve 176.816 198.743 217.1298 205.274 221.603 2425.964 10000 b
# qr 240.264 258.459 282.7005 270.123 285.518 4015.438 10000 c
# chol 131.562 148.824 163.3567 154.423 167.019 1593.205 10000 a
# svd 179.615 199.675 219.2781 208.074 223.469 1920.710 10000 b
#random data
dt2 = cbind(rnorm(1E4), rnorm(1E4)+2)
microbenchmark(
ppcor = ppcor::pcor(dt2)$estimate,
solve = pcor.solve(dt2),
qr = pcor.qr(dt2),
chol = pcor.chol(dt2),
svd = pcor.svd(dt2),
times = 10000L)
#Unit: microseconds
# expr min lq mean median uq max neval cld
# ppcor 243.063 267.323 306.4535 284.585 311.177 1833.936 10000 d
# solve 180.548 190.812 222.6685 198.277 216.004 84776.704 10000 a
# qr 229.068 248.662 282.8142 262.658 285.518 1954.301 10000 c
# chol 179.148 189.413 212.6551 198.277 216.005 1383.733 10000 a
# svd 213.672 230.933 262.5084 243.529 264.058 5261.543 10000 b
#uncorrelated data
dt3 = cbind(sin(seq(0, 2*pi, length.out = 1000L)), cos(seq(0, 2*pi, length.out = 1000L)))
microbenchmark(
ppcor = ppcor::pcor(dt3)$estimate,
solve = pcor.solve(dt3),
qr = pcor.qr(dt3),
chol = pcor.chol(dt3),
svd = pcor.svd(dt3),
times = 10000L)
#Unit: microseconds
# expr min lq mean median uq max neval cld
# ppcor 142.759 162.354 188.7767 172.1500 191.745 2230.021 10000 d
# solve 80.711 89.108 102.8269 92.3740 101.704 1709.372 10000 a
# qr 130.629 145.092 168.0627 153.0220 169.351 4914.910 10000 c
# chol 79.777 87.709 102.2984 92.3740 101.238 6731.117 10000 a
# svd 112.901 127.363 147.1913 134.1285 148.358 1401.928 10000 b
[更新]或者,換句話說, chol
< solve
< svd
< qr
< ppcor
現在。 可能會獲得一些加速,因為協方差矩陣是對稱的( chol
解決方案已經使用了這個事實),並且在協方差計算中也可以獲得時間。
當然, ppcor
庫更通用,處理協方差矩陣不可逆等情況,因此在比較中處於劣勢。 人們也可以做到這一點,盡管當部分相關性將被窮盡地計算並且知道協方差矩陣是正定的時,期望具有更簡單的解決方案。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.