Quant Memo
Core

Group Lasso

Group Lasso extends ordinary Lasso from selecting individual coefficients to selecting whole predefined groups of them together, which matters whenever features come bundled — like the dummy columns of one categorical variable.

Prerequisites: Elastic Net Regularization

Ordinary Lasso decides, coefficient by coefficient, which ones to zero out. That works fine when each feature stands alone, but breaks down when features naturally come in bundles — a one-hot-encoded categorical variable with five levels produces five coefficients that are really one decision ("does this categorical variable matter at all?"), yet Lasso is free to zero out three of the five and keep two, a nonsensical partial answer.

Group Lasso fixes this by penalizing predefined groups of coefficients together rather than individually:

minβ  Loss(β)+λg=1Gpgβg2\min_\beta \; \text{Loss}(\beta) + \lambda \sum_{g=1}^{G} \sqrt{p_g} \, \|\beta_g\|_2

In words: instead of penalizing the absolute size of each coefficient separately, penalize the overall magnitude (the 2\ell_2 norm) of each whole group gg of coefficients, scaled by that group's size — so an entire group is driven to zero together, or kept together, never split arbitrarily.

Group Lasso makes an all-or-nothing decision at the level of a meaningful feature (a whole categorical variable, a whole set of polynomial terms for one input), rather than letting ordinary Lasso make inconsistent partial decisions inside a group that should stand or fall as one unit.

Worked example. A model includes a "region" categorical variable one-hot encoded into 4 dummy columns and a "sector" variable encoded into 6 dummy columns. Fit with group Lasso treating each categorical variable as one group, the penalty might zero out all 4 region coefficients together (region isn't predictive) while keeping all 6 sector coefficients nonzero (sector is) — a clean, interpretable "keep this whole variable or drop it" outcome that ordinary per-coefficient Lasso cannot guarantee.

Related concepts

Further reading

  • Yuan & Lin, 'Model Selection and Estimation in Regression with Grouped Variables' (2006)
ShareTwitterLinkedIn