Getting Faster Results

Warning

  1. The solution described below is useful when you mathematically know a problem is DCP-compliant and none of your data inputs will change the nature of the problem. We recommend that users check the DCP-compliance of a problem (via a call to is_dcp(prob) for example) at least once to ensure this is the case. Not verifying DCP-compliance may result in garbage!

  2. Note also that the large speed gains in previous versions are no longer evident in the version 1.x because the new reduction framework has really made CVXR faster.

Introduction

As was remarked in the introduction to CVXR, its chief advantage is flexibility: you can specify a problem in close to mathematical form and CVXR solves it for you, if it can. Behind the scenes, CVXR compiles the domain specific language and verifies the convexity of the problem before sending it off to solvers. If the problem violates the rules of Disciplined Convex Programming it is rejected.

Therefore, it is generally slower than tailor-made solutions to a given problem.

An Example

To understand the speed issues, let us consider the global warming data from the Carbon Dioxide Information Analysis Center (CDIAC) again. The data points are the annual temperature anomalies relative to the 1961–1990 mean. We will fit the nearly-isotonic approximation \(\beta \in {\mathbf R}^m\) by solving

\[ \begin{array}{ll} \underset{\beta}{\mbox{Minimize}} & \frac{1}{2}\sum_{i=1}^m (y_i - \beta_i)^2 + \lambda \sum_{i=1}^{m-1}(\beta_i - \beta_{i+1})_+, \end{array} \] where \(\lambda \geq 0\) is a penalty parameter and \(x_+ =\max(x,0)\).

This can be solved as follows.

suppressMessages(suppressWarnings(library(CVXR)))
data(cdiac)
y <- cdiac$annual
m <- length(y)
lambda <- 0.44
beta <- Variable(m)
obj <- 0.5 * sum((y - beta)^2) + lambda * sum(pos(diff(beta)))
prob <- Problem(Minimize(obj))
soln <- solve(prob, solver = "ECOS")
betaHat <- soln$getValue(beta)

This is the recommended way to solve a problem.

However, suppose we wished to construct bootstrap confidence intervals for the estimate using 100 resamples. It is clear that this computation time can quickly become limiting .

Below, we show how one can get at the problem data and directly call a solver to get faster results.

Profile the code

Profiling a single fit to the model is useful to figure out where most of the time is spent.

library(profvis)
y <- cdiac$annual
profvis({
    beta <- Variable(m)
    obj <- Minimize(0.5 * sum((y - beta)^2) + lambda * sum(pos(diff(beta))))
    prob <- Problem(obj)
    soln <- solve(prob, solver = "ECOS")
    betaHat <- soln$getValue(beta)
})

It is especially instructive to click on the data tab and open up the tree for solve to see the sequence of calls and cumulative time used.

The profile shows that most of the total time (2400ms for one of our runs) time is spent in the call to is_dcp generic (about 2000ms). This generic is responsible to ensuring that all the problem is DCP-compliant by checking the nature of each of the components that make up the problem. The actual solving took a much smaller fraction of the time.

Directly Calling the Solver

We are mathematically certain that the above is convex and so we can avoid the is_dcp hit. We can obtain the the problem data for a particular solver (like OSQP, or ECOS or SCS) using the function get_problem_data and directly hand that data to the solver to get the solution.

prob_data <- get_problem_data(prob, solver = "ECOS")

ASIDE: How did we know ECOS was the solver to use? Future versions will provide a function to match a solver to a problem. (Actually, it is available already, but not exported yet!). For now, a single call to solve with the verbose option set to TRUE can provide that information.

soln <- solve(prob, verbose = TRUE)

Now that we have the problem data and know which solver to use, we can call the ECOS solver with the right arguments. (The ECOS solver is provided by the package ECOSolveR, which CVXR imports.)

if (packageVersion("CVXR") > "0.99-7") {
    ECOS_dims <- ECOS.dims_to_solver_dict(prob_data$data[["dims"]])
} else {
    ECOS_dims <- prob_data$data[["dims"]]
}
solver_output <- ECOSolveR::ECOS_csolve(c = prob_data$data[["c"]],
                                        G = prob_data$data[["G"]],
                                        h = prob_data$data[["h"]],
                                        dims = ECOS_dims,
                                        A = prob_data$data[["A"]],
                                        b = prob_data$data[["b"]])

