基于KNN的DSA圖像去噪及GPU的快速實(shí)現(xiàn)

王光磊，裴晨輝，苑　昊，王　斌，劉秀玲

(河北大學(xué) 電子信息工程學(xué)院 河北省數(shù)字醫(yī)療工程重點(diǎn)實(shí)驗(yàn)室，河北 保定 071002)

基于KNN的DSA圖像去噪及GPU的快速實(shí)現(xiàn)

王光磊，裴晨輝，苑昊，王斌，劉秀玲

(河北大學(xué) 電子信息工程學(xué)院 河北省數(shù)字醫(yī)療工程重點(diǎn)實(shí)驗(yàn)室，河北 保定 071002)

摘要:為快速地去除或減少DSA(Digital Subtraction Angiography)圖像的噪聲，對比評價(jià)KNN(K Nearest Neighbors)算法對高斯噪聲、泊松噪聲、斑點(diǎn)噪聲、椒鹽噪聲4種噪聲去除或減少的效果，幫助醫(yī)生快速準(zhǔn)確地為病人診斷疾病。提出的算法主要貢獻(xiàn)在于構(gòu)建了基于GPU(Graphics Processing Unit)的加速方法，使傳統(tǒng)圖像去噪的運(yùn)算速度得到大幅提升?；趫D像降質(zhì)、圖像還原過程建模，使用KNN算法對4種噪聲去除或減少，并對算法做并行化處理，利用GPU加速實(shí)現(xiàn)去噪的過程。通過實(shí)驗(yàn)得出，KNN算法能較好地去除或減少高斯噪聲、泊松噪聲來還原DSA圖像，使用CUDA(Compute Unified Device Architecture)編寫可在GPU上運(yùn)行的程序，利用GPU對1 024×1 024像素的24位深度的DSA圖像去噪，平均渲染幀率能達(dá)到190.53 f/s(幀/秒)，較傳統(tǒng)CPU(Central Processing Unit)串行，平均處理速度提高70.86倍。使用GPU加速能夠快速地處理數(shù)據(jù)量較大、計(jì)算密集的DSA噪聲圖像，實(shí)現(xiàn)有效并且快速的高斯噪聲去除，幫助醫(yī)生精、準(zhǔn)、快地診斷疾病。

關(guān)鍵詞:DSA；KNN；GPU；CUDA；圖像去噪

近幾年，心腦血管疾病發(fā)病率呈逐年上升趨勢，其危害已超過傳染性疾病，成為我國威脅居民健康最大類疾病。最近的一次統(tǒng)計(jì)數(shù)據(jù)顯示，中國心腦血管疾病死亡已占總死亡構(gòu)成的41%。在醫(yī)學(xué)界，對于心腦血管疾病能夠準(zhǔn)確診斷也是衛(wèi)生工作人員和醫(yī)療技術(shù)人員一直孜孜追求的。然而在各類疾病診斷中，腦血管疾病的診斷比較困難。

1980年，美國威斯康星大學(xué)Mistretta小組、亞利桑納大學(xué)Nadelman小組研制成功數(shù)字減影血管造影機(jī)，在血管疾病診斷方面得到了很好的應(yīng)用，尤其是心腦血管疾病的診斷。自此，醫(yī)學(xué)領(lǐng)域有了傷害性小、方便簡潔、安全實(shí)用，并且影像清楚分辨率高的血管造影系統(tǒng)。然而，DSA發(fā)展到今天仍存在一些問題，比如它在采集圖像數(shù)據(jù)的時(shí)候會受到一些干擾，從而會使圖像產(chǎn)生噪聲。改善圖像質(zhì)量可以通過添加濾光片、對近紅外光敏感的探測器、改良CCD感光元件等改進(jìn)前端設(shè)備的方法進(jìn)行預(yù)處理的調(diào)節(jié)。由于前端預(yù)處理的局限性，為了提高系統(tǒng)的魯棒性與可靠性，降低成本，需要在后端利用軟件進(jìn)行圖像去噪處理。然而，對于醫(yī)學(xué)成像又有其特點(diǎn)，數(shù)據(jù)量巨大，計(jì)算密集，還有對實(shí)時(shí)性的要求也很高，傳統(tǒng)的串行計(jì)算很難滿足這些要求。

近些年來，國內(nèi)外研究人員在圖像去噪方面做了許多探索與研究?，F(xiàn)如今已有的去噪方法大類幾十種，小類上百種，如直方圖均衡算法、經(jīng)典逆濾波法、基于Retinex的相關(guān)算法、小波及曲波變換方法、維納濾波、空間域的核回歸方法、非局部均值、基于參數(shù)估計(jì)的算法、雙邊濾波、基于概率框架的盲圖像復(fù)原算法等常見方法[1-4]。BUADES A，DELEDALLE C，COUPE P，GUO Y等人首先提出和應(yīng)用了非局部均值(Non-local Means， NLM)去噪算法[5-9]。時(shí)至今天，在NLM提出十年間里也得到了許多改進(jìn)，如DOWSON N，DARBON J，ORCHARD J，LIU Y L，WANG J，WANG Z M等人將算法改進(jìn)后提高了算法實(shí)現(xiàn)的速度[10-14]。CHEN Y，DONOHO D L，CHANG S G等人運(yùn)用小波閾值算法實(shí)現(xiàn)了圖像去噪[15-17]。ZHANG J等人將小波與雙邊濾波算法結(jié)合實(shí)現(xiàn)了醫(yī)學(xué)超聲圖像的去噪[18]。自2007年NVIDIA推出CUDA以來，研究人員才逐漸重視到并行計(jì)算在醫(yī)學(xué)領(lǐng)域存在著巨大潛力。美國通用電氣醫(yī)療集團(tuán)利用GPU實(shí)現(xiàn)了錐形束反投影與透視CT重建。西門子公司BATENBURG，SIJBERS等人利用GPU加速了CT重建速度，并將硬件成本壓縮了上百倍；STONE等人使用GPU加快高級MRI重建速度14.35倍。VERONICA GIDEN等人利用GPU依照場景圖對用于XIP的CUDA內(nèi)核管線進(jìn)行成功構(gòu)建，并提高了構(gòu)建速度[19]。WOLFGANG WEIN等人使用GPU成功實(shí)現(xiàn)了針對診斷成像和成像引導(dǎo)介入的自動CT超聲配準(zhǔn)。2014年德克薩斯大學(xué)由STEVE博士帶領(lǐng)的小組，將GPU應(yīng)用在癌癥治療領(lǐng)域，原本需要計(jì)算70個(gè)多小時(shí)才能夠完成的復(fù)雜質(zhì)子放射療法縮短到10 s，縮短時(shí)間的同時(shí)還提高了計(jì)算精度。GPU在醫(yī)學(xué)領(lǐng)域的應(yīng)用近十年間得到了空前的發(fā)展，為人類醫(yī)療事業(yè)做出了很大的貢獻(xiàn)，GPU在醫(yī)學(xué)方面的重要性也不言而喻。

因此，綜合考慮后，本文提出了一種針對DSA圖像去噪的GPU加速方案，并很好地應(yīng)用于KNN去噪算法。實(shí)驗(yàn)表明，使用GPU加速能夠快速處理數(shù)據(jù)量較大、計(jì)算密集的噪聲圖像，實(shí)現(xiàn)有效的噪聲去除。

1KNN算法及其實(shí)現(xiàn)

1.1DSA噪聲

DSA系統(tǒng)最主要的組成部分是控制處理機(jī)和快速圖像處理機(jī)，以及模擬數(shù)字轉(zhuǎn)換器、存儲器等電子器件，整個(gè)圖像的攝制、存儲、處理和傳遞都是用數(shù)字形式進(jìn)行的。所以，在這其中不可避免地會引入圖像的降質(zhì)、模糊以及噪聲的干擾，采集到的圖像信息及特征會衰減、被掩蓋，甚至出現(xiàn)模糊，嚴(yán)重影響了醫(yī)生對病人病情的診斷。其中最值得引起關(guān)注的是，在所有噪聲中高斯噪聲的比重最大，也就是說高斯噪聲對影像的影響最大。因此，若對高斯噪聲能夠較好地濾除或者減弱，對最終的成像將會有較大的意義，對醫(yī)生準(zhǔn)確診斷病人的病情幫助也是最大的。

1.2基于KNN算法的DSA圖像去噪

