fastpos

Travis build status Codecov test coverage CRAN status

The R package fastpos provides a fast algorithm to calculate the required sample size for a Pearson correlation to stabilize within a sequential framework (Schönbrodt & Perugini, 2013, 2018). Basically, one wants to find the sample size at which one can be sure that 1-α percent of many studies will fall into a specified corridor of stability around an assumed population correlation and stay inside that corridor if more participants are added to the study. For instance, find out how many participants per study are required so that, out of 100k studies, 90% would fall into the region between .4 to .6 (a Pearson correlation) and not leave this region again when more participants are added (under the assumption that the population correlation is .5). This sample size is also referred to as the critical point of stability for the specific parameters.

This approach is related to the AO-method of sample size planning (e.g. Algina & Olejnik, 2003) and as such can be seen as an alternative to power analysis. Unlike AO, the concept of stability incorporates the idea of sequentially adding participants to a study. Although the approach is young, it has already attracted a lot of interest in the psychological research community, which is evident in over 800 citations of the original publication (Schönbrodt & Perugini, 2013). Still, to date, there exists no easy way to use the stability approach for individual sample size planning because there is no analytical solution to the problem and a simulation approach is computationally expensive with \mathcal{O}(n^2). The presented package overcomes this limitation by speeding up the calculation of correlations and achieving \mathcal{O}(n). For typical parameters, the theoretical speedup should be at least around a factor of 250. An empirical benchmark for a typical scenario even shows a speedup of about 400, paving the way for a wider usage of the stability approach.

Installation

You can install the released version of fastpos from CRAN with:

install.packages("fastpos")

You can install the development version from GitHub with devtools:

devtools::install_github("johannes-titz/fastpos")

Using fastpos

Since you have found this page, I assume you either want to (1) calculate the critical point of stability for your own study or (2) explore the method in general. If this is the case, read on and you should find what you are looking for. Let us first load the package and set a seed for reproducibility:

library(fastpos)
set.seed(19950521)

In most cases you will just need the function find_critical_pos which will give you the critical point of stability for your specific parameters.

Let us reproduce Schönbrodt and Perugini’s quite famous and oft-cited table of the critical points of stability for a precision of 0.1. We reduce the number of studies to 10k so that it runs fairly quickly.

find_critical_pos(rho = seq(.1, .7, .1), sample_size_max = 1e3,
                  n_studies = 10e3)
#> Warning in find_critical_pos(rho = seq(0.1, 0.7, 0.1), sample_size_max = 1000, : 37 simulation[s] did not reach the corridor of
#>             stability.
#> Increase sample_size_max and rerun the simulation.
#>   rho_pop 80%   90%    95% sample_size_min sample_size_max lower_limit upper_limit n_studies n_not_breached precision
#> 1     0.1 253 361.0 479.05              20            1000         0.0         0.2     10000             14       0.1
#> 2     0.2 237 339.0 445.00              20            1000         0.1         0.3     10000             16       0.1
#> 3     0.3 212 304.1 402.00              20            1000         0.2         0.4     10000              5       0.1
#> 4     0.4 184 261.0 346.00              20            1000         0.3         0.5     10000              1       0.1
#> 5     0.5 142 205.1 273.00              20            1000         0.4         0.6     10000              0       0.1
#> 6     0.6 103 147.0 200.00              20            1000         0.5         0.7     10000              1       0.1
#> 7     0.7  64  96.0 127.05              20            1000         0.6         0.8     10000              0       0.1
#>   precision_rel
#> 1         FALSE
#> 2         FALSE
#> 3         FALSE
#> 4         FALSE
#> 5         FALSE
#> 6         FALSE
#> 7         FALSE

The results are very close to Schönbrodt and Perugini’s table (see https://github.com/nicebread/corEvol). Note that a warning is shown, because in some simulations the corridor of stability was not reached. As long as this number is low, this should not affect the estimates much. But if you want to get more accurate estimates, then increase the maximum sample size.

If you want to dig deeper, you can have a look at the functions that find_critical_pos builds upon. simulate_pos is the workhorse of the package. It calls a C++ function to calculate correlations sequentially and it does this pretty quickly (but you know that already, right?). A rawish approach would be to create a population with create_pop and pass it to simulate_pos:

pop <- create_pop(0.5, 1e6)
pos <- simulate_pos(x_pop = pop[,1],
                    y_pop = pop[,2],
                    n_studies = 1e4,
                    sample_size_min = 20,
                    sample_size_max = 1e3,
                    replace = T,
                    lower_limit = 0.4,
                    upper_limit = 0.6)
hist(pos, xlim = c(0, 1e3), xlab = c("Point of stability"),
     main = "Histogram of points of stability for rho = .5+-.1")

quantile(pos, c(.8, .9, .95), na.rm = T)
#> 80% 90% 95% 
#> 141 208 276

Note that no warning message appears if the corridor is not reached, but instead an NA value is returned. Pay careful attention if you work with this function, and adjust the maximum sample size as needed.

create_pop creates the population matrix by using a method described on SO (https://stats.stackexchange.com/questions/15011/generate-a-random-variable-with-a-defined-correlation-to-an-existing-variables/15040#15040). This is a much simpler way than Schönbrodt and Perugini’s approach, but the results do not seem to differ. If you are interested in how population parameters (e.g. skewness) affect the point of stability, you should instead refer to the population generating functions in Schönbrodt and Perugini’s work.

Parallelization

Since version 0.4.0 fastpos supports multiple cores. My first attempts to implement this were quite unsuccessful because of several reasons: (1) Higher-level parallelism in R makes it difficult to show progress in C++, which is where the important and time-demanding calculations happen (2) some parallelizing solutions do not work on all operating systems (e.g. mcpbapply) (3) overhead can be quite large, especially for a small number of simulation runs.

I thought the best solution is is to directly parallelize in C++. I tried to do it with RcppThread, but in the end this was even slower than singlethreading. I assume that a more experienced C++ programmer could make it work but to me parallelizing in C++ feels a bit like torture.

My final solution was quite simple and pragmatic: to use futures. I divide the number of studies by the available cores , then start simulations via futures in a multisession plan. Meanwhile the main R process also starts a simulation, wich shows a progress bar in C++. All simulations end at approximately the same time, the progress bar finishes and the futures resolve. The points of stability are combined and the rest of the program works as for the singlethreaded version.

Speed benefits are non-existent for a small numbers of studies, since fastpos is already too fast:

onecore <- function() {find_critical_pos(0.5)}
multicore <- function() {find_critical_pos(0.5, n_cores = future::availableCores())}
microbenchmark::microbenchmark(onecore(), multicore(), times = 10)
#> Unit: seconds
#>         expr      min       lq     mean   median       uq      max neval cld
#>    onecore() 1.742925 1.880799 2.010636 1.945818 2.133720 2.397736    10   a
#>  multicore() 1.700849 1.786334 2.058422 1.798521 1.838429 4.336336    10   a

When increasing the number of studies, the benefit becomes visible, but the difference is not gigantic:

onecore <- function() {find_critical_pos(0.5, n_studies = 1e5)}
multicore <- function() {find_critical_pos(0.5, n_studies = 1e5, 
                                           n_cores = future::availableCores())}
microbenchmark::microbenchmark(onecore(), multicore(),
                               times = 10)
#> Unit: seconds
#>         expr      min       lq     mean   median        uq       max neval cld
#>    onecore() 9.159248 9.346659 9.767050 9.604174 10.303442 10.417008    10   b
#>  multicore() 5.391069 5.988821 6.260544 6.511733  6.689804  6.732199    10  a

The test was done on my local computer with 4 cores.

How fast is fastpos?

In the introduction I boldly claimed that fastpos is much faster than the original implementation of Schönbrodt and Perugini (corEvol). The theoretical argument goes as follows:

corEvol calculates every correlation from scratch. If we take the sum formula for the correlation coefficient

r_{xy} = \frac{n\sum x_i y_i - \sum x_i \sum y_i} {\sqrt{n\sum x_i^2-(\sum x_i)^2} \sqrt{n\sum y_i2-(\sum y_i)2}}

we can see that several sums are calculated, each consisting of adding up (the sample size) terms. This has to be done for every sample size from the minimum to the maximum one. Thus, the total number of added terms for one sum is:

\sum _{n_\mathrm{min}}^{n_\mathrm{max}}n = \sum_{n=1}^{n_\mathrm{max}}n - \sum_{n=1}^{n_\mathrm{min}-1}n = n_\mathrm{max}(n_\mathrm{max}+1)/2 -(n_\mathrm{min}-1)(n_\mathrm{min}-1+1)/2

On the other hand, fastpos calculates the correlation for the maximum sample size first. This requires to add numbers for one sum. Then it subtracts one value from this sum to find the correlation for the sample size , which happens repeatedly until the minimum sample size is reached. Overall the total number of terms for one sum amounts to:

n_\mathrm{max}+n_\mathrm{max}-n_\mathrm{min}

The ratio between the two approaches is:

[\frac{n_\mathrm{max}(n_\mathrm{max}+1)/2 -(n_\mathrm{min}-1)n_\mathrm{min}/2}{2n_\mathrm{max}-n_\mathrm{min}}](https://latex.codecogs.com/png.latex?%5Cfrac%7Bn_%5Cmathrm%7Bmax%7D%28n_%5Cmathrm%7Bmax%7D%2B1%29%2F2%20-%28n_%5Cmathrm%7Bmin%7D-1%29n_%5Cmathrm%7Bmin%7D%2F2%7D%7B2n_%5Cmathrm%7Bmax%7D-n_%5Cmathrm%7Bmin%7D%7D%20 “\frac{n_\mathrm{max}(n_\mathrm{max}+1)/2 -(n_\mathrm{min}-1)n_\mathrm{min}/2}{2n_\mathrm{max}-n_\mathrm{min}}”)

For the typically used n_\mathrm{max} of 1000 and n_\mathrm{min} of 20, we can expect a speedup of about 250. This is only an approximation for several reasons. First, one can stop the process when the corridor is reached, which is done in fastpos but not in corEvol. Second, the main function of fastpos was written in C++ (via Rcpp), which is much faster than R. In a direct comparison between fastpos and corEvol we can expect fastpos to be at least 250 times faster.

The theoretical difference is so big that it should suffice to give a rough benchmark for which the following parameters were chosen: rho = .1, sample_size_max = 1000, sample_size_min = 20, n_studies = 10000.

Note that corEvol was written as a script for a simulation study and thus cannot be simply called via a function. Furthermore, a simulation run takes a lot of time and thus it is not practical to run it too many times. If you want to experiment with the benchmark, I have forked the original corEvol repository and made a benchmark branch (note that this will only work on GNU/Linux, since here I am using git through the bash):

git -C corEvol pull || git clone --single-branch --branch benchmark https://github.com/johannes-titz/corEvol

For corEvol, two files are “sourced” for the benchmark. The first file generates the simulations and the second is for calculating the critical point of stability. I turned off all messages produced by these source files.

library(microbenchmark)
setwd("corEvol")
corevol <- function(){
  source("01-simdata.R")
  source("02-analyse.R")
}
bm <- microbenchmark(corevol = corevol(),
                     fastpos = find_critical_pos(rho = .1,
                                                 sample_size_max = 1000,
                                                 n_studies = 10000),
                     times = 10, unit = "s")
bm
#> Unit: seconds
#>     expr        min         lq       mean     median         uq        max neval cld
#>  corevol 547.848236 726.241786 796.730368 871.701845 887.676597 903.194998    10   b
#>  fastpos   1.899546   4.159176   4.353587   4.268697   4.590466   6.157654    10  a

For the chosen parameters, fastpos is about 200 times faster than corEvol, for which there are two main reasons: (1) fastpos is built around a C++ function via Rcpp and (2) this function does not calculate every calculation from scratch, but only calculates the difference between the correlation at sample size and via the sum formula of the Pearson correlation (see above). There are some other factors that might play a role, but they cannot account for the large difference found. For instance, setting up a population takes quite long in corEvol (about 20s), but compared to the ~13min required overall, this is only a small fraction. There are other parts of the corEvol code that are fated to be slow, but again, a speedup by a factor of 200 cannot be achieved by improving these parts. The presented benchmark is definitely not comprehensive, but only demonstrates that fastpos can be used with no significant waiting time for a typical scenario, while for corEvol this is not the case.

FAQ

What does fastpos do if the corridor of stability is not reached for a simulation study?

In this case fastpos will return an NA value for the point of stability. When calculating the quantiles, fastpos will use the maximum sample size, which is a more reasonable estimate than ignoring the specific simulation study altogether.

Why does fastpos produce different estimates to corEvol?

If the same parameters are used, the differences are rather small. In general, differences cannot be avoided entirely due to the random nature of the whole process. Even if the same algorithm is used, the estimates will vary slightly from run to run. The other more important aspect is how studies are treated where the point of stability is not reached: corEvol ignores them, while fastpos assumes that the corridor was reached at the maximum sample size. Thus, if the parameters are the same, fastpos will tend to produce larger estimates, which is more accurate (and more conservative). But note that if the corridor of stability is not reached, then you should increase the maximum sample size. Previously, this was not feasible due to the computational demands, but with fastpos it usually can be done.

Issues and Support

If you find any bugs, please use the issue tracker at:

https://github.com/johannes-titz/fastpos/issues

If you need answers on how to use the package, drop me an e-mail at johannes at titz.science or johannes.titz at gmail.com

Contributing

Comments and feedback of any kind are very welcome! I will thoroughly consider every suggestion on how to improve the code, the documentation, and the presented examples. Even minor things, such as suggestions for better wording or improving grammar in any part of the package, are more than welcome.

If you want to make a pull request, please check that you can still build the package without any errors, warnings, or notes. Overall, simply stick to the R packages book: https://r-pkgs.org/ and follow the code style described here: http://r-pkgs.had.co.nz/r.html#style

References

Algina, J., & Olejnik, S. (2003). Sample size tables for correlation analysis with applications in partial correlation and multiple regression analysis. Multivariate Behavioral Research, 38, 309–323. https://doi.org/10.1207/S15327906MBR3803_02

Schönbrodt, F. D., & Perugini, M. (2013). At what sample size do correlations stabilize? Journal of Research in Personality, 47, 609–612. https://doi.org/10.1016/j.jrp.2013.05.009

Schönbrodt, F. D., & Perugini, M. (2018). Corrigendum to “At What Sample Size Do Correlations Stabilize?” [J. Res. Pers. 47 (2013) 609–612]. Journal of Research in Personality, 74, 194. https://doi.org/10.1016/j.jrp.2018.02.010