Medical Image Synthesis with Context-Aware Generative Adversarial Networks

摘要

計(jì)算機(jī)斷層掃描(CT)對(duì)于各種臨床應(yīng)用至關(guān)重要,例如放射治療計(jì)劃以及PET衰減校正拓挥。但是盖灸,CT在采集過(guò)程中會(huì)暴露放射線徙垫,這可能對(duì)患者造成副作用。與CT相比,磁共振成像(MRI)更安全治拿,并且不涉及任何輻射。因此,近來(lái),對(duì)于放射治療計(jì)劃的情況,研究人員被極大地動(dòng)機(jī)從同一對(duì)象的其對(duì)應(yīng)的MR圖像估計(jì)CT圖像。在本文中,我們提出了一種數(shù)據(jù)驅(qū)動(dòng)的方法來(lái)解決這一具有挑戰(zhàn)性的問(wèn)題凑队。特別是款青,通過(guò)訓(xùn)練性的卷積網(wǎng)絡(luò)來(lái)生成給定MR圖像的CT渠牲。為了更好地建模從MRI到CT的非線性關(guān)系并生成更逼真的圖像,我們建議使用對(duì)抗訓(xùn)練策略和圖像梯度差損失函數(shù)敲长。我們進(jìn)一步將AutoContextModel應(yīng)用于實(shí)現(xiàn)上下文感知的生成對(duì)抗網(wǎng)絡(luò)。實(shí)驗(yàn)結(jié)果表明,我們的方法從MRI圖像預(yù)測(cè)CT圖像時(shí)準(zhǔn)確,可靠越败,并且優(yōu)于目前的三種最新方法半哟。

1 簡(jiǎn)介

CT圖像廣泛地運(yùn)用在診斷和放療,在癌癥放射治療中,CT圖像提供單位制樟插,這對(duì)于治療計(jì)劃中的劑量計(jì)算至關(guān)重要负甸。此外,在流行的PET-CT掃描儀中悠就,CT圖像對(duì)于正電子發(fā)射斷層掃描(PET)的衰減校正也非常重要稿静。

用2D卷積神經(jīng)網(wǎng)絡(luò)去運(yùn)行3D醫(yī)學(xué)圖像時(shí)不合適的,因?yàn)?DCNN考慮到圖像是切片的母剥,所以有可能通過(guò)切片會(huì)產(chǎn)生不連續(xù)的預(yù)測(cè)憔披。為了解決這個(gè)問(wèn)題桐早,Ji提出3DCNN。

通常焚刺,假設(shè)數(shù)據(jù)是從高斯分布中提取的氓润,則將L2距離用作損失函數(shù)。在多峰分布的情況下這可能會(huì)出現(xiàn)問(wèn)題瞪讼,并且往往會(huì)在輸出圖像中產(chǎn)生模糊的結(jié)果

本文提出通過(guò)3D全卷積神經(jīng)網(wǎng)絡(luò)(FCN)來(lái)學(xué)習(xí)從MR到CT圖像的非線性映射,這是傳統(tǒng)CNN的一種變體。與CNN相比曾撤,F(xiàn)CN生成結(jié)構(gòu)化輸出寄锐,可以更好地保留預(yù)測(cè)信息中的鄰域信息 CT圖像怠褐。 此3D FCN在生成對(duì)抗框架中用作生成器磷杏,在該框架中选泻,對(duì)抗損失項(xiàng)是常規(guī)重建誤差的附加項(xiàng),目的是生成更真實(shí)的CT數(shù)據(jù)忿族。 網(wǎng)絡(luò)是在補(bǔ)丁程序上訓(xùn)練的锣笨,這使得其視圖僅限于補(bǔ)丁程序本身,因此無(wú)法提供遠(yuǎn)程信息道批。 我們?cè)诿總€(gè)階段都使用GAN框架進(jìn)行訓(xùn)練的Auto-ContextModel(ACM)使它能夠感知上下文错英。 據(jù)我們所知,這是GAN框架在合成醫(yī)學(xué)圖像生成領(lǐng)域的首次應(yīng)用隆豹。

該方法在兩個(gè)真實(shí)的CT / MR數(shù)據(jù)集上進(jìn)行了評(píng)估椭岩。 實(shí)驗(yàn)結(jié)果表明,我們的方法可以有效地根據(jù)MR圖像預(yù)測(cè)CT圖像璃赡,并且在比較下也優(yōu)于三種最新方法判哥。

2 方法

