Introduction

Simple Least Squares
Author

Anqi Fu and Balasubramanian Narasimhan

Consider a simple linear regression problem where it is desired to estimate a set of parameters using a least squares criterion.

We generate some synthetic data where we know the model completely, that is

Y=Xβ+ϵ

where Y is a 100×1 vector, X is a 100×10 matrix, β=[4,,1,0,1,,5] is a 10×1 vector, and ϵN(0,1).

set.seed(123)

n <- 100
p <- 10
beta <- -4:5   # beta is just -4 through 5.

X <- matrix(rnorm(n * p), nrow=n)
colnames(X) <- paste0("beta_", beta)
Y <- X %*% beta + rnorm(n)

Given the data X and Y, we can estimate the β vector using lm function in R that fits a standard regression model.

ls.model <- lm(Y ~ 0 + X)   # There is no intercept in our model above
m <- matrix(coef(ls.model), ncol = 1)
rownames(m) <- paste0("\\(\\beta_{", 1:p, "}\\)")
library(kableExtra)
knitr::kable(m, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:2, background = "#ececec")
β1 -3.9196886
β2 -3.0117048
β3 -2.1248242
β4 -0.8666048
β5 0.0914658
β6 0.9490454
β7 2.0764700
β8 3.1272275
β9 3.9609565
β10 5.1348845

These are the least-squares estimates and can be seen to be reasonably close to the original β values -4 through 5.

The CVXR formulation

The CVXR formulation states the above as an optimization problem:

minimizeβyXβ22, which directly translates into a problem that CVXR can solve as shown in the steps below.

  • Step 0. Load the CVXR library
suppressWarnings(library(CVXR, warn.conflicts=FALSE))
  • Step 1. Define the variable to be estimated
betaHat <- Variable(p)
  • Step 2. Define the objective to be optimized
objective <- Minimize(sum((Y - X %*% betaHat)^2))

Notice how the objective is specified using functions such as sum, *%* and ^, that are familiar to R users despite that fact that betaHat is no ordinary R expression but a CVXR expression.

  • Step 3. Create a problem to solve
problem <- Problem(objective)
  • Step 4. Solve it!
## psolve() returns the optimal value directly; value() extracts variable values
## (The deprecated solve()$getValue() pattern still works but warns.)
result <- psolve(problem)
check_solver_status(problem)
  • Step 5. Extract solution and objective value
Objective value: 97.847586

We can indeed satisfy ourselves that the results we get matches that from lm.

m <- cbind(value(betaHat), coef(ls.model))
colnames(m) <- c("CVXR est.", "lm est.")
rownames(m) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(m, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:3, background = "#ececec")
CVXR est. lm est.
β1 -3.9196886 -3.9196886
β2 -3.0117048 -3.0117048
β3 -2.1248242 -2.1248242
β4 -0.8666048 -0.8666048
β5 0.0914658 0.0914658
β6 0.9490454 0.9490454
β7 2.0764700 2.0764700
β8 3.1272275 3.1272275
β9 3.9609565 3.9609565
β10 5.1348845 5.1348845

Wait a minute! What have we gained?

On the surface, it appears that we have replaced one call to lm with at least five or six lines of new R code. On top of that, the code actually runs slower, and so it is not clear what was really achieved.

So suppose we knew that the βs were nonnegative and we wish to take this fact into account. This is nonnegative least squares regression and lm would no longer do the job.

In CVXR, the modified problem merely requires the addition of a constraint to the problem definition.

problem <- Problem(objective, constraints = list(betaHat >= 0))
result <- psolve(problem)
check_solver_status(problem)
m <- matrix(value(betaHat), ncol = 1)
rownames(m) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(m, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:2, background = "#ececec")
β1 0.0000000
β2 0.0000000
β3 0.0000000
β4 0.0000000
β5 1.2374488
β6 0.6234665
β7 2.1230663
β8 2.8035640
β9 4.4448016
β10 5.2073521

We can verify once again that these values are comparable to those obtained from another R package, say nnls.

library(nnls)
nnls.fit <- nnls(X, Y)$x
m <- cbind(value(betaHat), nnls.fit)
colnames(m) <- c("CVXR est.", "nnls est.")
rownames(m) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(m, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:3, background = "#ececec")
CVXR est. nnls est.
β1 0.0000000 0.0000000
β2 0.0000000 0.0000000
β3 0.0000000 0.0000000
β4 0.0000000 0.0000000
β5 1.2374488 1.2374488
β6 0.6234665 0.6234665
β7 2.1230663 2.1230663
β8 2.8035640 2.8035640
β9 4.4448016 4.4448016
β10 5.2073521 5.2073521

Okay that was cool, but…

As you no doubt noticed, we have done nothing that other R packages could not do.

So now suppose further, for some extraneous reason, that the sum of β2 and β3 is known to be negative and but all other βs are positive.

It is clear that this problem would not fit into any standard package. But in CVXR, this is easily done by adding a few constraints.

To express the fact that β2+β3 is negative, we construct a row matrix with zeros everywhere, except in positions 2 and 3 (for β2 and β3 respectively).

A <- matrix(c(0, 1, 1, rep(0, 7)), nrow = 1)
colnames(A) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(A, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:10, background = "#ececec")
β1 β2 β3 β4 β5 β6 β7 β8 β9 β10
0 1 1 0 0 0 0 0 0 0

