文獻(xiàn)標(biāo)識碼: A
DOI:10.16157/j.issn.0258-7998.2017.01.037
中文引用格式: 宋長明,,惠慶磊,,程東旭. 基于字典學(xué)習(xí)和加權(quán)TV的MRI重構(gòu)算法[J].電子技術(shù)應(yīng)用,,2017,,43(1):141-144.
英文引用格式: Song Changming,Hui Qinglei,,Cheng Dongxu. A novel MR image reconstruction algorithm based on dictionary learning and weighted total variation[J].Application of Electronic Technique,,2017,43(1):141-144.
0 引言
核磁共振圖像(Magnetic Resonance Image,,MRI)具有無輻射,、高分辨率,、多平面成像等優(yōu)點,廣泛應(yīng)用于臨床診斷,。然而,,緩慢的掃描速度會導(dǎo)致運動偽影的產(chǎn)生,致使圖像變得模糊和失真,,因此研究如何在保證高質(zhì)量圖像的情況下提高成像速度具有重要的理論和實際應(yīng)用價值,。壓縮感知(Compressive Sensing,CS)理論是DONOHO D L,、CAND?魬S E J和TAO T等[1-3]在2006年提出的一種全新的信號采樣和處理理論,。CS理論提供了一個利用少量測量數(shù)據(jù)實現(xiàn)稀疏信號重建的框架,極大地減少傅里葉變換域的采樣數(shù)據(jù),,縮短掃描時間,,提高成像速度。
在MRI領(lǐng)域,,CS理論得到極大的認(rèn)可,。LUSTIG M等[4]利用MRI在小波域的稀疏性和空間域的變分約束,將CS理論成功應(yīng)用于心臟成像,、腦成像,、快速三維血管造影等,并取得良好的重建效果,。在Lusting等人的研究基礎(chǔ)上,,DABOV K等[5]在對局部相似性研究的基礎(chǔ)上,提出一種新的塊匹配重建算法,。AK?覶AKAYA M等[6]利用塊匹配的重建算法對中心采樣的心臟圖像進(jìn)行重建,,實現(xiàn)了在4倍采樣的MRI的精確重建。RAVISHANKAR S等[7]利用K-SVD思想,,提出一種基于自適應(yīng)字典學(xué)習(xí)的MRI重建算法,,重建效果得到進(jìn)一步提高。RAJWADE A和LINGALA S G等[8,,9]提出的具有盲字典學(xué)習(xí)的高光譜成像和動態(tài)MRI算法自適應(yīng)學(xué)習(xí)字典,獲得了良好的重構(gòu)性能,。
字典學(xué)習(xí)[10]可以有效表征圖像的特征信息,全變分(Total Variation,,TV)正則項[11]可以保持圖像邊緣信息,。然而,在字典學(xué)習(xí)過程中稀疏編碼會產(chǎn)生混亂信息熵,,影響字典學(xué)習(xí)效率,,TV單向梯度處理平滑區(qū)域會產(chǎn)生階梯效應(yīng)。針對以上問題,,本文引入熵約束和加權(quán)的TV正則項,,充分利用圖像的先驗信息,,構(gòu)建圖像重構(gòu)模型。
1 壓縮感知基本理論
基于稀疏表示理論,,CS理論將信號采樣和壓縮合并進(jìn)行,,通過非自適應(yīng)線性投影有效獲取信號的特征信息,并利用求解最優(yōu)化問題高概率精確重構(gòu)原信號,,緩解了數(shù)據(jù)的采集,、存儲、傳輸和分析的壓力,。一般地,,對于原始信號為x∈CN,給定測量值y=Φx∈CM,,其中Φ∈CM×N(M<N)是非滿秩測量矩陣,,存在無窮多個可行解x,若Φ滿足約束等距條件,,則可確保重構(gòu)解唯一,。信號重構(gòu)過程轉(zhuǎn)換為最小化L0范數(shù)模型,即:
其中||·||0是零范數(shù),,表示x中非零元素的個數(shù),。求解式(1)不僅是一個NP難問題,而且極易受到噪聲的影響,。針對此問題,,研究者采用凸的L1范數(shù)代替非凸的L0范數(shù),即:
選擇一個合適的拉格朗日乘子λ,,式(2)可轉(zhuǎn)化為無約束優(yōu)化問題:
采用Bregman算法,、對偶算法以及交替迭代乘子法等算法均可有效求解式(3)。最近研究表明,,圖像信息中包含了大量的自相關(guān)結(jié)構(gòu),,具有高度結(jié)構(gòu)化稀疏性以及低秩特性,利用這種非局部相似特性構(gòu)建的低秩結(jié)構(gòu)化稀疏模型[12]顯示出更為優(yōu)異的圖像重構(gòu)效果,。
2 圖像重構(gòu)模型的提出
由于字典學(xué)習(xí)過程中稀疏系數(shù)索引會產(chǎn)生較高的信息熵,影響原子學(xué)習(xí)效率,。而單一的TV正則項又會對圖像邊緣紋理過平滑,,產(chǎn)生階梯效應(yīng)。為此,,基于熵約束和加權(quán)TV正則項提出一種新的圖像重構(gòu)模型,,如下所示:
其中E(T,u)是字典學(xué)習(xí)正則項,,G(u)是加權(quán)TV正則項,,y=Φu是數(shù)據(jù)約束項,。
2.1 基于熵約束的字典學(xué)習(xí)正則項
稀疏表示是影響圖像重構(gòu)質(zhì)量的關(guān)鍵因素,為克服傳統(tǒng)變換不能有效表示輪廓,、紋理,、振蕩等高維幾何特征的缺點,稀疏表示的研究從正交基擴(kuò)展到過完備字典,。式(4)中的第一項E(T,,u)是字典學(xué)習(xí)約束項,字典學(xué)習(xí)方法一般可通過優(yōu)化下式獲?。?/p>
在稀疏編碼過程中會得到非零系數(shù)和相對應(yīng)的位置索引,,這些索引值有很大的隨機(jī)性,導(dǎo)致其信息熵非常高,,降低了字典學(xué)習(xí)效率,。為此,引入熵的約束,,以提高字典學(xué)習(xí)效率,。對保真項、稀疏度以及原子選擇的熵作聯(lián)合優(yōu)化,,構(gòu)造新的優(yōu)化函數(shù):
2.2 加權(quán)各向同性與異性的TV正則項
TV模型廣泛應(yīng)用于圖像去噪,、修復(fù)、超分辨率等,其各向同性,、各向異性的數(shù)學(xué)表達(dá)如下:
其中Dx,、Dy分別表示水平和垂直方向上的導(dǎo)數(shù)。
對于分段常數(shù)函數(shù)或光滑區(qū)域,,其每個像素點的梯度是1-稀疏,,通過L1范數(shù)可以很好地重構(gòu),然而,,對于邊緣信息豐富的區(qū)域,,梯度并非1-稀疏,其效果并不理想,,它會產(chǎn)生階梯效應(yīng),。針對這種非稀疏的梯度向量,結(jié)合各向同性與各向異性,,用加權(quán)的TV正則項代替單一的TV項,,即:
其中γ∈[0,1],。其優(yōu)點在于具有Lipschitz正則性,,并且在求解時用現(xiàn)有的算法可充分確保其收斂性。
2.3 基于字典學(xué)習(xí)和加權(quán)TV的重構(gòu)模型
結(jié)合上述字典學(xué)習(xí)和TV模型,構(gòu)建如下重構(gòu)模型:
式(9)第一項保證圖像塊可以用過完備字典稀疏表示,,Rj∈Rn×N表示提取圖像u的第j塊的二值矩陣,,那么所有圖像塊的組合能覆蓋整個圖像,可能有重疊部分,,這里的圖像塊大小和過完備字典中的原子大小必須是一致的,,才能保證每個圖像塊都能被字典中的原子線性表示。第二項是字典原子熵約束項,,提高字典學(xué)習(xí)效率,。第三項是基于加權(quán)的TV正則項,有效保留圖像邊緣信息,,可抑制階梯效應(yīng),。前三項起到去除采樣過疏導(dǎo)致偽影和保留圖像局部結(jié)構(gòu)的作用。第四項是數(shù)據(jù)保真項,,用于增強(qiáng)k空間的數(shù)據(jù)保真度,,其中
是感知矩陣,P是采樣矩陣,,
是傅里葉變換,。
3 模型求解
本文采用交替最小化方法求解式(9)。首先固定u,,獲得學(xué)習(xí)字典和稀疏系數(shù),。然后,固定字典和系數(shù),依據(jù)測量數(shù)據(jù)更新重構(gòu)圖像u,。
(1)字典學(xué)習(xí)階段對應(yīng)的子問題目標(biāo)函數(shù)為:
采用K-SVD方法求解,,其相應(yīng)的OMP算法需要修改,在每次迭代計算殘差時應(yīng)加入選擇原子熵,,即:
(2)更新重構(gòu)階段對應(yīng)的子問題目標(biāo)函數(shù)為:
上述能量泛函是凸的,,通過交替迭代轉(zhuǎn)化為求解關(guān)于u、d的兩個子問題,。
對u問題,,利用變分法直接求導(dǎo),引入周期性邊界條件,并使用快速傅里葉變換(FFT)進(jìn)行求解,,可得:
4 仿真實驗與結(jié)果分析
實驗環(huán)境為MATLAB 2011a版本,,為了能夠有效對比重構(gòu)圖像的不同特征,選取光滑和紋理區(qū)域差異較大的兩張腦部圖像,,尺寸為256×256,,并采用兩種不同的采樣方式,如圖1所示,。實驗中待重構(gòu)數(shù)據(jù)是模擬對原始圖像的二維離散傅里葉變換,采樣得到,。本文與字典模型,、TV模型作對比,,以信噪比(SNR)、峰值信噪比(PSNR),、結(jié)構(gòu)相似度(SSIM)作為指標(biāo)衡量重構(gòu)效果,。圖像塊大小為8×8,字典原子維數(shù)M=64,,個數(shù)K=256,,以離散余弦變換字典為初始字典,平衡參數(shù)α=1,,β=1,,稀疏度T0=10,λ=140,,分別取γ=0.5,、1,迭代10次,。
利用零填充,、TV、字典學(xué)習(xí)3種不同算法,,分別對兩幅不同特征的MRI圖像在16%和25%采樣率下進(jìn)行了多組實驗,,實驗結(jié)果如圖2、圖3所示,。不同圖像重構(gòu)下SNR,、PSNR、SSIM指標(biāo)對比見表1,、表2,。從實驗結(jié)果看,TV算法會對圖像邊緣信息過平滑,,丟失紋理細(xì)節(jié)信息,。同時,為了對比本文算法在不同采樣矩陣下的恢復(fù)效果,。綜合數(shù)據(jù)看,,L1-L2恢復(fù)效果優(yōu)于L1-0.5L2的效果,本文算法重構(gòu)的圖像比TV或字典模型PSNR平均高出1~2 dB,,在平滑區(qū)和邊緣紋理區(qū)處理結(jié)果都優(yōu)于單純的TV和字典算法,。
5 結(jié)論
本文基于圖像塊稀疏性提出一個自適應(yīng)的字典學(xué)習(xí)框架,該框架直接由采樣的空間數(shù)據(jù)獲得,,對圖像樣本具有自適應(yīng)性,,又結(jié)合加權(quán)TV正則項,充分地利用各自的優(yōu)點,有效消除噪聲,,保留圖像邊緣紋理,,增強(qiáng)局部結(jié)構(gòu)。實驗結(jié)果表明,,本文對噪聲具有魯棒性,,在無噪聲和有噪聲的情況下均具有良好的性能。在高度欠采樣下,,圖像的各種特征有很好的重構(gòu)效果,,算法經(jīng)過少量的迭代就可以收斂,對參數(shù)的選取同樣具有魯棒性,。下一步將從參考圖像出發(fā),,選擇合適初始字典,考慮低秩流形結(jié)構(gòu)和塊稀疏字典兩者相結(jié)合,使MRI重構(gòu)圖像具有更豐富的結(jié)構(gòu)信息,。
參考文獻(xiàn)
[1] DONOHO D L.Compressed sensing[J].IEEE Transactions on Information Theory,,2006,52(4):1289-1306.
[2] CANDES E J.Compressive sampling[J].Marta Sanz Solé,,
2006,,17(2):1433-1452.
[3] CANDES E J,ROMBERG J,,TAO T.Robust uncertainty principles: exact signal reconstruction from highly incomplete frequency information[J].IEEE Transactions on Information Theory,,2006,52(2):489-509.
[4] LUSTIG M,,DONOHO D,,PAULY J M.Sparse MRI:The application of compressed sensing for rapid MR imaging[J].Magnetic Resonance in Medicine,2007,,58(6):1182-95.
[5] DABOV K,,F(xiàn)OI A,KATKOVNIK V,,et al.Image denoising by sparse 3-D transform-domain collaborative filtering[J].IEEE Transactions on Image Processing A Publication of the IEEE Signal Processing Society,,2007,16(8):2080-2095.
[6] AKCAKAYA M,,BASHA T A,,GODDU B,et al.Low-dimensional-structure self-learning and thresholding:Regularization beyond compressed sensing for MRI Reconstruction[J].Magnetic Resonance in Medicine Official Journal of the Society of Magnetic Resonance in Medicine,,2011,,66(3):756-767.
[7] RAVISHANKAR S,BRESLER Y.MR image reconstruction from highly undersampled k-space data by dictionary learning[J].IEEE Transactions on Medical Imaging,,2011,,30(5):1028-41.
[8] RAJWADE A,,KITTLE D,TSAI T H,,et al.Coded hyper-spectral imaging and blind compressive sensing[J].Siam Journal on Imaging Sciences,,2013,,6(2):782-812.
[9] LINGALA S G,,JACOB M.A blind compressive sensing frame work for accelerated dynamic MRI[J].Proceedings,2012,,88(1):1060-1063.
[10] KONDO S.Compressed sensing and redundant dictionaries[J].Information Theory IEEE Transactions on,,2008,54(5):2210-2219.
[11] RUDIN L I,,OSHER S,,F(xiàn)ATEMI E.Nonlinear total variation based noise removal algorithms[J].Physica D-nonlinear Phenomena,1992,,60(1-4):259-268.
[12] DONG W,,SHI G,LI X,,et al.Compressive sensing via nonlocal low-rank regularization[J].IEEE Transactions on Image Processing A Publication of the IEEE Signal Processing Society,,2014,23(8):3618-3632.
作者信息:
宋長明,,惠慶磊,,程東旭
(中原工學(xué)院 理學(xué)院,河南 鄭州450007)