我們提出了一種由對(duì)抗性網(wǎng)絡(luò)組成的對(duì)抗性網(wǎng)絡(luò),其中完全卷積的網(wǎng)絡(luò)構(gòu)成了生成器碉考。 首先塌计,我們提出一種基本的3D FCN結(jié)構(gòu),以根據(jù)MRI圖像估算CT侯谁。 3D操作可以更好地建模3D空間信息锌仅,從而可以解決2D CNN中可能存在的跨切片不連續(xù)性問(wèn)題。 其次墙贱,我們?cè)谠O(shè)計(jì)的網(wǎng)絡(luò)中采用對(duì)抗訓(xùn)練策略热芹,其中一個(gè)附加的鑒別器網(wǎng)絡(luò)可以促使發(fā)電機(jī)的輸出盡可能看起來(lái)像真實(shí)的CT。 我們將圖像梯度差項(xiàng)添加到生成器的損失函數(shù)中惨撇,以保持生成的CT的清晰度伊脓。 最后,我們使用自動(dòng)上下文模型來(lái)迭代定義生成器的輸出串纺。

例如丽旅,?Unsupervisedrepresentationlearningwith deep convolutional generative adversarial networks. 中的AlecRadford能夠通過(guò)使用FCN來(lái)獲得非常逼真的圖像椰棘,而無(wú)需使用最大池化,并且在G和D上的不同層之間進(jìn)行批歸一化榄笙,其方式與我們預(yù)想的類似邪狞。


合成圖片的GAN架構(gòu)

該網(wǎng)絡(luò)輸入MRI圖片,并且嘗試去生成對(duì)應(yīng)的CT圖像茅撞。它具有8個(gè)階段帆卓,分別包含卷積,批處理規(guī)范化和ReLu運(yùn)算米丘,分別具有過(guò)濾器32剑令、32、32拄查、64忠售、64簿姨、64溶推、32否灾、32的數(shù)量。最后一層僅包含32個(gè)卷積濾波器稍算,其輸出被視為估算的CT典尾。 最后,鑒別器是一種典型的CNN架構(gòu)糊探,包括三個(gè)階段的卷積+批處理規(guī)范化+ ReLu +最大池化以及具有三個(gè)完全連接層的卷積的組合钾埂,其中前兩個(gè)使用ReLu作為激活函數(shù),最后一個(gè)使用Sigmoid科平,其輸出表示從輸入數(shù)據(jù)中提取輸入數(shù)據(jù)的可能性 真實(shí)CT的分布褥紫。 濾波器的大小為5×5×5,卷積層的濾波器數(shù)目為32匠抗、64故源、128和256,完全連接層中的輸出節(jié)點(diǎn)數(shù)目為512汞贸、128和1。

2.2自動(dòng)上下文模型(ACM)進(jìn)行優(yōu)化

在這項(xiàng)工作中印机,我們證明了ACM也可以成功地應(yīng)用于回歸任務(wù)矢腻。由于從先前估計(jì)的CT中提取了上下文特征,因此它們可以對(duì)僅使用基于補(bǔ)丁的特征無(wú)法獲得的信息進(jìn)行編碼射赛。 在我們的工作中多柑,我們不使用多個(gè)生成器,而是使用生成器網(wǎng)絡(luò)楣责,并且也沒(méi)有連接概率圖竣灌,而是連接了先前生成器網(wǎng)絡(luò)的輸出聂沙。 具體來(lái)說(shuō),我們迭代訓(xùn)練幾個(gè)以MRI片為輸入并估計(jì)相應(yīng)CT片的GAN初嘹。 這些補(bǔ)片在MRI補(bǔ)片中串聯(lián)為第二個(gè)通道及汉,并且在訓(xùn)練下一個(gè)GAN時(shí)將這些新數(shù)據(jù)用作輸入。 圖3展示了該方案屯烦。


提出的ACM應(yīng)用于GAN的架構(gòu)


3 數(shù)據(jù)



數(shù)據(jù)詳情

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

