LASSO and Elastic Net Computation

Batch Processing

LASSO and Elastic Net algorithms follow the general workflow described in Regression Usage Model.

Training

For a description of common input and output parameters, refer to Regression Usage Model. Both LASSO and Elastic Net algorithms have the following input parameters in addition to the common input parameters:

Training Input for LASSO and Elastic Net (Batch Processing)

Input ID

Input

weights

Optional input.

Pointer to the \(1 \times n\) numeric table with weights of samples. The input can be an object of any class derived from NumericTable except for PackedTriangularMatrix, PackedSymmetricMatrix, and CSRNumericTable.

By default, all weights are equal to 1.

gramMatrix

Optional input.

Pointer to the \(p \times p\) numeric table with pre-computed Gram Matrix. The input can be an object of any class derived from NumericTable except for CSRNumericTable.

By default, the table is set to an empty numeric table. It is used only when the number of features is less than the number of observations.

Chosse the appropriate tab to see the parameters used in LASSO and Elastic Net batch training algorithms:

Training Parameters for LASSO (Batch Processing)

Parameter

Default Value

Description

algorithmFPType

float

The floating-point type that the algorithm uses for intermediate computations. Can be float or double.

method

defaultDense

The computation method used by the LASSO regression. The only training method supported so far is the default dense method.

interceptFlag

True

A flag that indicates whether or not to compute

lassoParameters

A numeric table of size \(1 \times 1\) that contains the default LASSO parameter equal to \(0.1\).

\(L_1\) coefficients: \(\lambda_i\)

A numeric table of size \(1 \times k\) (where \(k\) is the number of dependent variables) or \(1 \times 1\). The contents of the table depend on its size:

  • For the table of size \(1 \times k\), use the values of LASSO parameters \(\lambda_j\) for \(j = 1, \ldots, k\).

  • For the table of size \(1 \times 1\), use the value of LASSO parameter for each dependant variable \(\lambda_1 = \ldots = \lambda_k\).

This parameter can be an object of any class derived from NumericTable, except for PackedTriangularMatrix, PackedSymmetricMatrix, and CSRNumericTable.

optimizationSolver

Coordinate Descent solver

Optimization procedure used at the training stage.

optResultToCompute

\(0\)

The 64-bit integer flag that specifies which extra characteristics of the LASSO regression to compute.

Provide the following value to request a characteristic:

  • computeGramMatrix for Computation Gram matrix

dataUseInComputation

doNotUse

A flag that indicates a permission to overwrite input data. Provide the following value to restrict or allow modification of input data:

  • doNotUse – restricts modification

  • doUse – allows modification

In addition, both LASSO and Elastic Net algorithms have the following optional results:

Training Output for LASSO and Elastic Net (Batch Processing)

Result ID

Result

gramMatrix

Pointer to the computed Gram Matrix with size \(p \times p\)

Prediction

For a description of the input and output, refer to Regression Usage Model.

At the prediction stage, LASSO and Elastic Net algorithms have the following parameters:

Prediction Parameters for LASSO and Elastic Net (Batch Processing)

Parameter

Default Value

Description

algorithmFPType

float

The floating-point type that the algorithm uses for intermediate computations. Can be float or double.

method

defaultDense

Default performance-oriented computation method, the only method supported by the regression-based prediction.

Performance Considerations

For better performance when the number of samples is larger than the number of features in the training data set, certain coordinates of gradient and Hessian are computed via the component of Gram matrix. When the number of features is larger than the number of observations, the cost of each iteration via Gram matrix depends on the number of features. In this case, computation is performed via residual update [Friedman2010].

To get the best overall performance for LASSO and Elastic Net training, do the following:

  • If the number of features is less than the number of samples, use homogenous table.

  • If the number of features is greater than the number of samples, use SOA layout rather than AOS layout.