基於自然圖像先驗的貝葉斯超分重建

   基於自然圖像先驗的貝葉斯超分重建

一:摘要:

一般方法:用貝葉斯模型來描述問題,然後得到最大後驗概率解,就成爲一個正則化迴歸的問題,但是這並不能充分利用概率模型的優勢。另外,目前的貝葉斯SR方法用後驗均值估計,用比較簡單的後驗均值模型以保證效率。本文提出一種貝葉斯SR方法,用高階MRF模型作爲自然圖像的先驗。最後使用MMSE準則來估計SR圖像。使用MCMC採樣方法來得到最後的MMSE方法的解。

與別的文章的不同

(1)是一個完整的貝葉斯方法,可以HR圖像的先驗知識,或者其他不確定性,例如噪聲的級別。

(2)高階MRF模型用來模型自然圖像統計先驗,也有助於採樣。

(3)MMSE準則代替MAP準則,MAP對局部極大值更加敏感。尤其當用到重尾先驗,容易陷入局部最大值。

二:最近工作

三:自然圖像統計的先驗模型

自然圖像具有它獨有的特徵,和噪聲圖像不一樣。例如局部相關性、邊界、自相似性。自然圖像統計:(1)自然圖像加以梯度濾波後有重尾分佈。可以理解爲:自然圖像是局部平滑的,因此邊際分佈比高斯分佈下降的快。自然圖像有許多的特徵:邊界:導致比高斯分佈嚴重的重尾分佈。除了這些:例如尺度不變性:自然圖像在不同尺度下,重尾分佈是相似的。還有局部像素的相關性。

(1)高斯先驗:梯度圖像的高斯模型

(2)圖像滿足一定的平滑特性。過度平滑圖像

(3)拉普拉斯先驗

 

保持圖像的不連續性

(4)超拉普拉斯先驗

alpha控制圖像的稀疏性。

 

自然圖像的概率模型可以統稱爲MRF。在MRF中,自然圖像x的概率f.)輸入爲爲一個clique,輸出這個clique的能量或者勢。Z是變分函數使得概率之和等於1.成對MPF是典型的MRF:每個像素是一個結點。每個像素和它鄰域像素連接。

前面所說的都是成對MRF,因爲每個clique都只有兩個鄰域像素,因爲是求一階梯度。引入高階MRF:高階MRF的挑戰是勢函數比較難定義。因爲cliquesize變大。引入FOE作爲勢函數。它是低維空間的幾個函數的乘積:

 

KlL個梯度濾波器,可以從數據中學習得到。比之前定義好的梯度濾波器的好處:學習得到的濾波器更能得到圖像的統計特徵。Clique實際上是在c點被kl覆蓋的一個patchMRF模型的一個挑戰是歸一化項Z。方法參考文獻中對比差異和分數匹配。

 

四、基於自然統計特性的貝葉斯圖像超分

傳統的用正則化項的SR方法不能發揮概率模型的優勢,面臨非凸優化問題。這邊提出一個貝葉斯SR方法,從後驗使用MMSE方法估計HR圖像,使用比較好的通過自然圖像學習的模型。

算法框架:首先從訓練自然圖像中用MRF模型學習一個分佈,用GSM模型作爲它的勢函數。學習到的分佈作爲被估計的SR的先驗。然後HR圖像通過後驗採樣用MMSE標準得到。

由貝葉斯定理:

包含條件概率或者叫似然概率和先驗px)。

A.貝葉斯SR構想

(1)條件似然概率:

(2)圖像的先驗模型:FOE模型是一個高階MRF模型,可以捕捉自然圖像的統計信息通過從數據中學習高階濾波器。

因此採用高階FOE模型作爲待估計HR的先驗。

 

勢函數是高斯尺度混合模型

採用這個模型的兩點好處:(1)方便獲取自然圖像統計信息。(2)便於採樣。

是第l個濾波器的第j個成分的歸一化係數。Sj是第j個高斯成分的尺度參數。

裏面包含的參數:可以通過參考文獻[25]/[32]訓練數據學習得到。

其實是一個零均值的一維高斯混合模型,更因爲先驗模型是高斯混合,可以方便的採樣。

(3)超參數的先驗

把噪聲級別也看成是隨機變量,把它和其他隨機變量一起估計。使用的是gamma分佈:

 

使用gamma分佈的原因是gamma分佈是高斯分佈的共軛先驗。

(4)後驗

 

使用貝葉斯後驗平均來估計HR圖像。我們通過從後驗分佈中採樣在估計。

 

B.貝葉斯SR方法

直接從後驗分佈中採樣是比較難得。加入輔助變量Z,是一個m*L的矩陣,m是是圖像的維數(變成向量)。Z包含的圖像x每個位置每個專家的GSM模型的指標變量。

 

Zcl=j

 

使用MCMC方法進行採樣。

(1)採樣Z

 

(2)採樣T

是一個gamma分佈

(3)採樣X

(4)

 

由於協方差矩陣很大,採樣比較苦難。將x分解:

第一個分量u:

 

第2個分量V:先採樣

上述採樣方法都使用參考文章:Gaussian sampling by local perturbationsUv變量的求解使用共軛梯度法。

得到採樣

 

估計噪聲

 

或者:

 

 

實現細節:

H是一個高斯濾波器,當放大因子爲3時,標準差爲2。尺寸爲7.濾波在頻域進行,通過FFT完成。

8個專家相乘得到勢函數f(。),L=8.

GSM模型J=15來作爲專家函數。15個成分共享一個基方差參數

每一個有一個尺度參數

尺寸爲3*3.GSM從參考文獻[29]用對比散度算法從訓練圖像中學習得到。也就是說這邊高斯模型的方差和均值都是定的,只是係數需要從訓練圖像中學習。

下圖爲學習到的濾波器以及8個專家(具有重尾分佈)和對應的係數。

 

MCMC中比較重要的一點是收斂條件:我們令M=100N=20.如果是噪聲圖像M=200.

 

實驗和結果:

放大因子爲3.BINNFastScSRproposed

放大因子爲4,和KRR(通過基於塊的稀疏核迴歸得到HRLR的映射)、

變分貝葉斯VBSR(用一個簡單的TV模型,通過VB近似,用後驗平均估計HR圖像)。這個方法中,低通濾波器的矩陣是確定的,只能處理小圖像。

ScSR方法相比較,該方法不需要知道噪聲級別,而且能推斷出噪聲級別。

 

總結:

SR圖像是用MMSE標準估計而不是傳統的MAP估計的。是後驗平均,而不是最大後驗。

估計HR圖像,我們利用近似MMSE估計的有效採樣。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章