1.2.1圖像降質(zhì)

圖像增強(qiáng)的主要目的是要迎合人們的視覺感官，使人盡可能地從圖像中得到更多的信息或令視覺效果更好，具有藝術(shù)性。如圖像增強(qiáng)和圖像去噪主要是以預(yù)先確定的目標(biāo)來針對性地還原圖像。盡管兩者有相重疊的領(lǐng)域，但圖像增強(qiáng)主要是主觀處理，而圖像去噪原則大部分是客觀處理。圖像去噪是利用退化現(xiàn)象的某種先驗(yàn)知識來復(fù)原一幅帶噪聲的圖像，而不能夠使圖像有較大失真，最大程度地從客觀上進(jìn)行最真實(shí)的還原。因此，圖像去噪是面向一個(gè)退化的模型，而采用降質(zhì)過程的逆過程做出針對性的并且目的很明確的處理，來恢復(fù)圖像。

通過建模[20-21]，降質(zhì)過程可以看作一個(gè)退化函數(shù)和一個(gè)加性噪聲項(xiàng)，對一幅輸入圖像f(x,y)進(jìn)行處理，產(chǎn)生一幅退化后的圖像g(x,y)

g(x,y)=H[f(x,y)]+η(x,y)

(1)

給定函數(shù)g(x,y)和退化函數(shù)H以及加性噪聲項(xiàng)函數(shù)η(x,y)，從而圖像去噪的目的便成為了獲得一個(gè)預(yù)知的原圖像f^(x,y)。原則上，這種預(yù)知應(yīng)盡可能地接近原圖像，而且退化函數(shù)H、加性噪聲項(xiàng)函數(shù)η(x,y)所含信息越豐富，f^(x,y)，f(x,y)越相近。

g(x,y)=h(x,y)*f(x,y)+η(x,y)

(2)

式中：h(x,y)為H的空間表示?？臻g域的卷積和頻率的乘法組成了一個(gè)傅里葉變換對，所以可以用等價(jià)的頻率域來寫出前面的模型，如圖1所示。

圖1　降質(zhì)—去噪處理模型

假設(shè)H為線性、空間不變過程，便可以得到在空間域中表示的降質(zhì)圖像

G(u,v)=H(u,v)F(u,v)+N(u,v)

(3)

式中：用大寫字母表示空降質(zhì)間域中相應(yīng)項(xiàng)的傅里葉變換。由于線性、空間不變性，所以退化函數(shù)H可以由卷積來建模。在此次建模中，假設(shè)H是恒等算子，并只處理由噪聲造成的退化。

為了更好地觀察KNN算法對DSA圖像的去噪效果，分析算法的針對性、有效性、穩(wěn)定性以及評價(jià)本算法對去噪的成效，故實(shí)驗(yàn)將人為針對性地引入高斯噪聲、泊松噪聲、斑點(diǎn)噪聲、椒鹽噪聲4種噪聲來實(shí)驗(yàn)。這些噪聲都是空間噪聲，其值是隨機(jī)數(shù)，由概率密度函數(shù)p(z)或等價(jià)的積累分布函數(shù)F(z)表征，它們均由均值m和方差σ2決定。a，b均為常數(shù)。這4種噪聲的各參數(shù)如下：

1)高斯噪聲

概率密度函數(shù)

(4)

積累分布函數(shù)

(5)

均值和方差

m=a，σ2=b2

(6)

2)泊松噪聲

概率密度函數(shù)

(7)

積累分布函數(shù)

(8)

均值和方差

m=a，σ2=a

(9)

3)斑點(diǎn)噪聲

概率密度函數(shù)

(10)

積累分布函數(shù)

(11)

均值和方差

(12)

4)椒鹽噪聲

概率密度函數(shù)

(13)

積累分布函數(shù)

(14)

均值和方差

m=(0)Pp+k(1-Pp-Ps)+(2n-1)Ps

σ2=(0-m)2Pp+(k-m)2(1-Pp-Ps)+

(2n-1-m)2Ps

(15)

1.2.2KNN算法實(shí)現(xiàn)

針對一幅退化圖像的去噪，最簡單方法是在上文介紹的模型中忽略噪聲項(xiàng)，并形成如下形式的估計(jì)

(16)

式中：G(u,v)是降質(zhì)圖像的傅里葉變換，然后取F^(u,v)的傅里葉反變換得到圖像的相應(yīng)估計(jì)，來直接對噪聲圖像進(jìn)行逆濾波，考慮噪聲時(shí)，可以將估計(jì)表示為

(17)

KNN算法用來減少圖像噪聲也是基于高斯噪聲濾波器，但比其更復(fù)雜。假設(shè)讓u(x)作為最原始的噪聲圖像，KNNh,ru(x)是經(jīng)過KNN濾波器和參數(shù)h和r之后的結(jié)果。讓Ω(P)在空間上以一定的距離環(huán)繞像素點(diǎn)P，本文將它視為N×N大小的一系列像素，此時(shí)N=2M+1(自然數(shù)M=0，1，2，3…)，這樣P就是Ω(P)的中心像素，于是

(18)

式中：C(x)是正?；禂?shù)。由公式得知，周圍像素權(quán)重決定于h、r和Ω。r是一個(gè)傳統(tǒng)的高斯噪聲系數(shù)，由帶噪圖像決定，在用KNN算法處理時(shí)，r=N。h的選擇具有主觀性，它是用戶自己來選擇的，選擇的不同最后的結(jié)果也會不同，用戶可以根實(shí)驗(yàn)結(jié)果來調(diào)節(jié)h。對難以量化的h值，會根據(jù)圖像默認(rèn)一個(gè)值，最后的結(jié)果也是較為可觀的。Ω的大小取決于圖像的大小，為了達(dá)到一個(gè)良好的視覺效果，通常是5×5或7×7塊的像素塊兒。在本實(shí)驗(yàn)中，選擇了7×7塊的像素塊進(jìn)行實(shí)驗(yàn)，如圖2所示。

圖2　7×7像素塊

1.3GPU加速實(shí)現(xiàn)

1.3.1耗時(shí)分析

在KNN算法中，K近鄰搜索方法是暴力搜索法。本實(shí)驗(yàn)要處理的圖像是1 024×1 024像素點(diǎn)的RGB圖像，也就是說一幅圖像是一個(gè)1 024×1 024×3的彩色像素組，其中每個(gè)彩色像素是一個(gè)三值組，這三個(gè)值分別對應(yīng)一個(gè)特定空間位置處該RGB圖像的紅、綠、藍(lán)分量，如圖3所示。

圖3　RGB分量示意圖

RGB圖像可以看作是三色圖像的合成，把它們輸入顯示器對應(yīng)的三端時(shí)，會在屏幕上生成一幅人類可視的多彩圖像。其合成的三幅圖像分別稱作紅色、綠色、藍(lán)色分量圖像，每個(gè)分量的數(shù)據(jù)類型又是判斷三色分量值大小唯一的依據(jù)。本實(shí)驗(yàn)所采用的圖像是unit8類的，其灰度值為0～255。同時(shí)用來表示這些分量圖像像素值的比特?cái)?shù)決定了一幅圖RGB圖像的比特深度，所以本實(shí)驗(yàn)的RGB圖像的比特深度為24。因此，本實(shí)驗(yàn)的計(jì)算量非常大，傳統(tǒng)的串行計(jì)算模式難以滿足對速度的要求，為了提高計(jì)算效率，采用了GPU加速。

1.3.2GPU加速實(shí)現(xiàn)過程

在本實(shí)驗(yàn)中，為針對每個(gè)參考點(diǎn)的計(jì)算分配了一個(gè)線程，以發(fā)揮GPU優(yōu)勢。一個(gè)線程塊分配了1 024個(gè)線程，大于192個(gè)線程，這樣就可以在寄存器和共享內(nèi)存內(nèi)存取數(shù)據(jù)，1個(gè)時(shí)鐘周期的延遲也可以忽略掉了。無論CPU還是GPU在計(jì)算的時(shí)候主要是存取數(shù)據(jù)耗時(shí)，對于GPU而言，若一個(gè)線程塊有192個(gè)線程，正好可以在32個(gè)線程從啟動到結(jié)束的時(shí)候，其他5組32個(gè)線程會有一組準(zhǔn)備好數(shù)據(jù)，這樣就可以無延時(shí)的進(jìn)行計(jì)算，從而最大程度上利用了硬件資源。根據(jù)實(shí)驗(yàn)選取DSA圖像的大小正好分配1 024個(gè)線程塊，這樣一個(gè)SM(Streaming Multiprocessor)上能夠運(yùn)行多個(gè)線程塊，又線程數(shù)目大于了192個(gè)，這樣延遲被忽略掉，從而最大程度上利用共享內(nèi)存。參考點(diǎn)與線程對應(yīng)圖如圖4所示。

實(shí)驗(yàn)時(shí)，首先將數(shù)據(jù)傳輸?shù)紾PU中。然后再將一個(gè)SM里需要計(jì)算的線程的數(shù)據(jù)放進(jìn)共享內(nèi)存中，共享內(nèi)存再將運(yùn)行線程直接使用的數(shù)據(jù)放進(jìn)寄存器里面，來提高計(jì)算速度，最大程度挖掘GPU硬件性能。GPU端計(jì)算完畢后，再將計(jì)算結(jié)果傳回到CPU端，這樣就實(shí)現(xiàn)了GPU處理帶噪聲的DSA圖像。其實(shí)現(xiàn)過程圖如圖5所示。

2實(shí)驗(yàn)結(jié)果

2.1實(shí)驗(yàn)環(huán)境

硬件方面，本文選用英特爾公司出品的專門面向服務(wù)器和工作站的高端處理器Xeon E5-2680，同時(shí)，為了更好地使實(shí)驗(yàn)效果對比明顯，在原有的顯卡NVIDIA Quadro K5200的基礎(chǔ)上再加一塊NVIDIA Tesla K40c。

圖4　參考點(diǎn)與線程的對應(yīng)圖

圖5　GPU實(shí)現(xiàn)過程

軟件方面，系統(tǒng)選擇64位的Windows 7專業(yè)版，CUDA平臺選擇Visual Studio 2013 + CUDA 7.0這樣較新的組合搭配。環(huán)境的主要參數(shù)如表1。

表1實(shí)驗(yàn)環(huán)境參數(shù)

環(huán)境參數(shù)電腦型號戴爾PrecisionTower7910Tower操作系統(tǒng)Windows7專業(yè)版64位SP1(DirectX11)軟件平臺VisualStudio2013+CUDA7.0MATLABR2014a處理器英特爾Xeon(至強(qiáng))E5-2680v3@2.50GHz(X2)主板戴爾0215PR(英特爾Haswell-EDMI2)顯卡NVIDIAQuadroK5200(4Gbyte)加速卡NVIDIATeslaK40c(12Gbyte)內(nèi)存192Gbyte(三星DDR42133MHz)主硬盤LSI(512Gbyte)(X4)

2.2去噪效果

由于本實(shí)驗(yàn)在數(shù)據(jù)選取時(shí)選擇的DSA圖像較大，通常圖像的細(xì)節(jié)是醫(yī)生診斷疾病的關(guān)鍵，故本文將放大圖像的細(xì)節(jié)來做對比。截取部分如圖6所示。

圖6　DSA圖像細(xì)節(jié)

實(shí)驗(yàn)分別引進(jìn)了兩組高斯噪聲(m=0，σ2=0.05和σ2=0.03)、一組泊松噪聲、一組斑點(diǎn)噪聲(σ2=0.2)、一組椒鹽噪聲(d=0.03，d代表噪聲密度)。其5組實(shí)驗(yàn)結(jié)果如圖7所示。

從實(shí)驗(yàn)結(jié)果來看：KNN算法對選取的兩組帶高斯噪聲的DSA圖像具有較好去除效果，對泊松噪聲在細(xì)節(jié)方面去噪較好，對斑點(diǎn)噪聲幾乎無任何功效，對椒鹽噪聲也有一定的噪聲消除成效。

2.3加速效果

實(shí)驗(yàn)在對KNN去噪算法用CUDA實(shí)現(xiàn)后，做了一個(gè)對比實(shí)驗(yàn)，以展示算法用GPU運(yùn)算的優(yōu)勢。實(shí)驗(yàn)分別選取了5位病人的DSA圖像，每位病人選取2張，共10張，以準(zhǔn)確測算去噪速度。GPU和CPU對比實(shí)驗(yàn)數(shù)據(jù)如表2所示。

