圖像降噪算法——低秩聚類：WNNM算法

• 圖像降噪算法——低秩聚類：WNNM算法
• • 1. 基本原理
  • 2. matlab代碼
  • 3. 結(jié)論

圖像降噪算法——低秩聚類：WNNM算法

同樣是為了完善自己知識(shí)版圖的完整性，我決定再補(bǔ)充下低秩聚類算法的相關(guān)算法，低秩聚類算法同樣是一大類算法，這篇博客是挑選了其中最經(jīng)典的一種算法WNNM算法進(jìn)行展開學(xué)習(xí)，由于沒有在這方面做過(guò)太多相關(guān)的工作，因此可能理解相對(duì)膚淺，還請(qǐng)讀者見諒

1. 基本原理

在寫這篇博客之前，我先學(xué)習(xí)了稀疏表達(dá)相關(guān)的知識(shí)，這里我從稀疏表達(dá)算法出發(fā)，引入低秩聚類算法的相關(guān)解釋：

在稀疏表達(dá)相關(guān)算法中，將圖像建模成$\mathbf{Y}=\mathbf{D}\mathbf{X}$的形式，其中$\mathbf{Y}$是有單個(gè)樣本按列組成而成的樣本矩陣，注意這里的樣本并沒有要求是相似的，$\mathbf{D}$是字典矩陣，矩陣中每一列為一個(gè)基向量或者子空間，$\mathbf{X}$則為系數(shù)矩陣，在系數(shù)表達(dá)中給定的限制條件是要求系數(shù)矩陣是稀疏的。我們通過(guò)一個(gè)過(guò)完備的字典和稀疏的系數(shù)矩陣就可以還原樣本，而噪聲不行，因此可以通過(guò)稀疏表達(dá)進(jìn)行去噪。

而在低秩聚類相關(guān)算法中，是將圖像建模成$\mathbf{Y}=\mathbf{X}?\mathbf{N}$，其中$\mathbf{Y}$同樣是由帶有噪聲的相似樣本組成而成的樣本矩陣。$\mathbf{X}$和$\mathbf{N}$分別為對(duì)應(yīng)的的無(wú)噪聲的樣本矩陣以及噪聲。我們給出的限制條件是$\mathbf{X}$是低秩矩陣。由于相似樣本組成的矩陣具備低秩性，噪聲不具備低秩性，因此通過(guò)低秩聚類可以實(shí)現(xiàn)圖像降噪的效果。

由此可見，從降噪的本質(zhì)上來(lái)將，稀疏表達(dá)的稀疏和低秩聚類中的聚類是具有一定相關(guān)性的。

為什么相似樣本具備低秩性呢？參考圖像降噪方法綜述中的說(shuō)法，當(dāng)我們拍攝一張大草原的圖片時(shí)，草原是由草組成的，而草是相似的，如果圖片上全是草，那么這張圖片實(shí)際包含的信息是很少的，因此可以理解為草是草的復(fù)制品，這就是低秩性的一個(gè)直觀理解。

那么接下來(lái)就開始具體將WNNM算法的實(shí)現(xiàn)，主要參考
非局部相似性去噪算法研究
Weighted Nuclear Norm Minimization with Application to Image Denoising

WNNM的全稱是Weighted Nuclear Norm Minimization，中文翻譯成加權(quán)核范數(shù)最小化方法，算法的流程如下圖所示：

如果在圖像上搜索相似patch的流程就不用贅述了，方法多種多樣，重要的是，假定我們要降噪的圖像塊為${\mathbf{P}}_{\mathbf{i}}$，由該圖像塊以及圖像上與其相似的圖像塊組成的矩陣為${\mathbf{Y}}_{\mathbf{i}}$，對(duì)應(yīng)的降噪后的矩陣為${\mathbf{X}}_{\mathbf{i}}$，低秩矩陣最小化可以用來(lái)求矩陣的解，于是我們得到目標(biāo)函數(shù)：$${\mathbf{X}}_i={\mathrm{argmin}}_{x_i}{\Vert {\mathbf{Y}}_i?{\mathbf{X}}_i\Vert }_F^2+\mathrm{rank}\left({\mathbf{X}}_i\right)$$但是，其中${\Vert \mathbf{X}\Vert }_F$是F范數(shù)，F范數(shù)的定義是矩陣$\mathbf{X}$各項(xiàng)元素的絕對(duì)值平方的總和，即$$\Vert \mathbf{X}{\Vert }_F\equiv \sqrt{\sum _{i=1}^m\sum _{j=1}^n{\left|x_{ij}\right|}^2}$$上式是一個(gè)非凸函數(shù)，求解過(guò)程將是一個(gè)NP問題，因此需要對(duì)該問題轉(zhuǎn)為凸優(yōu)化問題后再求解，為此，前輩提出了標(biāo)準(zhǔn)核范數(shù)最小化的求解方法，也就是NNM——本文要介紹的WNNM算法的前身，NNM的目標(biāo)函數(shù)如下：$${\mathbf{X}}_i={\mathrm{argmin}}_{X_i}{\Vert {\mathbf{Y}}_i?{\mathbf{X}}_i\Vert }_F^2+\lambda {\Vert {\mathbf{X}}_i\Vert }_{?}$$式中，$\lambda$是一個(gè)正數(shù)，${\Vert {\mathbf{X}}_i\Vert }_{?}$是核范數(shù)，核范數(shù)的定義為$$\Vert \mathbf{X}{\Vert }_{?}=\mathrm{tr}\left(\sqrt{{\mathbf{X}}^T\mathbf{X}}\right)=\mathrm{tr}(\mathbf{\Sigma })$$

求解方法是對(duì)${\mathbf{Y}}_{\mathbf{i}}$進(jìn)行奇異值分解為${\mathbf{Y}}_j=\mathbf{U}\mathbf{\Sigma }\mathbf{V}$，然后對(duì)奇異值進(jìn)行軟閾值收縮：$${\mathcal{S}}_{\lambda }(\mathbf{\Sigma }{)}_{ii}=\max \left({\mathbf{\Sigma }}_{ii}?\lambda ,0\right)$$其中，${\mathbf{\Sigma }}_{ii}$為奇異矩陣$\mathbf{\Sigma }$對(duì)角線元素，于是得到目標(biāo)函數(shù)的解：$${\mathbf{X}}_i=\mathbf{U}{\mathcal{S}}_{\lambda }\left(\mathbf{\Sigma }\right)\mathbf{V}$$在NNM算法中，是使用同一個(gè)$\lambda$值對(duì)所有奇異值進(jìn)行軟閾值收縮，這樣做沒有考慮到圖像的信息主要集中在數(shù)值較大的上這個(gè)特點(diǎn)，因此會(huì)導(dǎo)致圖像細(xì)節(jié)過(guò)度平滑而變得模糊，為了解決這個(gè)問題，于是就誕生了WNNM算法，使用不同的$\lambda$值對(duì)奇異值進(jìn)行軟閾值收縮，數(shù)值大的奇異值對(duì)應(yīng)數(shù)值小的$\lambda$。由此我們得到目標(biāo)函數(shù)：$${\hat{\mathbf{X}}}_i={\mathrm{argmin}}_{X_i}\frac{1}{{\sigma }_n^2}{\Vert {\mathbf{Y}}_i?{\mathbf{X}}_i\Vert }_F^2+{\Vert {\mathbf{X}}_i\Vert }_{\mathbf{w},?}$$其中${\sigma }_n^2$為噪聲方差，用于歸一化F范數(shù)，而其中$\mathbf{w}=\left[w_1,\dots ,w_n\right]$中的每一項(xiàng)都為非負(fù)數(shù)，對(duì)應(yīng)每一個(gè)奇異值，如下：$$w_i=c\sqrt{k}/\left({\sigma }_i\left(\mathbf{X}\right)+\epsilon \right)$$$其中，{\sigma }_i\left(\mathbf{X}\right)$為$\mathbf{X}$的第$i$奇異值，可以觀察到，當(dāng)奇異值越大時(shí)權(quán)重越小。$k$w為相似圖像patch的數(shù)量，$\epsilon$為防止被零正常的小參數(shù)。但是這里有個(gè)問題是，${\sigma }_i\left(\mathbf{X}\right)$是未知的呀，我們可以假設(shè)在初始時(shí)刻，噪聲能量是在各個(gè)特征上分布是均勻的，因此初始化${\sigma }_i\left(\mathbf{X}\right)$為：$${\sigma }_i\left(\mathbf{X}\right)=\sqrt{\max \left({\sigma }_i^2\left(\mathbf{Y}\right)?k{\sigma }_n^2,0\right)}$$最后同樣我們對(duì)${\mathbf{Y}}_{\mathbf{i}}$進(jìn)行奇異值分解為${\mathbf{Y}}_j=\mathbf{U}\mathbf{\Sigma }\mathbf{V}$，然后目標(biāo)函數(shù)的解為：$${\mathbf{X}}_i=\mathbf{U}{\mathcal{S}}_{\mathbf{w}}\left(\mathbf{\Sigma }\right)\mathbf{V}$$

2. matlab代碼

matlab代碼我就直接貼上鏈接好了 csjunxu/WNNM_CVPR2014 ，這份matlab代碼我也沒有細(xì)讀，僅僅跑了一下，感興趣的同學(xué)可以多畫時(shí)間研究研究。

3. 結(jié)論

低秩聚類不僅僅可以用在圖像降噪，在圖像分割、分類等方面也有廣泛的應(yīng)用。

論文中的圖像效果如下：
我測(cè)試的結(jié)果也差不多，相對(duì)BM3D其文理細(xì)節(jié)保留會(huì)更好一些，但是一些紋理密集的區(qū)域會(huì)出現(xiàn)一些artifact，至于好不好，就要看大家對(duì)圖片質(zhì)量的要求了，這篇文章寫的比較淺顯，有問題歡迎交流

此外，這里我寫一個(gè)各種算法的總結(jié)目錄圖像降噪算法——圖像降噪算法總結(jié)，對(duì)圖像降噪算法感興趣的同學(xué)歡迎參考

總結(jié)

以上是生活随笔為你收集整理的图像降噪算法——低秩聚类：WNNM算法的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。