徹底理解了迭代硬閾值IHT以后,很自然的會想到:如果將軟閾值(SoftThresholding)函數與Majorization-Minimization優化框架相結合形成迭代軟閾值(Iterative Soft Thresholding,IST)算法(另一種常見簡稱為ISTA,即IterativeSoftThresholding Algorithm,另外Iterative有時也作Iterated),應該可以解決如下優化問題[1]:

此即基追蹤降噪(Basis Pursuit De-Noising, BPDN)問題。
1、迭代軟閾值算法流程
為了解決優化問題(將原a換為習慣的x)

執行迭代算法

其中Ψλ是對每一個數值計算軟閾值函數值

這個算法稱為迭代軟閾值(Iterative Soft Thresholding,IST)算法。
2、迭代軟閾值算法推導
基追蹤降噪(Basis Pursuit De-Noising, BPDN)問題的目標函數

由於目標函數f(x)並不容易優化,根據Majorization-Minimization優化框架,可以優化替代目標函數

這里要求||Φ||2<1,則有

下面,我們對替代目標函數進行變形化簡

其中
,是與x無關的項;所以優化替代目標函數u(x,z)時,可以等價於優化

其中
。看到這個問題熟悉么?
對!這正是軟閾值(SoftThresholding)函數要解決的以下優化問題

對於標准的軟閾值(Soft Thresholding)函數來說,這個優化問題的解是

注意:這里的B是一個向量,應該是逐個元素分別執行軟閾值函數;其中標准的軟閾值(SoftThresholding)函數是:

將符號換為此處的優化問題

則解為

注意:這里的x*是一個向量,應該是逐個元素分別執行軟閾值函數;其中

然后我們根據Majorization-Minimization優化框架的流程進行迭代即可,注意z代表xn,即當前迭代值,而優化解soft(x*,λ)代表xn+1,用於下次迭代。
由於這個算法的整個過程相當於迭代執行軟閾值(SoftThresholding)函數,所以把它稱為迭代軟閾值(Iterative Soft Thresholding)算法。
3、迭代軟閾值算法MATLAB代碼
在IST函數中,一共規定了三種IST跳出迭代的條件:第一個是重構結果經Phi變換后與原先y的差異,第二是最大迭代次數,第三個是重構結果x兩次相鄰迭代的差異。
以下為2016-08-12更新版本V1.1,相比於原先的V1.0改動之處為第1個跳出迭代循環的條件參考TwIST作了修改:
function [ x ] = IST_Basic( y,Phi,lambda,epsilon,loopmax )
% Detailed explanation goes here
%Version: 1.0 written by jbb0523 @2016-08-09
%Version: 1.1 modified by jbb0523 @2016-08-12
if nargin < 5
loopmax = 10000;
end
if nargin < 4
epsilon = 1e-2;
end
if nargin < 3
lambda = 0.1*max(abs(Phi'*y));
end
[y_rows,y_columns] = size(y);
if y_rows<y_columns
y = y';%y should be a column vector
end
soft = @(x,T) sign(x).*max(abs(x) - T,0);
n = size(Phi,2);
x = zeros(n,1);%Initialize x=0
f = 0.5*(y-Phi*x)'*(y-Phi*x)+lambda*sum(abs(x));%added in v1.1
loop = 0;
fprintf('\n');
while 1
x_pre = x;
x = soft(x + Phi'*(y-Phi*x),lambda);%update x
loop = loop + 1;
f_pre = f;%added in v1.1
f = 0.5*(y-Phi*x)'*(y-Phi*x)+lambda*sum(abs(x));%added in v1.1
if abs(f-f_pre)/f_pre<epsilon%modified in v1.1
fprintf('abs(f-f_pre)/f_pre<%f\n',epsilon);
fprintf('IST loop is %d\n',loop);
break;
end
if loop >= loopmax
fprintf('loop > %d\n',loopmax);
break;
end
if norm(x-x_pre)<epsilon
fprintf('norm(x-x_pre)<%f\n',epsilon);
fprintf('IST loop is %d\n',loop);
break;
end
end
end
原先的V1.0版本:
function [ x ] = IST_Basic( y,Phi,lambda,epsilon,loopmax )
% Detailed explanation goes here
%Version: 1.0 written by jbb0523 @2016-08-09
if nargin < 6
loopmax = 10000;
end
if nargin < 5
epsilon = 1e-6;
end
if nargin < 4
lambda = 0.1*max(abs(Phi'*y));
end
[y_rows,y_columns] = size(y);
if y_rows<y_columns
y = y';%y should be a column vector
end
soft = @(x,T) sign(x).*max(abs(x) - T,0);
n = size(Phi,2);
x = zeros(n,1);%Initialize x=0
loop = 0;
fprintf('\n');
while 1
x_pre = x;
x = soft(x + Phi'*(y-Phi*x),lambda);%update x
loop = loop + 1;
if norm(y-Phi*x)<epsilon
fprintf('norm(y-Phi*x)<%f\n',epsilon);
fprintf('IST loop is %d\n',loop);
break;
end
if loop >= loopmax
fprintf('loop > %d\n',loopmax);
break;
end
if norm(x-x_pre)<epsilon
fprintf('norm(x-x_pre)<%f\n',epsilon);
fprintf('IST loop is %d\n',loop);
break;
end
end
end
該函數非常簡單,核心程序就一行,其它均為配角,為這一行服務的:
x = soft(x + Phi'*(y-Phi*x),lambda);%update x
4、迭代軟閾值算法測試
這里首先按傳統的測試程序進行重構測試,這里的λ取值參考了文獻【2】作者主頁給出的代碼里的方法(可以自己試一下,這個值到底是1還是2其實影響不大):
clear all;close all;clc;
M = 64;%觀測值個數
N = 256;%信號x的長度
K = 10;%信號x的稀疏度
Index_K = randperm(N);
x = zeros(N,1);
x(Index_K(1:K)) = 5*randn(K,1);%x為K稀疏的,且位置是隨機的
%x(Index_K(1:K)) = sign(5*randn(K,1));
Phi = randn(M,N);%測量矩陣為高斯矩陣
Phi = orth(Phi')';
sigma = 0.005;
e = sigma*randn(M,1);
y = Phi * x + e;%得到觀測向量y
% y = Phi * x;%得到觀測向量y
%% 恢復重構信號x
tic
% lamda = sigma*sqrt(2*log(N));
lamda = 0.1*max(abs(Phi'*y));
fprintf('\nlamda = %f\n',lamda)
%x_r = BPDN_quadprog(y,Phi,lamda);
x_r = IST_Basic(y,Phi,lamda);
toc
%% 繪圖
figure;
plot(x_r,'k.-');%繪出x的恢復信號
hold on;
plot(x,'r');%繪出原信號x
hold off;
legend('Recovery','Original')
fprintf('\n恢復殘差:');
fprintf('%f\n',norm(x_r-x));%恢復殘差
運行結果如下:(信號為隨機生成,所以每次結果均不一樣)
1)圖

2)CommandWindows:
lamda= 0.177639
norm(x-x_pre)<0.000001
ISTloop is 106
Elapsedtime is 0.008842 seconds.
恢復殘差:1.946232
從圖中可以看出,重構結果的位置基本都是正確的,但確比原始信號都要小一些,為什么呢?從Command Windows的輸出結果來看,IST迭代的106次,跳出迭代的條件是x在相鄰兩次迭代中基本不變了(norm(x-x_pre)<0.000001),也就是說最優點x已經基本不變了,若同樣執行使用MATLAB自帶的quadprog的基追蹤降噪BPDN_quadprog函數(參見壓縮感知重構算法之基追蹤降噪),重構結果是正常的,到底為什么本文的IST重構結果得不到近似的x呢?難道理論推導有問題?
直到看了文獻【2】中的Debiasing:


文中提到:“In many situations, itis worthwhile to debias the solution as a postprocessing step, to eliminate theattenuation of signal magnitude due to the presence of the regularization term.”,直譯過來就是“在很多場合,作為一個后處理步驟對結果進行除偏(debias)是很值得的,用於消除由於正則項的存在對信號幅度造成的衰減”,讀到這里就明白了,為什么本文的IST重構結果總是與真實的x幅度差一些呢?這是因為IST求解的優化問題含有正則項(regularizationterm),即λ||x||1。直觀地來講,因為目標函數中存在λ||x||1項,在最優化min時這個也要盡量的小,所以IST的恢復的結果比實際的x幅度要小一些(與參數λ有關)。解決這個問題的辦法就是對結果進行除偏(debias,注:有道詞典查不到此單詞,也沒看中文文獻是如何翻譯的,直接根據詞根琢磨了一下自己瞎翻譯的)。
如何除偏(debias)呢?接着來看:“Inthe debiasing step, we fix at zero those individual components or groups thatare zero at the end of the SpaRSA process, and minimize the objective over theremaining elements.”,翻譯一下就是“將SpaRSA重構結果中的為零的項固定為零,然后在剩除項上對目標函數進行最小化”,簡單一點說就是優化文中的式(27)(正好與本文IST解決的問題相同,后來會專門分析SpaRSA算法):

其中AI的含義更清晰的解釋如下:
若SpaRSA重構的x為:

原來的矩陣A為:

則AI等於原矩陣A只保留第3列(對應x中的元素a)和第5列(對應x中的元素b)的子矩陣:

文獻【2】后面繼續談到,若要求解式(27)可以使用共軛梯度法(Conjugate gradient)。實際上,式(27)的最優解為最小二乘解(熟悉匹配追蹤系列算法的應該對這個很清楚,可參見壓縮感知中的數學知識:投影矩陣(projectionmatrix)),即

在SpaRSA算法中,作者給出的官方代碼里包括了debias過程,由一個參數來控制是否對重構結果執行debias過程,這里為了簡單,就不修改IST函數了,直接在測試代碼中加入debias代碼:
clear all;close all;clc;
M = 64;%觀測值個數
N = 256;%信號x的長度
K = 10;%信號x的稀疏度
Index_K = randperm(N);
x = zeros(N,1);
x(Index_K(1:K)) = 5*randn(K,1);%x為K稀疏的,且位置是隨機的
%x(Index_K(1:K)) = sign(5*randn(K,1));
Phi = randn(M,N);%測量矩陣為高斯矩陣
Phi = orth(Phi')';
sigma = 0.005;
e = sigma*randn(M,1);
y = Phi * x + e;%得到觀測向量y
% y = Phi * x;%得到觀測向量y
%% 恢復重構信號x
tic
% lamda = sigma*sqrt(2*log(N));
lamda = 0.1*max(abs(Phi'*y));
fprintf('\nlamda = %f\n',lamda)
%x_r = BPDN_quadprog(y,Phi,lamda);
x_r = IST_Basic(y,Phi,lamda);
toc
%Debias
[xsorted inds] = sort(abs(x_r), 'descend');
AI = Phi(:,inds(xsorted(:)>1e-3));
xI = pinv(AI'*AI)*AI'*y;
x_bias = zeros(length(x),1);
x_bias(inds(xsorted(:)>1e-3)) = xI;
%% 繪圖
figure;
plot(x_r,'k.-');%繪出x的恢復信號
hold on;
plot(x,'r');%繪出原信號x
hold off;
legend('Recovery','Original')
fprintf('\n恢復殘差(original):');
fprintf('%f\n',norm(x_r-x));%恢復殘差
%Debias
figure;
plot(x_bias,'k.-');%繪出x的恢復信號
hold on;
plot(x,'r');%繪出原信號x
hold off;
legend('Recovery-debise','Original')
fprintf('恢復殘差(debias):');
fprintf('%f\n',norm(x_bias-x));%恢復殘差
運行結果如下:(信號為隨機生成,所以每次結果均不一樣)
1)圖:
第一幅圖(IST重構結果與原信號對比):

第二幅圖(對IST重構結果debias后的結果與原信號對比):

2)CommandWindows:
lamda = 0.405767
norm(x-x_pre)<0.000001
IST loop is 74
Elapsed time is 0.007175 seconds.
恢復殘差(original):3.736317
恢復殘差(debias):0.847947
從第二幅圖的結果來看,debias后的結果已與BPDN_quadprog函數結果(參見壓縮感知重構算法之基追蹤降噪)在同一個水平了。
值得注意的是,在求最小二乘解的公式里包含矩陣求逆的過程,矩陣求逆在大規模(large scale)問題中是一個比較麻煩的問題,IST等迭代算法(包括IHTs等)本身相比於MP等算法的優勢即為不含矩陣求逆,計算效率高,因此實際中求解文獻【2】的式(27)不應該直接采用如上包含矩陣求逆的方式,尤其是大規模問題,否則IST算法計算量小的優勢就沒有了,此處僅是為了示范debias的效果。
5、結束語
值得注意的是,一般文獻中出現的IST或ISTA簡稱中的“S”並非指的是“soft”,而是“shrinkage”,即“IteratedShrinkage/ThresholdingAlgorithm”,至於“shrinkage”是什么意思呢?我們還是先來看一下有道詞典吧:

那么Iterative Soft Thresholding和Iterated Shrinkage/Thresholding有什么區別呢?你要認為它們是一樣的也沒問題,因為從文獻中來看,很多作者的確是這么認為的;另外,還可以認為Iterative Soft Thresholding是Iterated Shrinkage/Thresholding的一種特殊情況,即每次迭代時正好是求軟閾值函數時的特殊情況,而Iterated Shrinkage/Thresholding更是一種廣義的稱呼。
值得注意的是,本文參考文獻很少,且均是與IST“不相關”的,這是因為沒有哪一種文獻明確提出IST,所以也不知道引用哪個了。
在下一篇中,我們從多篇文獻中來看一看究竟什么是“Shrinkage”……
IST:Iterative Shrinkage/Thresholding和Iterative Soft Thresholding
6、參考文獻
【1】Chen S S, Donoho D L,Saunders M A.Atomic decomposition by basis pursuit[J]. SIAM review, 2001, 43(1): 129-159.
【2】WrightS J, Nowak R D, Figueiredo M A T. Sparse reconstruction by separable approximation.[J]. IEEE Transactions on Signal Processing, 2009,57(7):3373-3376.
