The biglasso Package: A Memory- and Computation-Efficient Solver for Lasso Model Fitting with Big Data in R

Penalized regression models such as the lasso have been extensively applied to analyzing high-dimensional data sets. However, due to memory limitations, existing R packages like glmnet and ncvreg are not capable of fitting lasso-type models for ultrahigh-dimensional, multi-gigabyte data sets that are increasingly seen in many areas such as genetics, genomics, biomedical imaging, and high-frequency finance. In this research, we implement an R package called biglasso that tackles this challenge. biglasso utilizes memory-mapped files to store the massive data on the disk, only reading data into memory when necessary during model fitting, and is thus able to handle out-of-core computation seamlessly. Moreover, it’s equipped with newly proposed, more efficient feature screening rules, which substantially accelerate the computation. Benchmarking experiments show that our biglasso package, as compared to existing popular ones like glmnet, is much more memory- and computation-efficient. We further analyze a 36 GB simulated GWAS data set on a laptop with only 16 GB RAM to demonstrate the out-of-core computation capability of biglasso in analyzing massive data sets that cannot be accommodated by existing R packages.

Yaohui Zeng (Department of Biostatistics) , Patrick Breheny (Department of Biostatistics)
2021-01-14

Supplementary materials

Supplementary materials are available in addition to this article. It can be downloaded at RJ-2021-001.zip

References

Reuse

Text and figures are licensed under Creative Commons Attribution CC BY 4.0. The figures that have been reused from other sources don't fall under this license and can be recognized by a note in their caption: "Figure from ...".

Citation

For attribution, please cite this work as

Zeng & Breheny, "The biglasso Package: A Memory- and Computation-Efficient Solver for Lasso Model Fitting with Big Data in R", The R Journal, 2021

BibTeX citation

@article{RJ-2021-001,
  author = {Zeng, Yaohui and Breheny, Patrick},
  title = {The biglasso Package: A Memory- and Computation-Efficient Solver for Lasso Model Fitting with Big Data in R},
  journal = {The R Journal},
  year = {2021},
  note = {https://doi.org/10.32614/RJ-2021-001},
  doi = {10.32614/RJ-2021-001},
  volume = {12},
  issue = {2},
  issn = {2073-4859},
  pages = {6-19}
}