從5個(gè)病人的10組對比實(shí)驗(yàn)結(jié)果來看，GPU的處理速度明顯高于CPU的處理速度。其中，用GPU計(jì)算平均渲染幀率為190.53 f/s(幀/秒)，用CPU計(jì)算平均渲染幀率為2.69 f/s(幀/秒)，平均加速比為70.83，即應(yīng)用本文的GPU加速策略可以將計(jì)算速度提高70倍以上。因此，應(yīng)用本文的方法通過GPU的處理速度可以滿足醫(yī)學(xué)成像對實(shí)時(shí)性的要求，實(shí)現(xiàn)醫(yī)生對病人的實(shí)時(shí)監(jiān)測診斷。

e　椒鹽噪聲去噪前后圖7　5組DSA圖像去噪結(jié)果

比較項(xiàng)圖像1圖像2圖像3圖像4圖像5圖像編號se003se004se028se040se019se021se011se016se006se009圖像規(guī)格1024×1024GPU渲染幀率/(f·s-1)190.9190.8190.6189.7190.6190.3190.4190.7190.8190.5CPU渲染幀率/(f·s-1)2.72.62.72.72.82.72.72.72.62.7加速比70.7073.3870.5970.2668.0770.4870.5270.6373.3870.56

3結(jié)論

DSA圖像在采集過程中不可以避免地會引入一些噪聲，為解決此問題，本實(shí)驗(yàn)應(yīng)用了KNN算法來進(jìn)行去噪。但由于DSA圖像數(shù)據(jù)量大、計(jì)算密集，又考慮到實(shí)際情況，去噪的實(shí)時(shí)性也要好，即使對KNN算法進(jìn)行改進(jìn)仍很難滿足這些要求。于是可以將解決的重點(diǎn)放在處理速度上，考慮到KNN算法的特點(diǎn)，可并行程度較高，使用CUDA編寫可在GPU上運(yùn)行的程序，用GPU處理后，效果較為理想。

參考文獻(xiàn)：

[1]AI D， YANG J， FAN J， et al. Denoising filters evaluation for magnetic resonance images [J]. Optik-international journal for light and electron optics，2015(23)：3844-3850.

[2]SINGH C， AGGARWAL A. An efficient approach for image sequence denoising using Zernike moments-based nonlocal means approach[J]. Computers & electrical engineering，2015(11)：56-60.

[3]IFTIKHAR M A，JALIL A，RATHORE S，et al. An extended non-local means algorithm： application to brain MRI[J]. International journal of imaging systems and technology，2014(24)：293-305.

[4]KUANG Y，ZHANG L，ZHANG Y. An adaptive rank-sparsity K-SVD algorithm for image sequence denoising[J]. Pattern recognition letters，2014(45)：16-54.

[5]BUADES A，COLL B，MOREL J M. A review of image denoising algorithms，with a new one [J]. Multiscale model simul，2005，4(2)：490-530.

[6]BUADES A，COLL B，MOREL J M. Nonlocal image and movie denoising [J]. International journal of computer vision， 2008，76(2)：123-139.

[7]DELEDALLE C，DENIS L，TUPIN F. Iterative weighted maximum likelihood denoising with probabilistic patch based weights[J]. IEEE transactions on image processing，2009，18(12)：2661-2672.

[8]COUPE P，HELLIER P，KERVRANN C，et al. Nonlocal means-based speckle filtering for ultrasound Images [J].IEEE transactions on image processing，2009，18(10)：2221-2229.

[9]GUO Y，WANG Y，HOU T. Speckle filtering of ultrasonic images using a modified non local-based algorithm [J]. Biomedical signal processing and control，2011(6)：129-138.

[10]DOWSON N， SALVADO O. Hashed nonlocal means for rapid image filtering [J].IEEE transactions on pattern analysis and machine intelligence，2011，33(3)：485-499.

[11]DARBON J，CUNHA A，CHAN T F，et al. Fast nonlocal filtering applied to electron cryomicroscopy[C]//Proc. IEEE International Symposium on Biomedical Imaging： From Nano to Macro， 2008. [S.l.]：IEEE， 2008： 1331-1334.

[12]ORCHARD J，EBRAHIMI M，WONG A. Faster nonlocal-means image denoising using the FFT[J]. IEEE transactions on image processing，2008(6)：45-49.

