Abstract
apeglm provides empirical Bayes shrinkage estimators for effect sizes for a variety of GLM models; apeglm stands for “Approximate Posterior Estimation for GLM”. apeglm package version: 1.12.0Note: the typical RNA-seq workflow for users would be to call apeglm estimation from within the lfcShrink
function from the DESeq2 package. The unevaluated code chunk shows how to obtain apeglm shrinkage estimates after running DESeq
. See the DESeq2 vignette for more details. The lfcShrink
wrapper function takes care of many details below, and unifies the interface for multiple shrinkage estimators. The coefficient to shrink can be specified either by name or by number (following the order in resultsNames(dds)
). Be aware that DESeq2’s lfcShrink
interface provides LFCs on the log2 scale, while apeglm provides coefficients on the natural log scale.
Joshua Zitovsky contributed fast C++ code for the beta-binomial likelihood, demonstrated in a later section of the vignette.
We have benefited in the development of apeglm
from feedback or contributions from the following individuals:
Wolfgang Huber, Cecile Le Sueur, Charlotte Soneson
Here we show example code which mimics what will happen inside the lfcShrink
function when using the apeglm method (Zhu, Ibrahim, and Love 2018).
Load a prepared SummarizedExperiment
:
## SRR1039508 SRR1039509 SRR1039512 SRR1039513 SRR1039516
## ENSG00000000003 679 448 873 408 1138
## ENSG00000000005 0 0 0 0 0
## ENSG00000000419 467 515 621 365 587
## ENSG00000000457 260 211 263 164 245
## ENSG00000000460 60 55 40 35 78
## ENSG00000000938 0 0 2 0 1
## SRR1039517 SRR1039520 SRR1039521
## ENSG00000000003 1047 770 572
## ENSG00000000005 0 0 0
## ENSG00000000419 799 417 508
## ENSG00000000457 331 233 229
## ENSG00000000460 63 76 60
## ENSG00000000938 0 0 0
For demonstration, we will use 5000 genes of the airway
dataset, the first from those genes with at least 10 counts across all samples.
First run a DESeq2 differential expression analysis (Love, Huber, and Anders 2014) (size factors, and dispersion estimates could similarly be estimated using edgeR):
library(DESeq2)
dds <- DESeqDataSet(airway, ~cell + dex)
dds$dex <- relevel(dds$dex, "untrt")
dds <- DESeq(dds)
res <- results(dds)
Defining data and parameter objects necessary for apeglm
. We must multiply the coefficients from DESeq2 by a factor, because apeglm provides natural log coefficients. Again, this would be handled inside of lfcShrink
in DESeq2 for a typical RNA-seq analysis.
Here apeglm
on 5000 genes takes less than a minute on a laptop. It scales with number of genes, the number of samples and the number of variables in the design formula, where here we have 5 coefficients (one for the four cell cultures and one for the difference due to dexamethasone treatment).
We provide apeglm
with the SummarizedExperiment although the function can also run on a matrix of counts or other observed data. We specify a coef
as well as a threshold
which we discuss below. Note that we multiple the threshold
by log(2)
to convert from log2 scale to natural log scale.
library(apeglm)
system.time({
fit <- apeglm(Y=airway, x=x, log.lik=logLikNB, param=param, coef=ncol(x),
threshold=log(2) * 1, mle=mle, offset=offset)
})
## user system elapsed
## 32.648 0.068 32.819
## List of 7
## $ no.shrink : int [1:4] 1 2 3 4
## $ prior.mean : num 0
## $ prior.scale : num 0.299
## $ prior.df : num 1
## $ prior.no.shrink.mean : num 0
## $ prior.no.shrink.scale: num 15
## $ prior.var : num 0.0894
There are faster implementations of apeglm specifically for negative binomial likelihoods. The version nbinomR
is ~5 times faster than the default general
.
system.time({
fitR <- apeglm(Y=assay(airway), x=x, log.lik=NULL, param=param, coef=ncol(x),
threshold=log(2) * 1, mle=mle, offset=offset, method="nbinomR")
})
## user system elapsed
## 6.052 0.004 6.063
The version nbinomCR
is ~10 times faster than the default general
.
system.time({
fitCR <- apeglm(Y=assay(airway), x=x, log.lik=NULL, param=param, coef=ncol(x),
threshold=log(2) * 1, mle=mle, offset=offset, method="nbinomCR")
})
## user system elapsed
## 3.260 0.000 3.264
The version nbinomC
returns only the MAP coefficients and can be ~50-100 times faster than the default general
. The MAP coefficients are the same as returned by nbinomCR
above, we just skip the calculation of posterior SD. A variant of nbinomC
is nbinomC*
which includes random starts.
system.time({
fitC <- apeglm(Y=assay(airway), x=x, log.lik=NULL, param=param, coef=ncol(x),
threshold=log(2) * 1, mle=mle, offset=offset, method="nbinomC")
})
## user system elapsed
## 0.188 0.000 0.188
Among other output, we have the estimated coefficients attached to the ranges of the SummarizedExperiment used as input:
## [1] "CompressedGRangesList"
## attr(,"package")
## [1] "GenomicRanges"
## DataFrame with 5000 rows and 5 columns
## X.Intercept. cellN061011 cellN080611 cellN61311 dextrt
## <numeric> <numeric> <numeric> <numeric> <numeric>
## ENSG00000000003 6.64566 0.0607881 0.2296723 -0.1520394 -0.26470148
## ENSG00000000419 6.23452 -0.0849991 -0.0151415 -0.0476622 0.11513214
## ENSG00000000457 5.45295 0.0601670 -0.0556310 0.0484623 0.00907001
## ENSG00000000460 3.77048 0.5452039 0.2613561 0.3494298 -0.04252381
## ENSG00000000938 -0.31469 -4.1774898 -0.5396787 -4.2008464 -0.00836574
## ... ... ... ... ... ...
## ENSG00000122756 -3.62985 4.0391091 -2.0059134 2.9579939 -0.000701743
## ENSG00000122778 5.23906 0.0800614 0.7857182 0.2400986 -0.285457154
## ENSG00000122779 5.66861 -0.0569747 0.0501293 -0.0441315 -0.109945277
## ENSG00000122783 6.01535 0.0368918 -0.1578283 -0.2973999 0.069012038
## ENSG00000122786 9.78251 -0.1823493 0.0942307 -0.1202800 0.523514860
We can compare the coefficients from apeglm with the "normal"
shrinkage type from the original DESeq2 paper (2014). This method, which makes use of a Normal-based prior, is no longer the default shrinkage estimator for lfcShrink
. apeglm provides coefficients on the natural log scale, so we must convert to log2 scale by multiplying by log2(exp(1))
. Note that DESeq2’s lfcShrink
function converts apeglm coefficients to the log2 scale internally.
## user system elapsed
## 2.892 0.000 2.898
Here we plot apeglm estimators against DESeq2:
Here we plot MLE, DESeq2 and apeglm estimators against the mean of normalized counts:
par(mfrow=c(1,3))
lims <- c(-8,8)
hline <- function() abline(h=c(-4:4 * 2),col=rgb(0,0,0,.2))
xlab <- "mean of normalized counts"
plot(res$baseMean, res$log2FoldChange, log="x",
ylim=lims, main="MLE", xlab=xlab)
hline()
plot(res$baseMean, DESeq2.lfc, log="x",
ylim=lims, main="DESeq2", xlab=xlab)
hline()
plot(res$baseMean, apeglm.lfc, log="x",
ylim=lims, main="apeglm", xlab=xlab)
hline()
Note that p-values and FSR define different events, and are not on the same scale. An FSR of 0.5 means that the estimated sign is as bad as random guess.
par(mfrow=c(1,2),mar=c(5,5,1,1))
plot(res$pvalue, fit$fsr, col="blue",
xlab="DESeq2 pvalue", ylab="apeglm local FSR",
xlim=c(0,1), ylim=c(0,.5))
abline(0,1)
plot(-log10(res$pvalue), -log10(fit$fsr),
xlab="-log10 DESeq2 pvalue", ylab="-log10 apeglm local FSR",
col="blue")
abline(0,1)
The s-value was proposed by Stephens (2016), as a statistic giving the aggregate false sign rate for tests with equal or lower s-value than the one considered. We recommend using a lower threshold on s-values than typically used for adjusted p-values, for example one might be interested in sets with 0.01 or 0.005 aggregate FSR.
plot(res$padj, fit$svalue, col="blue",
xlab="DESeq2 padj", ylab="apeglm svalue",
xlim=c(0,.2), ylim=c(0,.02))
More scrutiny can be applied by using an LFC threshold greater than zero, and asking for the probability of a “false-sign-or-small” (FSOS) event: that the effect size is not further from zero in distance than the threshold amount. We can run the svalue
function on these per-gene probabilities to produce s-values that bound the FSOS rate for sets of genes. By specifying threshold=log(2) * 1
above, apeglm
will then output a vector thresh
in the results list that gives the per-gene probabilities of false-sign-or-small events.
We have created a separate GitHub repository giving an example of how the apeglm estimator can be used for Zero-Inflated Negative Binomial data. This approach uses the zinbwave method and Bioconductor package to estimate the probability of each zero belonging to the excess zero component. We compare using a Negative Binomial likelihood with the excess zeros down-weighted and using a Zero-Inflated Negative Binomial likelihood. These two approaches with apeglm perform similarly but we note that the first approach involves less additional code and is faster to compute.
We also show an short example using an alternative likelihood to the negative binomial. Suppose we have allele-specific counts for n=20 vs 20 samples across 5000 genes. We can define a binomial model and test for the allelic balance across groups of samples.
Here we will simulate allele counts from our existing dataset for demonstration. We spike in 10 genes with strong allelic imbalance (instead of an allelic ratio close to 0.5, these will have a ratio of 0.75).
library(emdbook)
n <- 20
f <- factor(rep(1:2,each=n))
mu <- ifelse(res$baseMean > 50, res$baseMean, 50)
set.seed(1)
cts <- matrix(rnbinom(nrow(dds)*2*n,
mu=mu,
size=1/dispersions(dds)),
ncol=2*n)
theta <- runif(nrow(cts),1,1000)
prob <- rnorm(nrow(cts),.5,.05) # close to 0.5
ase.cts <- matrix(rbetabinom(prod(dim(cts)), prob=prob,
size=cts, theta=rep(theta,ncol(cts))),
nrow=nrow(cts))
idx <- 1:10
idx2 <- which(f == 2)
theta[idx] <- 1000
prob[idx] <- 0.75
# the spiked in genes have an allelic ratio of 0.75
ase.cts[idx,idx2] <- matrix(rbetabinom(length(idx)*length(idx2), prob=prob[idx],
size=cts[idx,idx2], theta=theta[idx]),
nrow=length(idx))
We first need to estimate MLE coefficients and standard errors.
One option to run apeglm
would be to define a beta-binomial likelihood function which uses the total counts as a parameter, and the logit function as a link. And then this function could be provided to the log.lik
argument.
betabinom.log.lik <- function(y, x, beta, param, offset) {
xbeta <- x %*% beta
p.hat <- (1+exp(-xbeta))^-1
dbetabinom(y, prob=p.hat, size=param[-1], theta=param[1], log=TRUE)
}
However, apeglm
has faster C++ implementations for the beta-binomial, which were implemented and tested by Joshua Zitovsky. Here, using method="betabinCR"
is 3 times faster than using the R-defined log.lik
, and the "betabinCR"
method also scales significantly better with more samples and more coefficients. As with the negative binomial, method="betabinC"
can be used if the standard errors are not needed, and this will be 5 times faster than the R-defined log.lik
approach on this dataset.
The following code performs two iterations of estimating the MLE coefficients, and then estimating the beta-binomial dispersion.
theta.hat <- 100 # rough initial estimate of dispersion
x <- model.matrix(~f)
niter <- 3
system.time({
for (i in 1:niter) {
param <- cbind(theta.hat, cts)
fit.mle <- apeglm(Y=ase.cts, x=x, log.lik=NULL, param=param,
no.shrink=TRUE, log.link=FALSE, method="betabinCR")
theta.hat <- bbEstDisp(success=ase.cts, size=cts,
x=x, beta=fit.mle$map,
minDisp=.01, maxDisp=5000)
}
})
## user system elapsed
## 24.248 0.020 24.317
We can then plot the MLE estimates over the mean:
coef <- 2
xlab <- "mean of normalized counts"
plot(res$baseMean, fit.mle$map[,coef], log="x", xlab=xlab, ylab="log odds")
points(res$baseMean[idx], fit.mle$map[idx,coef], col="dodgerblue", cex=3)
Now we run the posterior estimation, including a prior on the second coefficient:
mle <- cbind(fit.mle$map[,coef], fit.mle$sd[,coef])
param <- cbind(theta.hat, cts)
system.time({
fit2 <- apeglm(Y=ase.cts, x=x, log.lik=NULL, param=param,
coef=coef, mle=mle, threshold=0.5,
log.link=FALSE, method="betabinCR")
})
## user system elapsed
## 5.516 0.008 5.534
In the apeglm
plot, we color in red the genes with a low aggregate probability of false-sign-or-small (FSOS) events (s-value < .01), where we’ve again defined “small” on the log odds scale using the threshold
argument above.
par(mfrow=c(1,2))
ylim <- c(-1,1.5)
s.val <- svalue(fit2$thresh) # small-or-false-sign value
plot(res$baseMean, fit.mle$map[,coef], main="MLE",
log="x", xlab=xlab, ylab="log odds", ylim=ylim)
points(res$baseMean[idx], fit.mle$map[idx,coef], col="dodgerblue", cex=3)
abline(h=0,col=rgb(1,0,0,.5))
cols <- ifelse(s.val < .01, "red", "black")
plot(res$baseMean, fit2$map[,coef], main="apeglm",
log="x", xlab=xlab, ylab="log odds", col=cols, ylim=ylim)
points(res$baseMean[idx], fit2$map[idx,coef], col="dodgerblue", cex=3)
abline(h=0,col=rgb(1,0,0,.5))
## [1] 1.098612
##
## TRUE
## 10
## R version 4.0.3 (2020-10-10)
## Platform: x86_64-pc-linux-gnu (64-bit)
## Running under: Ubuntu 18.04.5 LTS
##
## Matrix products: default
## BLAS: /home/biocbuild/bbs-3.12-bioc/R/lib/libRblas.so
## LAPACK: /home/biocbuild/bbs-3.12-bioc/R/lib/libRlapack.so
##
## locale:
## [1] LC_CTYPE=en_US.UTF-8 LC_NUMERIC=C
## [3] LC_TIME=en_US.UTF-8 LC_COLLATE=C
## [5] LC_MONETARY=en_US.UTF-8 LC_MESSAGES=en_US.UTF-8
## [7] LC_PAPER=en_US.UTF-8 LC_NAME=C
## [9] LC_ADDRESS=C LC_TELEPHONE=C
## [11] LC_MEASUREMENT=en_US.UTF-8 LC_IDENTIFICATION=C
##
## attached base packages:
## [1] parallel stats4 stats graphics grDevices utils datasets
## [8] methods base
##
## other attached packages:
## [1] emdbook_1.3.12 apeglm_1.12.0
## [3] DESeq2_1.30.0 airway_1.9.0
## [5] SummarizedExperiment_1.20.0 Biobase_2.50.0
## [7] GenomicRanges_1.42.0 GenomeInfoDb_1.26.0
## [9] IRanges_2.24.0 S4Vectors_0.28.0
## [11] BiocGenerics_0.36.0 MatrixGenerics_1.2.0
## [13] matrixStats_0.57.0
##
## loaded via a namespace (and not attached):
## [1] bdsmatrix_1.3-4 Rcpp_1.0.5 locfit_1.5-9.4
## [4] mvtnorm_1.1-1 lattice_0.20-41 digest_0.6.27
## [7] plyr_1.8.6 R6_2.4.1 coda_0.19-4
## [10] RSQLite_2.2.1 evaluate_0.14 httr_1.4.2
## [13] ggplot2_3.3.2 pillar_1.4.6 zlibbioc_1.36.0
## [16] rlang_0.4.8 annotate_1.68.0 blob_1.2.1
## [19] Matrix_1.2-18 bbmle_1.0.23.1 rmarkdown_2.5
## [22] splines_4.0.3 BiocParallel_1.24.0 geneplotter_1.68.0
## [25] stringr_1.4.0 RCurl_1.98-1.2 bit_4.0.4
## [28] munsell_0.5.0 DelayedArray_0.16.0 numDeriv_2016.8-1.1
## [31] compiler_4.0.3 xfun_0.18 pkgconfig_2.0.3
## [34] htmltools_0.5.0 tidyselect_1.1.0 tibble_3.0.4
## [37] GenomeInfoDbData_1.2.4 XML_3.99-0.5 crayon_1.3.4
## [40] dplyr_1.0.2 MASS_7.3-53 bitops_1.0-6
## [43] grid_4.0.3 xtable_1.8-4 gtable_0.3.0
## [46] lifecycle_0.2.0 DBI_1.1.0 magrittr_1.5
## [49] scales_1.1.1 stringi_1.5.3 XVector_0.30.0
## [52] genefilter_1.72.0 ellipsis_0.3.1 vctrs_0.3.4
## [55] generics_0.0.2 RColorBrewer_1.1-2 tools_4.0.3
## [58] bit64_4.0.5 glue_1.4.2 purrr_0.3.4
## [61] survival_3.2-7 yaml_2.2.1 AnnotationDbi_1.52.0
## [64] colorspace_1.4-1 memoise_1.1.0 knitr_1.30
Love, Michael I., Wolfgang Huber, and Simon Anders. 2014. “Moderated estimation of fold change and dispersion for RNA-seq data with DESeq2.” Genome Biology 15 (12):550. https://doi.org/10.1186/s13059-014-0550-8.
Stephens, Matthew. 2016. “False Discovery Rates: A New Deal.” Biostatistics 18 (2). https://doi.org/10.1093/biostatistics/kxw041.
Zhu, Anqi, Joseph G. Ibrahim, and Michael I. Love. 2018. “Heavy-Tailed Prior Distributions for Sequence Count Data: Removing the Noise and Preserving Large Differences.” Bioinformatics. https://doi.org/10.1093/bioinformatics/bty895.