 Software
 Open access
 Published:
Seagull: lasso, group lasso and sparsegroup lasso regularization for linear regression models via proximal gradient descent
BMC Bioinformatics volume 21, Article number: 407 (2020)
Abstract
Background
Statistical analyses of biological problems in life sciences often lead to highdimensional linear models. To solve the corresponding system of equations, penalization approaches are often the methods of choice. They are especially useful in case of multicollinearity, which appears if the number of explanatory variables exceeds the number of observations or for some biological reason. Then, the model goodness of fit is penalized by some suitable function of interest. Prominent examples are the lasso, group lasso and sparsegroup lasso. Here, we offer a fast and numerically cheap implementation of these operators via proximal gradient descent. The grid search for the penalty parameter is realized by warm starts. The step size between consecutive iterations is determined with backtracking line search. Finally, seagull the R package presented here produces complete regularization paths.
Results
Publicly available highdimensional methylation data are used to compare seagull to the established R package SGL. The results of both packages enabled a precise prediction of biological age from DNA methylation status. But even though the results of seagull and SGL were very similar (R^{2} > 0.99), seagull computed the solution in a fraction of the time needed by SGL. Additionally, seagull enables the incorporation of weights for each penalized feature.
Conclusions
The following operators for linear regression models are available in seagull: lasso, group lasso, sparsegroup lasso and Integrative LASSO with Penalty Factors (IPFlasso). Thus, seagull is a convenient envelope of lasso variants.
Background
Linear regression is a widely used tool to explore the dependence between a response variable and explanatory variables. For example, in genomewide association studies, counts of genetic variants along the genome are related to records of a disease or performance trait. The high throughput of modern biotechnological procedures enables studying an extremely large amount of explanatory variables (p). However, this often goes along with relatively few observations (n; p ≫ n), making the estimation of effects a challenge. As an example, highdimensional methylation data are recently used to build regression models termed epigenetic clocks, which enable biological age to be predicted from DNA methylation status. Especially in the presence of multicollinearity, penalization methods have proved to be useful; Tikhonov, elastic net [1] and lasso [2] regularization are famous examples.
Standard approaches for epigenetic clocks employ elastic net regression, which performs well but typically results in only ~ 100 methylation sites with nonzero effect, limiting the potential for their genomewide annotation and interpretation [3]. To simultaneously detect nonzero effects and account for the relatedness of explanatory variables, the lasso has been modified and enhanced to the group lasso [4], the sparsegroup lasso [5] and the “Integrative LASSO with Penalty Factors” (IPFlasso) [6]. These particular modifications of the lasso assume an underlying group structure within the explanatory variables. For instance, in genomewide association studies, a group structure can be identified from linkage and linkage disequilibrium among chromosome regions. Thus, a method that exploits this structure such as the (sparse)group lasso has the potential to improve the accuracy of results. seagull the R package presented here contains implementations of the lasso variants mentioned above focusing on precision of parameter estimation and computational efficiency.
Hereinafter we briefly describe the optimization problem and all relevant input parameters of seagull. We then use public data to evaluate our package and to compare it to the established R package SGL [7].
Implementation
The R package seagull offers regularization paths for optimization problems of the form:
This is also known as the sparsegroup lasso [5]. The first term expresses the “goodness of fit”. The second and third term are penalties, both of which are multiplied with the penalty parameter λ > 0. The vector y contains n observations of the response variable. The vectors b and u represent nonpenalized and penalized effects, respectively; X and Z are the corresponding design matrices. Moreover, α ∈ [0, 1] is the mixing parameter which convexly links the penalties.
In the two limiting cases of α = 1 and α = 0, the resulting objective function is the lasso [2] and the group lasso [4], respectively. However, if α is chosen to be less than 1, it is assumed that the explanatory variables have an underlying group/cluster structure (with nonoverlapping groups). Groups need to be determined prior to the call of seagull, for instance, by applying a suitable cluster algorithm to the explanatory variables or by grouping them according to the source of measurement (RNA expression, SNP genotypes, etc.). Referring to this structure, the entries of u can be separated into the corresponding groups, say u^{(l)} for group l and p_{l} is the size of group l (L is the total number of groups). Hence:
The penalty operators lasso, group lasso and sparsegroup lasso are available in seagull. Furthermore, it is possible to consider weights for each explanatory variable and group. Thus, the implemented extension of the optimization problem (1) is:
where \( {\omega}_j^F \) and \( {\omega}_l^G \) are positive weights for feature j and group l, respectively. The weights for groups are defined as:
where the average over weights of features is taken over those features that belong to group l, i.e., \( \overline{\omega_j^F}=\frac{1}{p_l}{\sum}_{j\ \mathrm{in}\ \mathrm{group}\ l}{\omega}_j^F \). Hence, if all weights \( {\omega}_j^F \) are set to 1, the optimization problem (2) yields problem (1).
The option of including weights can be used for any reason but it also enables the user to apply the strategy of IPFlasso. In order to show this, we go back to optimization problem (2) with α = 1:
For convenience, we assume the absence of any effects b and multiply the entire expression by 2n. Thus:
where a simplification can be obtained via \( {\lambda}_j=2 n\lambda {\omega}_j^F \):
As a last step we assume that the entries of u are obtained from M different sources, i.e., “modalities” – as called by the authors of [6]. Then, we let all λ’s which belong to the same modality m have the same value λ^{(m)}. Therefore, the last term in the above expression can be written as a sum over modalities:
And this immediately leads to the IPFlasso. So in the seagull package, this particular lasso variant is implicitly included. The weights for features just need to be set accordingly, i.e., the same weight for features that belong to the same modality.
The penalty parameter λ > 0 reflects the strength of the penalization. Our package provides the opportunity to calculate a maximal value for λ (i.e., λ_{max}) and to perform a grid search by gradually decreasing this value down to a minimal value (i.e., λ_{min}). This minimum value is determined as a userspecified proportion ξ of λ_{max}, i.e., λ_{min} = ξλ_{max}. The sequence of penalty parameters is then calculated on a logarithmic scale. To efficiently accelerate the corresponding grid search, we implemented warm starts. Thus, the solution of b and u for the current value of λ is used as starting point for the subsequent value of λ. Eventually, seagull provides a sequence of penalty parameters and calculates the corresponding path of solutions.
The optimization problem is solved via proximal gradient descent (PGD; e.g., [8]). PGD is an extension of gradient descent for optimization problems which contain nonsmooth parts, i.e., problems where the gradient is not available for the entire objective function. More details about this algorithm are presented in Additional file 4. As PGD is an iterative algorithm, a proper step size between consecutive iterations is crucial for convergence. This step size is determined with backtracking line search.
In the best case, an iterative algorithm such as PGD converges to the solution of the optimization problem. But typically, in the neighborhood of the solution the gain from one iteration to the next iteration decreases. Thus, a stopping criterion is implemented. Such a criterion is often based on a measurement of gain itself. In seagull, we implemented a stopping criterion which measures the gain from iteration k − 1 to k and scales it with the estimates at iteration k:
We refer to ε_{rel} as the relative accuracy, due to its definition as a ratio.
All implemented algorithms are based on the R package Rcpp 1.0.3 [9].
Data and evaluation criteria
We analyzed blood DNA methylation profiles at about 1.9 million CpG sites and its association with chronological age in mice (n = 141). The data set is publicly available and described in detail in [10]. We split the data set into training (n = 75) and validation (n = 66) data, where all age classes appeared almost equally in both sets, and applied the sparsegroup lasso variant of seagull 1.0.5. R scripts for processing and analyzing the data are available in the supplementary material (Additional files 1 and 2). Readytouse data are also available at Code Ocean (see Availability of data and materials).
We compared the outcome of seagull to that of the established R package SGL 1.3 [7]. Its implementation is based on accelerated generalized gradient descent. Both packages offer regularization paths to the same optimization problem (1). Thus, the input parameters for both packages are very similar. For example, we set the mixing parameter α to 0.95, a grid of 50 values for λ, and the ratio ξ between minimal and maximal λ equal to 0.001. However, despite the similarities between seagull and SGL, the implemented convergence criteria differ due to different meanings of accuracy parameters. In the SGL package, this parameter is an upper bound for the ℓ_{1}norm of the estimates of b and u. Unless stated otherwise, we set the accuracy parameter for SGL and seagull to 10^{− 4} and 10^{− 6}, respectively.
We used the following criteria to evaluate the two packages: the minimum mean squared error (MSE) of predicted age based on methylation data (i.e., methylation age) and measured chronological age in the validation set along the regularization path, the squared correlation coefficient R^{2} between predicted and chronological age, the number of features with an estimated effect different from zero (i.e., nonzeros), and the execution time needed to compute the entire regularization path.
Another example for the application of seagull in genomewide association studies is given in Additional file 3. It is shown how parameters (i.e., weights) can be tuned for IPFlasso.
Results and discussion
Figure 1a shows the model fit based on regression coefficients which led to the minimum mean squared error of chronological age in the validation set. The correlation between the chronological and the predicted age (“methylation age”) was 95.8%, and 5095 nonzero effects were identified with seagull. Hence, using only the identified fraction of CpG sites enabled a precise prediction of age. As an option for regulating the sparsity, increasing the accuracy parameter of seagull by two magnitudes (10^{− 6} to 10^{− 4}) increased the number of nonzero effects by one magnitude. Though the implemented convergence criteria differed between both packages, results were similar. The correlation between regression coefficients leading to the minimum mean squared error was 99.5% (Fig. 1b). The number of nonzero effects obtained with SGL was 8822. In contrast to SGL, seagull computed the solution in a fraction of the time (seagull: ~ 2 h; SGL: ~ 45 h).
Table 1 displays the impact of the accuracy parameter on evaluation criteria in detail. If the accuracy parameter of seagull was set to 10^{− 6} or 10^{− 8}, seagull outran SGL with respect to all evaluation criteria. The measures for R^{2} and nonzeros are both based on the value of λ for which the minimum MSE of prediction was obtained. The dependence between λ and the corresponding MSE is shown in Fig. 2.
In addition to SGL, seagull enables the opportunity to introduce weights for each penalized feature. This option was recently investigated in [11], where an optimization problem similar to (2) was used to estimate effects of SNP genotypes in a flowering plant breed (Arabidopsis thaliana). In that study, weights were defined according to the minor allele frequency (MAF) of genetic variants at each locus j, i.e., \( {\omega}_j^F=2\sqrt{MAF_j\left(1{MAF}_j\right)} \). Unlike seagull, the optimization problem described in [11] does not involve the incorporation of group weights other than the square root of the size of each group.
Conclusions
Here we introduced our R package seagull, which offers regularization paths for the lasso, group lasso, sparsegroup lasso, and IPFlasso for linear regression models. We compared seagull to the established R package SGL. Both packages delivered similar results in terms of mean squared error, squared correlation coefficient, and sparsity pattern. Despite these similarities, seagull computed the solution in a fraction of time that SGL required. Furthermore, only seagull offered the opportunity to incorporate weights for each penalized variable which enables further variants of the lasso such as the IPFlasso. In summary, seagull is a convenient envelope of lasso variants.
Availability and requirements
Project name: seagull.
Project home page: https://CRAN.Rproject.org/package=seagull
Source code: https://github.com/jklosa/seagull
Operating system(s): Platform independent.
Programming language: R, Rcpp.
Other requirements: R (> = 3.5.0).
License: GPL (> = 2).
Any restrictions to use by nonacademics: None.
Availability of data and materials
The methylation dataset is available in the Gene Expression Omnibus (GEO) database, https://www.ncbi.nlm.nih.gov/geo/query/acc.cgi?acc=GSE80672. seagull is an R package that is freely available on the Comprehensive R Archive Network (CRAN; https://CRAN.Rproject.org/package=seagull; vignette included). The source code is available on https://github.com/jklosa/seagull. The processed dataset, R scripts and results are also available at Code Ocean: https://codeocean.com/capsule/6412387/tree/v1.
Abbreviations
 IPFlasso:

Integrative LASSO with Penalty Factors
 lasso:

Least Absolute Shrinkage and Selection Operator
 MAF:

Minor Allele Frequency
 MSE:

Mean Squared Error
 PGD:

Proximal Gradient Descent
 SNP:

Single Nucleotide Polymorphism(s)
References
Zou H, Hastie T. Regularization and variable selection via the elastic net. J Royal Statistical Soc B. 2005 Apr;67(2):301–20. https://doi.org/10.1111/j.14679868.2005.00503.x.
Tibshirani R. Regression shrinkage and selection via the lasso. J Royal Stat Soc B (Methodological). 1996;58(1):267–88.
Bell CG, Lowe R, Adams PD, Baccarelli AA, Beck S, Bell JT, et al. DNA methylation aging clocks: challenges and recommendations. Genome Biol. 2019;25:20. https://doi.org/10.1186/s130590191824y.
Yuan M, Lin Y. Model selection and estimation in regression with grouped variables. J Royal Statistical Soc B. 2006 Feb;68(1):49–67. https://doi.org/10.1111/j.14679868.2005.00532.x.
Simon N, Friedman J, Hastie T, Tibshirani R. A sparsegroup lasso. J Comput Graph Stat. 2013 Apr;22(2):231–45. https://doi.org/10.1080/10618600.2012.681250.
Boulesteix AL, De Bin R, Jiang X, Fuchs M. IPFLASSO: integrative L1 penalized regression with penalty factors for prediction based on multiOmics data. Comput Mathematical Methods in Med. 2017;2017:1–14. https://doi.org/10.1155/2017/7691937.
Simon N, Friedman J, Hastie T, Tibshirani R. SGL: Fit a GLM (or Cox Model) with a Combination of Lasso and Group Lasso Regularization. 2019. https://CRAN.Rproject.org/package=SGL.
Parikh N, Boyd S. Proximal algorithms. FNT in Optimization. 2014;1(3):127–239.
Eddelbuettel D, Francois R, Allaire JJ, Ushey K, Kou Q, Russell N, et al. Rcpp: Seamless R and C++ Integration. 2019. https://CRAN.Rproject.org/package=Rcpp.
Petkovich DA, Podolskiy DI, Lobanov AV, Lee SG, Miller RA, Gladyshev VN. Using DNA Methylation Profiling to Evaluate Biological Age and Longevity Interventions. Cell Metab. 2017;25(4):954–60.e6. https://doi.org/10.1016/j.cmet.2017.03.016.
Che K, Chen X, Guo M, Wang C, Liu X. Genetic variants detection based on weighted sparse group lasso. Front Genet. 2020;11. https://doi.org/10.3389/fgene.2020.00155.
Acknowledgements
We thank the reviewers for their helpful comments.
Funding
This work was supported by the German Research Foundation [grant number WI 4450/2–1]. The funder had no role in the design of the study and collection, analysis, and interpretation of data and in writing the manuscript. Open access funding provided by Projekt DEAL.
Author information
Authors and Affiliations
Contributions
JK derived the algorithms, developed the package, performed the analysis, and wrote the manuscript. NS initialized the development of the algorithms, contributed to their implementation, and gave important suggestions. POW wrote the R script “Additional file 1” and contributed to the discussion. VL was involved in theoretical investigations. DW supervised the work and edited the manuscript. All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Ethics approval and consent to participate
Not applicable.
Consent for publication
Not applicable.
Competing interests
The authors declare that they have no competing interests.
Additional information
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Supplementary information
Additional file 1.
An R script for downloading and processing the methylation data used in this study.
Additional file 2.
An R script for the analysis of the processed data to generate Fig. 1.
Additional file 3.
An R script for performing an exemplary genomewide association study.
Additional file 4.
A document with information about proximal gradient descent for the sparsegroup lasso.
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/. The Creative Commons Public Domain Dedication waiver (http://creativecommons.org/publicdomain/zero/1.0/) applies to the data made available in this article, unless otherwise stated in a credit line to the data.
About this article
Cite this article
Klosa, J., Simon, N., Westermark, P.O. et al. Seagull: lasso, group lasso and sparsegroup lasso regularization for linear regression models via proximal gradient descent. BMC Bioinformatics 21, 407 (2020). https://doi.org/10.1186/s1285902003725w
Received:
Accepted:
Published:
DOI: https://doi.org/10.1186/s1285902003725w