arXiv reaDer
The Domain Shift Problem of Medical Image Segmentation and Vendor-Adaptation by Unet-GAN
 畳み込みニューラルネットワーク(CNN)、特にUnetは、医療画像セグメンテーションの強力な方法です。これまで、Unetは、特にトレーニングデータとテストデータが同じ分布を共有している(つまり、同じソースドメインからの)条件下で、多くの複雑な医療画像セグメンテーションタスクで最先端のパフォーマンスを実証してきました。ただし、臨床診療では、医療画像はさまざまなベンダーやセンターから取得されます。特定のソースドメインからトレーニングされたU-Netのパフォーマンスは、異なるターゲットドメイン(異なるベンダー、取得パラメーターなど)に転送されると、予期せず低下する場合があります。 U-Netを再トレーニングするために新しいドメインごとに大量の注釈を収集するのは、費用がかかり、退屈で、実際には不可能です。この作業では、この問題に対処するための一般的なフレームワークを提案しました。これは、(1)ベンダー適応用の不対生成的敵対ネットワーク(GAN)、および(2)オブジェクトセグメンテーション用のUnetで構成されます。提案されているUnet-GANアーキテクチャでは、GANはセグメンテーション固有の機能レベルでUnetから学習します。例として心臓シネMRIを使用し、3つの主要ベンダー(Philips、Siemens、およびGE)を3つのドメインとして使用しましたが、方法論は一般に医療画像のセグメンテーションに拡張できます。提案された方法は、ベンダー全体でセグメンテーション結果の大幅な改善を示しました。提案されているUnet-GANは、クロスベンダー医療画像セグメンテーション問題に対する注釈不要のソリューションを提供し、トレーニング済みの深層学習モデルを実際の臨床シナリオでのマルチセンターおよびマルチベンダー使用に拡張する可能性があります。
Convolutional neural network (CNN), in particular the Unet, is a powerful method for medical image segmentation. To date Unet has demonstrated state-of-art performance in many complex medical image segmentation tasks, especially under the condition when the training and testing data share the same distribution (i.e. come from the same source domain). However, in clinical practice, medical images are acquired from different vendors and centers. The performance of a U-Net trained from a particular source domain, when transferred to a different target domain (e.g. different vendor, acquisition parameter), can drop unexpectedly. Collecting a large amount of annotation from each new domain to retrain the U-Net is expensive, tedious, and practically impossible. In this work, we proposed a generic framework to address this problem, consisting of (1) an unpaired generative adversarial network (GAN) for vendor-adaptation, and (2) a Unet for object segmentation. In the proposed Unet-GAN architecture, GAN learns from Unet at the feature level that is segmentation-specific. We used cardiac cine MRI as the example, with three major vendors (Philips, Siemens, and GE) as three domains, while the methodology can be extended to medical images segmentation in general. The proposed method showed significant improvement of the segmentation results across vendors. The proposed Unet-GAN provides an annotation-free solution to the cross-vendor medical image segmentation problem, potentially extending a trained deep learning model to multi-center and multi-vendor use in real clinical scenario.
updated: Wed Oct 30 2019 05:36:21 GMT+0000 (UTC)
published: Wed Oct 30 2019 05:36:21 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト