使用并行优化阵列R的循环

时间:2015-09-25 13:37:13

标签: arrays r performance loops parallel-processing

我有一个数组data = array [1:50,1:50,1:50]里面的值是介于-1,1之间的实数。

“数据”可以视为立方体50x50x50。

我需要根据这个等式创建一个相关矩阵(删除所有零)=>

value =(x + y) - | x-y |矩阵大小是可能组合的2倍(50x50x50)*((50x50x50)-1)/ 2 = 7.812.437.500这2倍=相关矩阵。

我这样做了:

假设我们有3x3x3:

arr = array(rnorm(10), dim=c(3,3,3))

data = data.frame(array(arr))


data$voxel <- rownames(data) 

#remove zeros
data<-data[!(data[,1]==0),]

rownames(data) = data$voxel

data$voxel = NULL


#######################################################################################
#Create cluster

no_cores <- detectCores() #- 1

clus <- makeCluster(no_cores)

clusterExport(clus, list("data") , envir=environment())

clusterEvalQ(clus,
             compare_strings <- function(j,i) {
               value <- (data[i,]+data[j,])-abs(data[i,]- data[j,])   
               pair <- rbind(rownames(data)[j],rownames(data)[i],value)
               return(pair)
             })

i = 0 # start 0
kk = 1
table <- data.frame()

ptm <- proc.time()

while(kk<nrow(data)) {

  out <-NULL  
  i = i+1 # fix row
  j = c((kk+1):nrow(data)) # rows to be compared

  #Apply the declared function  
  out = matrix(unlist(parRapply(clus,expand.grid(i,j), function(x,y) compare_strings(x[1],x[2]))),ncol=3, byrow = T)

  table <- rbind(table,out)

  kk = kk +1

}

proc.time() - ptm

结果是data.frame:

v1  v2  v3
1   2   2.70430114250358
1   3   0.199941717684129
... up to 351 rows

但这需要几天......

另外,我想为这种相关性创建一个矩阵:

   1                         2              3...
1  1                  2.70430114250358 
2  2.70430114250358          1
3...

有更快的方法吗?

由于

1 个答案:

答案 0 :(得分:0)

您的代码中存在许多性能错误:

  1. 当你应该依赖矢量化时循环。
  2. 你在一个循环中成长一个对象。
  3. 您可以并行化循环的每个迭代,而不是并行化外部循环。
  4. 如果避免第一个问题,可以避免所有这些问题。

    显然,您想要比较每个行组合。为此,您应首先获得行索引的所有组合:

    combs <- t(combn(1:27, 2))
    

    然后你可以将比较函数应用于这些:

    compare <- function(j,i, data) {
      as.vector((data[i,]+data[j,])-abs(data[i,]- data[j,]))
    }
    
    res <- data.frame(V1 = combs[,1], V2 = combs[,2], 
                      V3 = compare(combs[,1], combs[,2], data))
    

    现在,如果我们想检查这是否与您的代码产生相同的结果,我们首先需要修复您的输出。通过将字符(rownames)与数字组合在矩阵中,您将得到一个字符矩阵,最终data.frame的列都是字符。我们之后可以使用type.convert来解决这个问题(尽管从一开始就应该避免):

    table[] <- lapply(table, function(x) type.convert(as.character(x)))
    

    现在我们可以看到结果是一样的:

    all.equal(res, table)
    #[1] TRUE
    

    如果您愿意,可以将结果转换为稀疏矩阵:

    library(Matrix)
    m <- sparseMatrix(i = res$V1, j = res$V2, x = res$V3, 
                      dims = c(27, 27), symmetric = TRUE)
    diag(m) <- 1