The sum constraint is nothing but Aβ<=0

which we express in R as

constraint1 <- A %*% betaHat <= 0

NOTE: The above constraint can also be expressed simply as

constraint1 <- betaHat[2] + betaHat[3] <= 0

but it is easier working with matrices in general with CVXR.

For the positivity for rest of the variables, we construct a 10×10 matrix A to have 1’s along the diagonal everywhere except rows 2 and 3 and zeros everywhere.

B <- diag(c(1, 0, 0, rep(1, 7)))
colnames(B) <- rownames(B) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(B, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:11, background = "#ececec")
β1 β2 β3 β4 β5 β6 β7 β8 β9 β10
β1 1 0 0 0 0 0 0 0 0 0
β2 0 0 0 0 0 0 0 0 0 0
β3 0 0 0 0 0 0 0 0 0 0
β4 0 0 0 1 0 0 0 0 0 0
β5 0 0 0 0 1 0 0 0 0 0
β6 0 0 0 0 0 1 0 0 0 0
β7 0 0 0 0 0 0 1 0 0 0
β8 0 0 0 0 0 0 0 1 0 0
β9 0 0 0 0 0 0 0 0 1 0
β10 0 0 0 0 0 0 0 0 0 1

The constraint for positivity is Bβ>=0

which we express in R as

constraint2 <- B %*% betaHat >= 0

Now we are ready to solve the problem just as before.

problem <- Problem(objective, constraints = list(constraint1, constraint2))
result <- psolve(problem)
check_solver_status(problem)

And we can get the estimates of β.

m <- matrix(value(betaHat), ncol = 1)
rownames(m) <- paste0("\\(\\beta_{", 1:p, "}\\)")
knitr::kable(m, format = "html", escape = FALSE) |>
    kable_styling("striped") |>
    column_spec(1:2, background = "#ececec")
β1 0.0000000
β2 -2.8446952
β3 -1.7109771
β4 0.0000000
β5 0.6641308
β6 1.1781109
β7 2.3286139
β8 2.4144893
β9 4.2119052
β10 4.9483245

This demonstrates the chief advantage of CVXR: flexibility. Users can quickly modify and re-solve a problem, making our package ideal for prototyping new statistical methods. Its syntax is simple and mathematically intuitive. Furthermore, CVXR combines seamlessly with native R code as well as several popular packages, allowing it to be incorporated easily into a larger analytical framework. The user is free to construct statistical estimators that are solutions to a convex optimization problem where there may not be a closed form solution or even an implementation. Such solutions can then be combined with resampling techniques like the bootstrap to estimate variability.

Further Reading

We hope we have whet your appetite. You may wish to read a longer introduction with more examples.

We also have a number of tutorial examples available to study and mimic.

Session Info

R version 4.5.2 (2025-10-31)
Platform: aarch64-apple-darwin20
Running under: macOS Tahoe 26.3

Matrix products: default
BLAS:   /Library/Frameworks/R.framework/Versions/4.5-arm64/Resources/lib/libRblas.0.dylib 
LAPACK: /Library/Frameworks/R.framework/Versions/4.5-arm64/Resources/lib/libRlapack.dylib;  LAPACK version 3.12.1

locale:
[1] en_US.UTF-8/en_US.UTF-8/en_US.UTF-8/C/en_US.UTF-8/en_US.UTF-8

time zone: America/Los_Angeles
tzcode source: internal

attached base packages:
[1] stats     graphics  grDevices utils     datasets  methods   base     

other attached packages:
[1] nnls_1.6         CVXR_1.8.1       kableExtra_1.4.0

loaded via a namespace (and not attached):
 [1] gmp_0.7-5.1        clarabel_0.11.2    xml2_1.5.2         slam_0.1-55       
 [5] stringi_1.8.7      lattice_0.22-9     digest_0.6.39      magrittr_2.0.4    
 [9] evaluate_1.0.5     grid_4.5.2         RColorBrewer_1.1-3 fastmap_1.2.0     
[13] rprojroot_2.1.1    jsonlite_2.0.0     Matrix_1.7-4       ECOSolveR_0.6.1   
[17] backports_1.5.0    scs_3.2.7          Rmosek_11.1.1      viridisLite_0.4.3 
[21] scales_1.4.0       codetools_0.2-20   textshaping_1.0.4  cli_3.6.5         
[25] rlang_1.1.7        Rglpk_0.6-5.1      yaml_2.3.12        otel_0.2.0        
[29] tools_4.5.2        osqp_1.0.0         Rcplex_0.3-8       checkmate_2.3.4   
[33] here_1.0.2         gurobi_13.0-1      vctrs_0.7.1        R6_2.6.1          
[37] lifecycle_1.0.5    stringr_1.6.0      htmlwidgets_1.6.4  cccp_0.3-3        
[41] glue_1.8.0         Rcpp_1.1.1         systemfonts_1.3.1  xfun_0.56         
[45] rstudioapi_0.18.0  knitr_1.51         dichromat_2.0-0.1  highs_1.12.0-3    
[49] farver_2.1.2       htmltools_0.5.9    rmarkdown_2.30     svglite_2.2.2     
[53] piqp_0.6.2         compiler_4.5.2     S7_0.2.1          

References