利用GAN實(shí)現(xiàn)非成對的多模態(tài)MR圖像生成2019-10-28

原文:Unpaired Multi-contrast MR Image Synthesis Using Generative Adversarial Networks
論文來自2019MICCAI

摘要:

一些醫(yī)學(xué)圖像在某些情況下可能不容易獲得,因此從能得到的圖像出發(fā)生成不能得到的圖像有較大的醫(yī)學(xué)價(jià)值亮蛔。普通的GAN不能使用一個(gè)生成器和判別器生成多種與之相對應(yīng)的不同模態(tài)的圖像诵肛,因此對于生成多個(gè)模態(tài)的圖像需要多個(gè)模型紊册。針對這個(gè)問題假颇,本文提出了新的模型漂辐,使用Star-GAN來實(shí)現(xiàn)一到多的生成亚铁。本文引入了新的損失函數(shù)失息,它強(qiáng)制生成器生成高質(zhì)量的圖像,在視覺上更真實(shí)募舟,并且有很好的結(jié)構(gòu)相似性祠斧。在IXI數(shù)據(jù)集上學(xué)習(xí)所有可能的映射(T1,T2,PD,MRA),定性和定量都比較好拱礁。

介紹

因?yàn)樯疃葘W(xué)習(xí)的訓(xùn)練需要很多數(shù)據(jù)琢锋,但是標(biāo)注代價(jià)昂貴因此為了提高深度學(xué)習(xí)的表現(xiàn),通過生成一些圖像的方法來實(shí)現(xiàn)數(shù)據(jù)增廣是有研究價(jià)值的呢灶。傳統(tǒng)方法通過cycle-gan 吩蔑、c-gan 、wasserstein-gan或者pix2pix來實(shí)現(xiàn)一對一的生成填抬。我們使用star-gan和U-NET來實(shí)現(xiàn)一對多的生成烛芬。模型能夠以無監(jiān)督的形式訓(xùn)練,這樣能夠使生成器學(xué)習(xí)不同種模態(tài)的通用的幾何特征飒责。無監(jiān)督的方式也消除了成對數(shù)據(jù)的要求赘娄,因此對數(shù)據(jù)的限制較小。
在損失方面宏蛉,使用結(jié)構(gòu)相似性來約束小細(xì)節(jié)特征遣臼。除此之外還采用了‘學(xué)習(xí)感知圖像塊相似性’( Learned Perceptual Image Patch Similarity (LPIPS)不知道怎么翻譯,目前第一次看到拾并。)模型實(shí)現(xiàn)1輸入4輸出揍堰。

方法:

能夠?qū)崿F(xiàn)四種模態(tài)之間的轉(zhuǎn)化,輸入一個(gè)和一個(gè)目標(biāo)域能夠產(chǎn)生出相對的圖像嗅义。


Fig. 1. U-NET generator performs two synthesis: (i) generating a fake image given depth-wise concatenated real image and target contrast; (ii) reconstructing real image given fake image concatenated depth-wise with original contrast. Fake image is used to measure two losses: (i) Adversarial loss and (ii) contrast classification loss using PatchGAN discriminator. Reconstructed and real image are used to measure reconstruction loss to observe how close reconstructed image is to the real image in terms of structural (SSIM), perceptual (LPIPS) and global (L1) similarity

U-net用來實(shí)現(xiàn)兩種生成屏歹,一種是輸入一個(gè)域和另一個(gè)域的標(biāo)簽,通過深度級(jí)聯(lián)(depth-wise concatenat)然后生成另一個(gè)域的假的圖像之碗。另一中是輸入假的圖像和原始的標(biāo)簽生成由假圖重建的圖像蝙眶。第一步的生成用來計(jì)算對抗損失和分類損失。第二部分的生成中用來計(jì)算相似性的損失包括(L1范數(shù)褪那,DSSIM,LPIPS)
過程和star-gan一樣


star-gan

損失函數(shù)

對抗損失

本文使用正則化的帶有梯度懲罰的Wassersteing GAN (WGAN-GP)幽纷,能夠穩(wěn)定學(xué)習(xí),和增強(qiáng)生成圖像的質(zhì)量博敬。定義如下:



第一項(xiàng)損失是WGAN-GP損失友浸,第二項(xiàng)是正則化項(xiàng)



G(x,c)表示輸入x圖像和目標(biāo)域的標(biāo)簽生成的假圖像。

x' and x''是與x相近的相距很近的數(shù)據(jù)偏窝。D_是從第二層到最后一層判別器的輸出收恢。

這部分看的不是太明白武学。具體可能因?yàn)閃GAN沒有看過把。應(yīng)該原始論文中有派诬,需要后面完善劳淆。

分類損失

使生成器生成正確域的圖像




第一個(gè)式子使真圖的分類損失链沼,第二個(gè)是假圖的分類損失默赂。

生成損失

cycle損失

因?yàn)長1范數(shù)關(guān)注點(diǎn)在整張圖像而忽視了區(qū)塊(patch)之間的相似性。為此我們增加了兩項(xiàng)來強(qiáng)化約束小區(qū)域括勺,和結(jié)構(gòu)相似性缆八。
1)DSSIM損失:是SSIM損失的升級(jí)
2)LPIPS損失:https://arxiv.org/abs/1801.03924 還未看。


因此重建損失為
重建損失

總的損失為
判別器損失

生成器損失

網(wǎng)絡(luò)結(jié)構(gòu)

生成器是Unet結(jié)構(gòu)疾捍。判別器是基于PatchGAN的判別器奈辰。

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

使用LXI數(shù)據(jù)集 IXI dataset

定量分析:
定性分析:
Fig. 2. Synthesis of MRA, PD-weighted and T1-weighted images using a single T2- weighted image as input.
個(gè)人看法:

emmmm不應(yīng)該和star-gan比較把。應(yīng)該和介紹中提到的其他方法比較把乱豆。cycle奖恰,cgan之類的把。star-gan直接應(yīng)用在這上面肯定效果不是很好啊宛裕。這個(gè)比較個(gè)人覺得無意義瑟啃。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市揩尸,隨后出現(xiàn)的幾起案子蛹屿,更是在濱河造成了極大的恐慌,老刑警劉巖岩榆,帶你破解...
    沈念sama閱讀 219,366評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件错负,死亡現(xiàn)場離奇詭異,居然都是意外死亡勇边,警方通過查閱死者的電腦和手機(jī)犹撒,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,521評論 3 395
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來粒褒,“玉大人油航,你說我怎么就攤上這事』辰” “怎么了谊囚?”我有些...
    開封第一講書人閱讀 165,689評論 0 356
  • 文/不壞的土叔 我叫張陵,是天一觀的道長执赡。 經(jīng)常有香客問我镰踏,道長,這世上最難降的妖魔是什么沙合? 我笑而不...
    開封第一講書人閱讀 58,925評論 1 295
  • 正文 為了忘掉前任奠伪,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘绊率。我一直安慰自己谨敛,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,942評論 6 392
  • 文/花漫 我一把揭開白布滤否。 她就那樣靜靜地躺著脸狸,像睡著了一般。 火紅的嫁衣襯著肌膚如雪藐俺。 梳的紋絲不亂的頭發(fā)上炊甲,一...
    開封第一講書人閱讀 51,727評論 1 305
  • 那天,我揣著相機(jī)與錄音欲芹,去河邊找鬼卿啡。 笑死,一個(gè)胖子當(dāng)著我的面吹牛菱父,可吹牛的內(nèi)容都是我干的颈娜。 我是一名探鬼主播,決...
    沈念sama閱讀 40,447評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼浙宜,長吁一口氣:“原來是場噩夢啊……” “哼官辽!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起梆奈,我...
    開封第一講書人閱讀 39,349評論 0 276
  • 序言:老撾萬榮一對情侶失蹤野崇,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后亩钟,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體乓梨,經(jīng)...
    沈念sama閱讀 45,820評論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,990評論 3 337
  • 正文 我和宋清朗相戀三年清酥,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了扶镀。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,127評論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡焰轻,死狀恐怖臭觉,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情辱志,我是刑警寧澤蝠筑,帶...
    沈念sama閱讀 35,812評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站揩懒,受9級(jí)特大地震影響什乙,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜已球,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,471評論 3 331
  • 文/蒙蒙 一臣镣、第九天 我趴在偏房一處隱蔽的房頂上張望辅愿。 院中可真熱鬧,春花似錦忆某、人聲如沸点待。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,017評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽癞埠。三九已至,卻和暖如春棒坏,著一層夾襖步出監(jiān)牢的瞬間燕差,已是汗流浹背遭笋。 一陣腳步聲響...
    開封第一講書人閱讀 33,142評論 1 272
  • 我被黑心中介騙來泰國打工坝冕, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人瓦呼。 一個(gè)月前我還...
    沈念sama閱讀 48,388評論 3 373
  • 正文 我出身青樓喂窟,卻偏偏與公主長得像,于是被迫代替她去往敵國和親央串。 傳聞我的和親對象是個(gè)殘疾皇子磨澡,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,066評論 2 355

推薦閱讀更多精彩內(nèi)容