同时包含一范数和二范数的公式求解问题

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)=21bAx22+λxu
其中的变量都为向量,求 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(xx0)2+λxu
式子中有绝对值,需要分类讨论这个问题

  1. x − u > 0 x- u>0xu>0x > 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(xx0)2+λ(xu)

    f ′ ( x ) = x − x 0 + λ f'(x)=x-x_0 + \lambdaf(x)=xx0+λ
    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λ

  2. 同理,当 x − u < 0 x- u<0xu<0x < 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(xx0)2λ(xu)

    f ′ ( x ) = x − x 0 − λ f'(x)=x-x_0 - \lambdaf(x)=xx0λ
    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λx0u+λ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λtu+λ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)=21bAx22+λxu
二次项的导数为(如果不会求导的详见这里
∂ f 1 ( x ) ∂ x = A T ( A x − b ) \frac{\partial f_1(x)}{\partial x}=A^T(Ax-b)xf1(x)=AT(Axb)

则梯度下降求其迭代解(非闭式解)为
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(Axb)=x(t)+c1AT(bAx)
所以将上式代入软阈值函数里可得

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(bAx))
这就是目标函数的迭代阈值解法。


版权声明:本文为Mr_Lowbee原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。