[Rcpp-devel] RcppArmadillo -- crude benchmark of various mcmc vs rcppbugs
Dirk Eddelbuettel
edd at debian.org
Fri Apr 27 05:27:55 CEST 2012
On 26 April 2012 at 15:58, Whit Armstrong wrote:
| Slightly off topic, but Dirk kindly allowed me to make a quick
| benchmark post on rcppbugs.
"Allowed" ? Come on, benchmarks are always ok, as are posts on Rcpp. This is
after all the Rcpp list...
|
| Just a short post to the list showing some performance numbers (for
| running a simple linear regression) of rcppbugs (with an RcppArmadillo
| core) vs MCMCpack (Scythe core) vs JAGS (?? core).
|
| I was fairly optimistic about the speed of rcppbugs, yet this exceeds
| my expectations by so much that I fear there may be a bug in my
| benchmark code.
|
| I'm gearing up for my presentation at R/Finance, so I would be
| grateful to anyone who wants to kick the tires a bit. The core has
| been tested (we've been using the pure c++ implementation for over a
| year), but the R binding (via rcppbugs) is completely new.
|
| Thanks in advance. Code below (you'll need to install MCMCpack,
| rjags, and rcppbugs (now on CRAN) to run it).
|
| The quick summary is a 5x speedup over MCMCpack and a 77x speedup (I'm
| not sure I believe this yet) over JAGS.
A bit more modest here (two-year old i7 running Ubuntu amd64)
R> print(round(all.times.ratio,2))
time ratio
lm 0.09 0.08
rcppbugs 1.18 1.00
mcmcpack 4.05 3.45
jags 57.06 48.56
But still pretty good compared to MCMCpack. That is possibly legit -- modern
C++ versus a C++ design from a decade ago. Jags is a little harder to
believe. Maybe time to write an email to Martyn?
Thanks for sharing this. Looking forward to your talk next month.
Dirk
| Comments welcome.
|
| -Whit
|
|
| library(rcppbugs,quietly=TRUE,verbose=FALSE)
| library(rjags,quietly=TRUE,verbose=FALSE)
| library(MCMCpack,quietly=TRUE,verbose=FALSE)
|
| NR <- 1e2L
| NC <- 2L
|
| y <- rnorm(NR,1) + 10
| X <- matrix(nr=NR,nc=NC)
| X[,1] <- 1
| X[,2] <- y + rnorm(NR)/2 - 10
|
|
| lm.time <- system.time(lm.res <- lm.fit(X,y))
| ##print(coef(lm.res))
|
| b <- mcmc.normal(rnorm(NC),mu=0,tau=0.0001)
| tau.y <- mcmc.uniform(sd(as.vector(y)),lower=0,upper=100)
| ##y.hat <- deterministic(function(X,b) { X %*% b }, X, b)
| y.hat <- linear(X,b)
| y.lik <- mcmc.normal(y,mu=y.hat,tau=tau.y,observed=TRUE)
| m <- create.model(b, tau.y, y.hat, y.lik)
|
|
| iterations <- 1e5L
| burn <- iterations
| adapt <- 1e3L
| thin <- 10L
|
| cat("running rcppbugs model...\n")
| rcppbugs.time <- system.time(ans <- run.model(m,
| iterations=iterations, burn=burn, adapt=adapt, thin=thin))
| rcppbugs.coefs <- apply(ans[["b"]],2,mean)
| ##print(rcppbugs.coefs)
| print(rcppbugs.time)
|
| cat("running MCMCpack model...\n")
| mcmcpack.time <- system.time(mcmcpack.out <- MCMCregress(y ~
| X.2,data=data.frame(y=y,X=X),burnin = burn, mcmc = iterations,
| thin=thin))
| print(mcmcpack.time)
|
| cat("running jags model...\n")
|
| bug.model <- '
| model {
| for (i in 1:NR){
| y[i] ~ dnorm(y.hat[i], tau.y)
| y.hat[i] <- inprod(b, X[i,])
| }
| for (j in 1:NC){
| b[j] ~ dnorm(0, .0001)
| }
| tau.y ~ dunif(0, 100)
| }
| '
| bug.file <- tempfile(fileext=".bug")
| cat(bug.model,file=bug.file)
|
| jags.time1 <- system.time(
| jags <- jags.model(bug.file,
| data = list('X' = X,
| 'y' = y,
| 'NR' = NR,
| 'NC' = NC),
| n.chains = 1,
| n.adapt = adapt,
| quiet=TRUE
| ))
| jags.time2 <- system.time(update(jags, n.iter=burn, progress.bar="none"))
| jags.time3 <- system.time(jags.trace <-
| jags.samples(jags,c('b','tau.y'),n.iter=iterations,thin=thin,progress.bar="none"))
|
| jags.time <- jags.time1 + jags.time2 + jags.time3
| print(jags.time)
| jags.coefs <- apply(jags.trace$b,1,mean)
|
| ##print(jags.trace)
|
| cat("rcppbugs speedup:\n")
| all.times <- rbind(lm.time["elapsed"],rcppbugs.time["elapsed"],mcmcpack.time["elapsed"],jags.time["elapsed"])
| all.times.ratio <- cbind(all.times,all.times/rcppbugs.time["elapsed"])
| colnames(all.times.ratio) <- c("time","ratio")
| rownames(all.times.ratio) <- c("lm","rcppbugs","mcmcpack","jags")
| print(round(all.times.ratio,2))
|
| cat("coef comparison:\n")
| coef.compare <-
| rbind(coef(lm.res),rcppbugs.coefs,summary(mcmcpack.out)$statistics[,"Mean"][1:2],jags.coefs)
| rownames(coef.compare) <- c("lm","rcppbugs","mcmcpack","jags")
| print(coef.compare)
| _______________________________________________
| Rcpp-devel mailing list
| Rcpp-devel at lists.r-forge.r-project.org
| https://lists.r-forge.r-project.org/cgi-bin/mailman/listinfo/rcpp-devel
--
R/Finance 2012 Conference on May 11 and 12, 2012 at UIC in Chicago, IL
See agenda, registration details and more at http://www.RinFinance.com
More information about the Rcpp-devel
mailing list