arXiv reaDer
心臓セグメンテーションのための変分近似による教師なしドメイン適応
Unsupervised Domain Adaptation with Variational Approximation for Cardiac Segmentation
教師なしドメイン適応は、医療画像のセグメンテーションに役立ちます。特に、ターゲット画像のグラウンドトゥルースが利用できない場合、ドメイン適応は、他のモダリティからの既存のラベル付き画像を利用することにより、ターゲット固有のモデルをトレーニングできます。報告された作品のほとんどは、ソースドメインとターゲットドメインの両方の画像を共通の潜在的特徴空間にマッピングし、敵対的なトレーニングで暗黙的に、または不一致メトリックを直接最小化することで明示的に不一致を減らしました。この作業では、両方のドメインの潜在的な特徴が、画像に与えられた条件付き分布がガウス分布である、共通のパラメーター化された変分形式に向かって駆動される新しいフレームワークを提案します。これは、変分オートエンコーダー(VAE)とこの変分近似の正則化に基づく2つのネットワークによって実現されます。両方のVAEは、それぞれ1つのドメインに対応しており、セグメンテーションモジュールが含まれています。このモジュールでは、ソースセグメンテーションが教師ありの方法でトレーニングされ、ターゲットのセグメンテーションは教師なしでトレーニングされます。 2つの心臓セグメンテーションタスク、すなわち、クロスモダリティ(CTおよびMR)全心臓セグメンテーションとクロスシーケンス心臓MRセグメンテーションを使用して、提案されたドメイン適応方法を検証しました。結果は、提案された方法が2つの最先端のアプローチと比較してより良い精度を達成し、心臓セグメンテーションの優れた可能性を示したことを示しています。さらに、提案された明示的な正則化は、ドメイン間の分布ギャップを狭めるのに効果的かつ効率的であることが示されました。これは、教師なしドメインの適応に役立ちます。コードとデータはhttps://zmiclab.github.io/projects.htmlからリリースされています。
Unsupervised domain adaptation is useful in medical image segmentation. Particularly, when ground truths of the target images are not available, domain adaptation can train a target-specific model by utilizing the existing labeled images from other modalities. Most of the reported works mapped images of both the source and target domains into a common latent feature space, and then reduced their discrepancy either implicitly with adversarial training or explicitly by directly minimizing a discrepancy metric. In this work, we propose a new framework, where the latent features of both domains are driven towards a common and parameterized variational form, whose conditional distribution given the image is Gaussian. This is achieved by two networks based on variational auto-encoders (VAEs) and a regularization for this variational approximation. Both of the VAEs, each for one domain, contain a segmentation module, where the source segmentation is trained in a supervised manner, while the target one is trained unsupervisedly. We validated the proposed domain adaptation method using two cardiac segmentation tasks, i.e., the cross-modality (CT and MR) whole heart segmentation and the cross-sequence cardiac MR segmentation. Results show that the proposed method achieved better accuracies compared to two state-of-the-art approaches and demonstrated good potential for cardiac segmentation. Furthermore, the proposed explicit regularization was shown to be effective and efficient in narrowing down the distribution gap between domains, which is useful for unsupervised domain adaptation. Our code and data has been released via https://zmiclab.github.io/projects.html.
updated: Wed Jun 16 2021 13:00:39 GMT+0000 (UTC)
published: Wed Jun 16 2021 13:00:39 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト