论文标题

贝叶斯有条件gan用于MRI大脑图像合成

Bayesian Conditional GAN for MRI Brain Image Synthesis

论文作者

Zhao, Gengyan, Meyerand, Mary E., Birn, Rasmus M.

论文摘要

作为医学成像中的一种强大技术,图像合成被广泛用于诸如denoing,超级分辨率和模态转换等应用中。最近,深度神经网络的复兴在医学成像领域取得了巨大进步。尽管已经提出了许多基于深层倾斜的模型来提高图像合成的准确性,但对医疗应用非常重要的模型不确定性的评估一直是缺失的部分。在这项工作中,我们建议使用具有混凝土掉落的贝叶斯条件生成对抗网络(GAN)来提高图像合成精度。同时,整个管道中涉及一种不确定性校准方法,以使贝叶斯网络产生的不确定性可解释。该方法用T1W至T2W MR图像翻译验证,具有102名受试者的脑瘤数据集。与传统的贝叶斯神经网络(带有蒙特卡洛辍学)相比,该方法的结果达到了较低的RMSE,p值为0.0186。还说明了通过不确定性重新校准方法改善生成的不确定性的校准。

As a powerful technique in medical imaging, image synthesis is widely used in applications such as denoising, super resolution and modality transformation etc. Recently, the revival of deep neural networks made immense progress in the field of medical imaging. Although many deep leaning based models have been proposed to improve the image synthesis accuracy, the evaluation of the model uncertainty, which is highly important for medical applications, has been a missing part. In this work, we propose to use Bayesian conditional generative adversarial network (GAN) with concrete dropout to improve image synthesis accuracy. Meanwhile, an uncertainty calibration approach is involved in the whole pipeline to make the uncertainty generated by Bayesian network interpretable. The method is validated with the T1w to T2w MR image translation with a brain tumor dataset of 102 subjects. Compared with the conventional Bayesian neural network with Monte Carlo dropout, results of the proposed method reach a significant lower RMSE with a p-value of 0.0186. Improvement of the calibration of the generated uncertainty by the uncertainty recalibration method is also illustrated.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源