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>0 x−u>0 即 x > u x>u x>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 + \lambda f′(x)=x−x0+λ
令 f ′ ( x ) = 0 f'(x)=0 f′(x)=0,则
x = x 0 − λ x=x_0 - \lambda x=x0−λ
前提是需要满足 x > u x>u x>u,即 x 0 − λ > u x_0 - \lambda >u x0−λ>u
所以当 x 0 > u + λ x_0 > u+\lambda x0>u+λ 时 x = x 0 − λ x=x_0 - \lambda x=x0−λ -
同理,当 x − u < 0 x- u<0 x−u<0 即 x < u x<u x<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 - \lambda f′(x)=x−x0−λ
令 f ′ ( x ) = 0 f'(x)=0 f′(x)=0,则
x = x 0 + λ x=x_0 + \lambda x=x0+λ
前提是需要满足 x < u x<u x<u,即 x 0 + λ < u x_0 + \lambda <u x0+λ<u
所以当 x 0 < u − λ x_0 < u-\lambda x0<u−λ 时 x = x 0 + λ x=x_0 + \lambda x=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
u
u,将边界值
x
0
=
u
+
λ
x_0=u+\lambda
x0=u+λ 代入到
x
=
x
0
−
λ
x=x_0-\lambda
x=x0−λ中,以及
x
0
=
u
−
λ
x_0=u-\lambda
x0=u−λ 代入到
x
=
x
0
+
λ
x=x_0+\lambda
x=x0+λ中, 恰好可以得到
x
=
u
x=u
x=u
以自变量为
x
0
x_0
x0 , 应变量为
x
x
x , 画出函数图像是这样的

这其实就是一个软阈值函数,用符号
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_0
x0 是二次项的极值。
让我们回到目标函数:
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))
这就是目标函数的迭代阈值解法。