WeiYa's Work Yard

A dog, who fell into the ocean of statistics, tries to write down his ideas and notes to save himself.

Dantzig Selector

Posted on (Update: )
Tags: Linear Programming, Linear Model, Dantzig

This post is based on Candes, E., & Tao, T. (2007). The Dantzig selector: Statistical estimation when $p$ is much larger than $n$. The Annals of Statistics, 35(6), 2313–2351.

Consider $y=X\beta+z$, where $\beta\in\IR^p$ is a parameter vector of interest, $X$ is a data matrix with possibly fewer rows than columns, $n< < p$, and $z_i$ are i.i.d. $N(0,\sigma^2)$. The Dantzig selector solves

where $r$ is the residual vector $y-X\tilde\beta$ and $t$ is a positive scalar. The paper shows that if $X$ obeys a uniform uncertainty principle (with unitnormed columns) and if the true parameter vector $\beta$ is sufficiently sparse (which here roughly guarantees that the model is identifiable), then with very large probability,

The DS program can be written as

for some $\lambda_p > 0$.

For the name, the author wrote, “with this name, we intend to pay tribute to the father of linear programming who passed away while we were finalizing this manuscript, and to underscore that the convex program (DS) is effectively a variable selection technique.”

Note that the DS program is convex, and can easily be recast as a linear program (LP),

It seems that the solution space is larger, but actually no effect on the original solution. Take a toy example, suppose we want to $\min \vert x\vert+\vert y\vert$, which can be converted to $\min u+v\text{ s.t. }-u\le x\le u, -v\le y\le v$, the original optimization problem aims to find the minimum contour, whose value equals to $u+v$, while the transformed problem aims to find the minimum vertex on the axis. Since the contour is determined by the vertex on the axis, these two problems are equivalent.

For a general linear program with inequality constraints

define

  • $f(\beta) = X\beta-b$
  • $r_{dual} = c+F^*\lambda$
  • $r_{cent} = -\diag(\lambda)f(\beta)-\1/t$

There might be a typo in the original paper, in which $r_{dual} = c+X^*\lambda$, but it should be $F^*$.

the primal-dual method Refer to the next post for more details updates the current $(\beta,\lambda)$ by means of a Newton step, and solves

that is,

The current guess is then updated via $(\beta^+,\lambda^+)=(\beta,\lambda)+s(\Delta\beta,\Delta s)$.

Letting $U=X^*X$ and $\tilde y=X^*y$, the problem parameters have the block structure

which gives

where $D_i=\diag(\lambda_i/f_i), 1\le i\le 4$, and

Put $\begin{bmatrix}r_1\ r_2\end{bmatrix} = r_{dual} + F^*\diag(1/f(\beta))r_{cent}$, it is convenient to solve the system

by elimination and obtain

In other words, each step may involve solving a $p$ by $p$ system of linear equations. In fact, when $n$ is less than $p$, it is possible to solve a smaller system thanks to the Sherman-Woodbury-Morrison formula. Write $U^*(D_3+D_4)U=X^*B$, where $B=X(D_3+D_4)X^*X$ and put $D_{12}=4(D_1+D_2)^{-1}D_1D_2$. Then

The advantage is that one needs to solve the smaller $n\times n$ system $(I+BD_{12}^{-1}X^*)\beta’=b’$.

The source code can be found at the author’s homepage, the url mentioned in the paper is invalid now.


Published in categories Memo