Support union recovery in high-dimensional multivariate regression

Abstract

In multivariate regression, a K-dimensional response vector is regressed upon a common set of p covariates, with a matrix B^∗ ∈ ℝ^p × K of regression coefficients. We study the behavior of the multivariate group Lasso, in which block regularization based on the ℓ₁ ∕ ℓ₂ norm is used for support union recovery, or recovery of the set of s rows for which B^∗ is nonzero. Under high-dimensional scaling, we show that the multivariate group Lasso exhibits a threshold for the recovery of the exact row pattern with high probability over the random design and noise that is specified by the sample complexity parameter θ(n, p, s) := n ∕ [2ψ(B^∗) log(p − s)]. Here n is the sample size, and ψ(B^∗) is a sparsity-overlap function measuring a combination of the sparsities and overlaps of the K-regression coefficient vectors that constitute the model. We prove that the multivariate group Lasso succeeds for problem sequences (n, p, s) such that θ(n, p, s) exceeds a critical level θ_u, and fails for sequences such that θ(n, p, s) lies below a critical level θ_ℓ. For the special case of the standard Gaussian ensemble, we show that θ_ℓ = θ_u so that the characterization is sharp. The sparsity-overlap function ψ(B^∗) reveals that, if the design is uncorrelated on the active rows, ℓ₁ ∕ ℓ₂ regularization for multivariate regression never harms performance relative to an ordinary Lasso approach and can yield substantial improvements in sample complexity (up to a factor of K) when the coefficient vectors are suitably orthogonal. For more general designs, it is possible for the ordinary Lasso to outperform the multivariate group Lasso. We complement our analysis with simulations that demonstrate the sharpness of our theoretical results, even for relatively small problems.