[轉]壓縮感知重構算法之迭代軟閾值(IST)


  徹底理解了迭代硬閾值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.


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM