RcppEigen svd非常慢

时间:2013-05-03 17:56:33

标签: r inline eigen rcpp armadillo

升级到3.0后,RcppEigen的JacobiSVD变慢了吗? 我的图书馆使用RcppEigen现在正在快速工作。

> n<-1000
> m<-matrix(rnorm(n*n),n,n)

> unix.time(s1<-svd(m))       # R
   user  system elapsed 
 10.376   0.028  10.407 

> unix.time(s2<-svdArma(m))   # RcppArmadillo
   user  system elapsed 
 22.997   0.000  23.001 

> unix.time(s3<-svdEigen(m))  # RcppEigen
   user  system elapsed 
180.708   0.000 180.712 

这是R上的测试代码:

library(inline)

codeArma='
    arma::mat    m = Rcpp::as<arma::mat>(m_);

    arma::mat u;
    arma::vec s;
    arma::mat v;

    arma::svd(u,s,v,m); 
    return List::create( Rcpp::Named("u")=u,
                         Rcpp::Named("d")=s,
                         Rcpp::Named("v")=v );
'
svdArma <- cxxfunction(signature(m_="matrix"),codeArma, plugin="RcppArmadillo")

#-----------------------------------------------------------------------

codeEigen='
  const Eigen::Map<Eigen::MatrixXd> m (as<Eigen::Map<Eigen::MatrixXd> >(m_ ));

  Eigen::JacobiSVD <Eigen::MatrixXd>svd(m,
                   Eigen::ComputeThinU|Eigen::ComputeThinV);
  return List::create( Rcpp::Named("u")=svd.matrixU(),
                       Rcpp::Named("d")=svd.singularValues(),
                       Rcpp::Named("v")=svd.matrixV() );
'
svdEigen <- cxxfunction(signature(m_="matrix"), codeEigen, plugin="RcppEigen")

#------------------------------------------------------------------------
n<-1000
m<-matrix(rnorm(n*n),n,n)

system.time(s1<-svd(m))       # R
m1<-s1$u %*% diag(s1$d) %*% t(s1$v)
all.equal(m,m1)

system.time(s2<-svdArma(m))   # Armadillo
m2<-s2$u %*% diag(array(s2$d)) %*% t(s2$v)
all.equal(m,m2)

system.time(s3<-svdEigen(m))  # Eigen
m3<-s3$u %*% diag(s3$d) %*% t(s3$v)
all.equal(m,m3)

----------------------------------------------- -----------

1 个答案:

答案 0 :(得分:5)

切换到R 3.0.0不应该对RcppEigen这样的包的执行方式产生影响。如果你看到了性能的回归,那么可能会发生其他事情。

您也可以尝试使用Armadillo和/或Eigen直接在C ++中编译SVD(如果您将它们安装在R之外,和/或您可以从R包中使用一些修补程序获得标题)