我們使用與訓(xùn)練樣本相同的中心點(diǎn)隨機(jī)提取大小為32×32×32的MRI斑塊坷随,并具有相應(yīng)的大小為16×16×16的CT。 使用Adam最優(yōu)化器對(duì)網(wǎng)絡(luò)進(jìn)行訓(xùn)練驻龟,其凈利率為10-6温眉,β1= 0.5(在[15]中建議),最小批量大小為10翁狐。生成器使用λ1= 0.5类溢,λ2=λ3= 1進(jìn)行訓(xùn)練。 該代碼使用TensorFlow庫(kù)實(shí)現(xiàn)露懒,接受后將公開(kāi)發(fā)布闯冷。為了證明預(yù)測(cè)準(zhǔn)確性的建議方法的優(yōu)勢(shì),我們將其與三種廣泛使用的方法進(jìn)行比較:基于Anatlas的方法隐锭,基于稀疏表示的方法以及具有自動(dòng)上下文模型的結(jié)構(gòu)化隨機(jī)森林窃躲。 在Estimating CT Image from MRI Data Using 3D Fully Convolutional Networks - 道客巴巴中聲明。 而且所有實(shí)驗(yàn)都是一勞永逸的钦睡。

為了比較不同方法的性能蒂窒,我們利用平均絕對(duì)誤差(MAE)和峰值信噪比(PSNR)作為度量。

5 結(jié)論

我們已經(jīng)開(kāi)發(fā)了3D GAN模型荞怒,可通過(guò)直接將MR圖像塊作為輸入而將CT塊作為輸出來(lái)從MRI圖像估計(jì)CT圖像洒琢。 通過(guò)使用ACM可以提高性能,因?yàn)樵谟?xùn)練過(guò)程中會(huì)有效地?cái)U(kuò)大GAN的上下文褐桌,這使其具有上下文意識(shí)衰抑。 我們已經(jīng)將該模型用于從其相應(yīng)的MR圖像中預(yù)測(cè)CT圖像,其中實(shí)驗(yàn)證明了我們提出的方法可以明顯優(yōu)于三種最新方法荧嵌,從而表明其在回歸任務(wù)中的適用性呛踊。 盡管我們僅考慮了CT圖像預(yù)測(cè)的任務(wù),但是我們提出的模型也可以應(yīng)用于涉及醫(yī)學(xué)圖像分析中生成過(guò)程的其他相關(guān)任務(wù)啦撮,例如超分辨率谭网,圖像去噪等。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末赃春,一起剝皮案震驚了整個(gè)濱河市愉择,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖锥涕,帶你破解...
    沈念sama閱讀 221,635評(píng)論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件衷戈,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡层坠,警方通過(guò)查閱死者的電腦和手機(jī)殖妇,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,543評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門(mén),熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)窿春,“玉大人拉一,你說(shuō)我怎么就攤上這事【善颍” “怎么了蔚润?”我有些...
    開(kāi)封第一講書(shū)人閱讀 168,083評(píng)論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)尺栖。 經(jīng)常有香客問(wèn)我嫡纠,道長(zhǎng),這世上最難降的妖魔是什么延赌? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 59,640評(píng)論 1 296
  • 正文 為了忘掉前任除盏,我火速辦了婚禮,結(jié)果婚禮上挫以,老公的妹妹穿的比我還像新娘者蠕。我一直安慰自己,他們只是感情好掐松,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,640評(píng)論 6 397
  • 文/花漫 我一把揭開(kāi)白布踱侣。 她就那樣靜靜地躺著,像睡著了一般大磺。 火紅的嫁衣襯著肌膚如雪抡句。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 52,262評(píng)論 1 308
  • 那天杠愧,我揣著相機(jī)與錄音待榔,去河邊找鬼。 笑死流济,一個(gè)胖子當(dāng)著我的面吹牛锐锣,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播绳瘟,決...
    沈念sama閱讀 40,833評(píng)論 3 421
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼刺下,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了稽荧?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書(shū)人閱讀 39,736評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎姨丈,沒(méi)想到半個(gè)月后畅卓,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,280評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡蟋恬,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,369評(píng)論 3 340
  • 正文 我和宋清朗相戀三年翁潘,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片歼争。...
    茶點(diǎn)故事閱讀 40,503評(píng)論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡拜马,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出沐绒,到底是詐尸還是另有隱情俩莽,我是刑警寧澤,帶...
    沈念sama閱讀 36,185評(píng)論 5 350
  • 正文 年R本政府宣布乔遮,位于F島的核電站扮超,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏蹋肮。R本人自食惡果不足惜出刷,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,870評(píng)論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望坯辩。 院中可真熱鬧馁龟,春花似錦、人聲如沸漆魔。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,340評(píng)論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)有送。三九已至淌喻,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間雀摘,已是汗流浹背裸删。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,460評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留阵赠,地道東北人涯塔。 一個(gè)月前我還...
    沈念sama閱讀 48,909評(píng)論 3 376
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像清蚀,于是被迫代替她去往敵國(guó)和親匕荸。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,512評(píng)論 2 359