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

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>0 xu>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(xx0)2+λ(xu)

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

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

    f ′ ( x ) = x − x 0 − λ f'(x)=x-x_0 - \lambda f(x)=xx0λ
    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λx0u+λ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λtu+λ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)=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))
这就是目标函数的迭代阈值解法。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值