arXiv reaDer
クロスモダリティ腹部多臓器セグメンテーションのためのソースフリーの教師なしドメイン適応
Source-free unsupervised domain adaptation for cross-modality abdominal multi-organ segmentation
腹部の多臓器セグメンテーションのために、ソースラベル付きCTデータセットからターゲットラベルなしMRデータセットに学習した知識を転送するためのドメイン適応を達成することは価値があります。一方、ターゲットデータセットの高い注釈コストを回避し、ソースデータセットのプライバシーを保護することが非常に望ましいです。したがって、ソースデータセットにアクセスせずにクロスモダリティ腹部多臓器セグメンテーションのための効果的なソースフリー教師なしドメイン適応方法を提案します。提案されたフレームワークのプロセスには、2つの段階があります。最初の段階では、特徴マップの統計損失を使用して、トップセグメンテーションネットワーク内のソースとターゲットの特徴の分布を調整し、エントロピー最小化損失を使用して、信頼性の高いセグメンテーションを促進します。トップセグメンテーションネットワークから出力された疑似ラベルは、ソースのような画像を生成するためにスタイル補正ネットワークをガイドするために使用されます。中央のセグメンテーションネットワークから出力された疑似ラベルは、目的のモデル(下部のセグメンテーションネットワーク)の学習を監視するために使用されます。第2段階では、循環学習とピクセル適応マスクの改良を使用して、目的のモデルのパフォーマンスをさらに向上させます。このアプローチにより、肝臓、右腎臓、左腎臓、および脾臓のセグメンテーションで、それぞれ0.884、0.891、0.864、および0.911のダイス類似度係数で満足のいくパフォーマンスを達成します。さらに、提案されたアプローチは、ターゲット注釈データが存在する状況に簡単に拡張できます。パフォーマンスは、平均ダイス類似度係数が0.888から0.922に向上し、教師あり学習(0.929)に近く、ラベル付けされたMRボリュームは1つだけです。
It is valuable to achieve domain adaptation to transfer the learned knowledge from the source labeled CT dataset to the target unlabeled MR dataset for abdominal multi-organ segmentation. Meanwhile, it is highly desirable to avoid high annotation cost of target dataset and protect privacy of source dataset. Therefore, we propose an effective source-free unsupervised domain adaptation method for cross-modality abdominal multi-organ segmentation without accessing the source dataset. The process of the proposed framework includes two stages. At the first stage, the feature map statistics loss is used to align the distributions of the source and target features in the top segmentation network, and entropy minimization loss is used to encourage high confidence segmentations. The pseudo-labels outputted from the top segmentation network is used to guide the style compensation network to generate source-like images. The pseudo-labels outputted from the middle segmentation network is used to supervise the learning of the desired model (the bottom segmentation network). At the second stage, the circular learning and the pixel-adaptive mask refinement are used to further improve the performance of the desired model. With this approach, we achieve satisfactory performances on the segmentations of liver, right kidney, left kidney, and spleen with the dice similarity coefficients of 0.884, 0.891, 0.864, and 0.911, respectively. In addition, the proposed approach can be easily extended to the situation when there exists target annotation data. The performance improves from 0.888 to 0.922 in average dice similarity coefficient, close to the supervised learning (0.929), with only one labeled MR volume.
updated: Tue Nov 30 2021 04:48:21 GMT+0000 (UTC)
published: Wed Nov 24 2021 01:42:07 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アソシエイト