[13]LIU Y L，WANG J， CHEN X，et al. A robust and fast non-local means algorithm for image denoising [J]. Journal of computer science and technology， 2008，23(2)：270-279.

[14]WANG Z M，ZHANG L. An adaptive fast nonlocal image

denoising algorithm[J]. Journal of image and graphics，2009，14(4)：669-675.

[15]CHEN Y，HAN C. Adaptive wave let threshold for image denoising[J]. Electronics letters，2005，41(10)： 586-587.

[16]DONOHO D L. Denoising by soft-thresholding [J]. IEEE transactions on information theory，1995，41(3)： 613-627.

[17]CHANG S G，YU B，VETTERLI M. Adaptive wave let thresholding for image denoising and compression[J]. IEEE transactions on image processing，2000，9(9)： 1532-1546.

[18]ZHANG J，WANG C，CHENG Y. Despeckling for medical ultrasound images based on wavelet and bilateral filter [J]. Journal of image and graphics，2014，19(1)：126-132.

[19]WOLFGANG W，SHELBY B，ALI K，et al. Automatic CT-ultrasound registration for diagnostic imaging and image-guided intervention[J]. Medical image analysis，2008(12)：577-585.

[20]FU B，LI W W，F(xiàn)U Y P. An image topic model for image denoising. [J]. Neurocomputing，2015(169)：119-123.

[21]HE N，WANGA J B，ZHANG L L，et al. An improved fractional-order differentiation model for image denoising. [J]. Signal processing，2015(112)：180-188.

責(zé)任編輯：時(shí)雯

DSA image denoising based on KNN and rapid implementation of GPU

WANG Guanglei，PEI Chenhui，YUAN Hao，WANG Bin，LIU Xiuling

(KeyLaboratoryofDigitalMedicalEngineeringofHebeiProvince，CollegeofElectronicandInformationEngineering，HebeiUniversity，HebeiBaoding071002，China)

Abstract:The purpose of this paper is to efficiently reduce or eliminate the noise of DSA (Digital Subtraction Angiography) image and comparatively evaluate the efficiency of noise reduction of including Gaussian noise， Poisson noise， Speckle noise and Salt and Pepper noise based on KNN (K Nearest Neighbors) algorithm. The results can be used to provide high quality image to doctors for quick and accurate disease diagnosis for the patient. The main contribution of the proposed paper constructs an acceleration structure using GPU， which can dramatically reduce the time cost of the traditional image denoising algorithm. The process modeling of image quality and image reconstruction is supplemented in KNN algorithm to reduce or eliminate the four kinds of noises. Additionally， the parallel processing based on GPU (Graphics Processing Unit) is executed to accelerate the denoising procedure. The experiments prove that Gaussian noise， Poisson noise can be removed or reduced effectively to reconstruct DSA image by KNN algorithm. The developed programming codes using CUDA (Compute Unified Device Architecture) works well in GPU to realize the average rendering frame rate 190.53 f/s for denoising of 1 024×1 024 DSA image with 24 bit depth. Compare with the traditional CPU (Central Processing Unit) serial， the average processing speed rate is increased by 70.86 times. Using GPU acceleration can quickly process DSA noise image with large amount of data and intensive computational mission. The results can provide accurate and fast diagnosis to doctors with the efficient Gaussian noise reduction.

Key words:DSA；KNN；GPU；CUDA；image denoising

中圖分類號：TN911.73

文獻(xiàn)標(biāo)志碼:A

DOI：10.16280/j.videoe.2016.06.003

基金項(xiàng)目：國家自然科學(xué)基金項(xiàng)目(61473112;61203160);河北省教育廳項(xiàng)目(QN2014166)；河北省自然科學(xué)項(xiàng)目(F2015201196)

收稿日期：2015-12-10

文獻(xiàn)引用格式：王光磊，裴晨輝，苑昊，等.基于KNN的DSA圖像去噪及GPU的快速實(shí)現(xiàn)[J].電視技術(shù)，2016，40(6)：10-16.

WANG G L，PEI C H，YUAN H, et al.DSA image denoising based on KNN and rapid implementation of GPU[J].Video engineering，2016,40(6)：10-16.