Finally, we can obtain the results by asking CVXR to unpack the solver results for us. (See ?unpack_results for further examples.)

if (packageVersion("CVXR") > "0.99-7") {
    direct_soln <- unpack_results(prob, solver_output, prob_data$chain, prob_data$inverse_data)
} else {
    direct_soln <- unpack_results(prob, "ECOS", solver_output)
}

Profile the Direct Call

We can profile this direct call now.

profvis({
    beta <- Variable(m)
    obj <- Minimize(0.5 * sum((y - beta)^2) + lambda * sum(pos(diff(beta))))
    prob <- Problem(obj)
    prob_data <- get_problem_data(prob, solver = "ECOS")
    if (packageVersion("CVXR") > "0.99-7") {
        ECOS_dims <- ECOS.dims_to_solver_dict(prob_data$data[["dims"]])
    } else {
        ECOS_dims <- prob_data$data[["dims"]]
    }
    solver_output <- ECOSolveR::ECOS_csolve(c = prob_data$data[["c"]],
                                            G = prob_data$data[["G"]],
                                            h = prob_data$data[["h"]],
                                            dims = ECOS_dims,
                                            A = prob_data$data[["A"]],
                                            b = prob_data$data[["b"]])
    if (packageVersion("CVXR") > "0.99-7") {
        direct_soln <- unpack_results(prob, solver_output, prob_data$chain, prob_data$inverse_data)
    } else {
        direct_soln <- unpack_results(prob, "ECOS", solver_output)
    }
})

For one of our runs, the total time went down from \(2400ms\) to \(690ms\), more than a 3-fold speedup! In cases where the objective function and constraints are more complex, the speedup can be more than 10-fold.

Same Answer?

Of course, we should also verify that the results obtained in both cases are same.

identical(betaHat, direct_soln$getValue(beta))
## [1] TRUE

Session Info

sessionInfo()
## R version 4.2.1 (2022-06-23)
## Platform: x86_64-apple-darwin21.6.0 (64-bit)
## Running under: macOS Ventura 13.0
## 
## Matrix products: default
## BLAS:   /usr/local/Cellar/openblas/0.3.21/lib/libopenblasp-r0.3.21.dylib
## LAPACK: /usr/local/Cellar/r/4.2.1_4/lib/R/lib/libRlapack.dylib
## 
## locale:
## [1] en_US.UTF-8/en_US.UTF-8/en_US.UTF-8/C/en_US.UTF-8/en_US.UTF-8
## 
## attached base packages:
## [1] stats     graphics  grDevices datasets  utils     methods   base     
## 
## other attached packages:
## [1] profvis_0.3.7 CVXR_1.0-11  
## 
## loaded via a namespace (and not attached):
##  [1] gmp_0.6-6         Rcpp_1.0.9        bslib_0.4.0       compiler_4.2.1   
##  [5] jquerylib_0.1.4   tools_4.2.1       digest_0.6.30     bit_4.0.4        
##  [9] jsonlite_1.8.3    evaluate_0.17     lattice_0.20-45   rlang_1.0.6      
## [13] Matrix_1.5-1      gurobi_9.5-2      cli_3.4.1         Rglpk_0.6-4      
## [17] yaml_2.3.6        blogdown_1.13     xfun_0.34         cccp_0.2-9       
## [21] fastmap_1.1.0     ECOSolveR_0.5.4   Rmpfr_0.8-9       stringr_1.4.1    
## [25] knitr_1.40        htmlwidgets_1.5.4 sass_0.4.2        bit64_4.0.5      
## [29] grid_4.2.1        R6_2.5.1          rmarkdown_2.17    bookdown_0.29    
## [33] magrittr_2.0.3    rcbc_0.1.0.9001   codetools_0.2-18  htmltools_0.5.3  
## [37] assertthat_0.2.1  Rcplex_0.3-5      stringi_1.7.8     Rmosek_10.0.25   
## [41] cachem_1.0.6      slam_0.1-50

Source

R Markdown

References