Iterative Shrinkage Thresholding Algorithm (ISTA)求解
现有目标函数:
f ( x ) = 1 2 ∥ b − A x ∥ 2 2 + λ ∣ x − u ∣ f(x)=\frac{1}{2} \parallel \boldsymbol{b}-\mathbf{A}\boldsymbol{x} \parallel_2^2 + \lambda | \boldsymbol{x}- \boldsymbol{u}|f(x)=21∥b−Ax∥22+λ∣x−u∣
其中的变量都为向量,求 x \boldsymbol{x}x的解
将这样一个问题转化为标量形式
f ( x ) = 1 2 ( x − x 0 ) 2 + λ ∣ x − u ∣ f(x)=\frac{1}{2} ( x-x_0)^2 + \lambda | x- u|f(x)=21(x−x0)2+λ∣x−u∣
式子中有绝对值,需要分类讨论这个问题
当 x − u > 0 x- u>0x−u>0 即 x > u x>ux>u 时
f ( x ) = 1 2 ( x − x 0 ) 2 + λ ( x − u ) f(x)=\frac{1}{2} ( x-x_0)^2 + \lambda (x- u)f(x)=21(x−x0)2+λ(x−u)
有
f ′ ( x ) = x − x 0 + λ f'(x)=x-x_0 + \lambdaf′(x)=x−x0+λ
令f ′ ( x ) = 0 f'(x)=0f′(x)=0,则
x = x 0 − λ x=x_0 - \lambdax=x0−λ
前提是需要满足 x > u x>ux>u,即 x 0 − λ > u x_0 - \lambda >ux0−λ>u
所以当 x 0 > u + λ x_0 > u+\lambdax0>u+λ 时 x = x 0 − λ x=x_0 - \lambdax=x0−λ同理,当 x − u < 0 x- u<0x−u<0 即 x < u x<ux<u 时
f ( x ) = 1 2 ( x − x 0 ) 2 − λ ( x − u ) f(x)=\frac{1}{2} ( x-x_0)^2 - \lambda (x- u)f(x)=21(x−x0)2−λ(x−u)
有
f ′ ( x ) = x − x 0 − λ f'(x)=x-x_0 - \lambdaf′(x)=x−x0−λ
令f ′ ( x ) = 0 f'(x)=0f′(x)=0,则
x = x 0 + λ x=x_0 + \lambdax=x0+λ
前提是需要满足 x < u x<ux<u,即 x 0 + λ < u x_0 + \lambda <ux0+λ<u
所以当 x 0 < u − λ x_0 < u-\lambdax0<u−λ 时 x = x 0 + λ x=x_0 + \lambdax=x0+λ
综上,
x = { x 0 + λ , x 0 < u − λ u , u − λ ≤ x 0 ≤ u + λ x 0 − λ , x 0 > u + λ x= \begin{cases} x_0 + \lambda, \qquad x_0 < u-\lambda \\ u , \qquad u-\lambda \leq x_0 \leq u+\lambda\\ x_0 - \lambda, \qquad x_0 > u+\lambda \end{cases}x=⎩⎪⎨⎪⎧x0+λ,x0<u−λu,u−λ≤x0≤u+λx0−λ,x0>u+λ
这里解释一下,中间值为什么是 u uu,将边界值 x 0 = u + λ x_0=u+\lambdax0=u+λ 代入到 x = x 0 − λ x=x_0-\lambdax=x0−λ中,以及 x 0 = u − λ x_0=u-\lambdax0=u−λ 代入到 x = x 0 + λ x=x_0+\lambdax=x0+λ中, 恰好可以得到 x = u x=ux=u
以自变量为 x 0 x_0x0 , 应变量为 x xx , 画出函数图像是这样的

这其实就是一个软阈值函数,用符号 S ( ⋅ ) S(\cdot)S(⋅) 来表示, 定义为
S λ , u ( t ) = { t + λ , t < u − λ u , u − λ ≤ t ≤ u + λ t − λ , t > u + λ S_{\lambda, u}(t)= \begin{cases} t + \lambda, \qquad t < u-\lambda \\ u , \qquad u-\lambda \leq t \leq u+\lambda\\ t - \lambda, \qquad t > u+\lambda \end{cases}Sλ,u(t)=⎩⎪⎨⎪⎧t+λ,t<u−λu,u−λ≤t≤u+λt−λ,t>u+λ
则上式的解可以表示为
x = S λ , u ( x 0 ) x=S_{\lambda, u}(x_0)x=Sλ,u(x0)
这里的 x 0 x_0x0 是二次项的极值。
让我们回到目标函数:
f ( x ) = 1 2 ∥ b − A x ∥ 2 2 + λ ∣ x − u ∣ f(x)=\frac{1}{2} \parallel \boldsymbol{b}-\mathbf{A}\boldsymbol{x} \parallel_2^2 + \lambda | \boldsymbol{x}- \boldsymbol{u}|f(x)=21∥b−Ax∥22+λ∣x−u∣
二次项的导数为(如果不会求导的详见这里)
∂ f 1 ( x ) ∂ x = A T ( A x − b ) \frac{\partial f_1(x)}{\partial x}=A^T(Ax-b)∂x∂f1(x)=AT(Ax−b)
则梯度下降求其迭代解(非闭式解)为
x ( t + 1 ) = x ( t ) − 1 c A T ( A x − b ) = x ( t ) + 1 c A T ( b − A x ) x^{(t+1)}=x^{(t)} - \frac{1}{c} \boldsymbol{A}^T(\boldsymbol{A}\boldsymbol{x}-\boldsymbol{b})=\boldsymbol{x}^{(t)} + \frac{1}{c} \boldsymbol{A}^T(\boldsymbol{b}-\boldsymbol{Ax})x(t+1)=x(t)−c1AT(Ax−b)=x(t)+c1AT(b−Ax)
所以将上式代入软阈值函数里可得
x ( t + 1 ) = S λ , u ( x ( t ) + 1 c A T ( b − A x ) ) x^{(t+1)}=S_{\lambda, u}(\boldsymbol{x}^{(t)} + \frac{1}{c} \boldsymbol{A}^T (\boldsymbol{b}-\boldsymbol{Ax}))x(t+1)=Sλ,u(x(t)+c1AT(b−Ax))
这就是目标函数的迭代阈值解法。