arXiv reaDer
類似の解剖学的構造にわたるドメイン適応セグメンテーションのための対照的な半教師あり学習
Contrastive Semi-supervised Learning for Domain Adaptive Segmentation Across Similar Anatomical Structures
畳み込みニューラル ネットワーク (CNN) は、医療画像のセグメンテーションで最先端のパフォーマンスを達成しましたが、トレーニングには多くの手動注釈が必要です。半教師あり学習 (SSL) メソッドは、アノテーションの要件を減らすことを約束していますが、データセットのサイズとアノテーション付きの画像の数が少ない場合、そのパフォーマンスは依然として制限されています。類似の解剖学的構造を持つ既存の注釈付きデータセットを活用してトレーニングを支援すると、モデルのパフォーマンスが向上する可能性があります。ただし、ターゲット構造とは外観が異なり、イメージングモダリティでさえあるため、解剖学的ドメインのシフトによってさらに課題が生じます。この問題を解決するために、クロス解剖ドメイン適応 (CS-CADA) の対照的半教師あり学習を提案します。これは、モデルを適応させて、ターゲット ドメイン内の類似の構造をセグメント化します。ソース ドメイン内の類似構造の注釈付き画像。ドメイン固有のバッチ正規化 (DSBN) を使用して、2 つの解剖学的ドメインの特徴マップを個別に正規化し、ドメイン不変特徴の抽出を促進するクロスドメイン対照学習戦略を提案します。それらは自己アンサンブル平均教師 (SE-MT) フレームワークに統合され、予測の一貫性の制約を使用して、ラベルのないターゲット ドメイン イメージを活用します。大規模な実験により、当社の CS-CADA は解剖学的領域をまたがる困難なドメイン シフトの問題を解決できることが示されています。網膜血管画像を使用して X 線画像で冠状動脈を正確に分割し、眼底画像を使用して心臓 MR 画像を正確に分割できます。それぞれ、ターゲット ドメイン内の少数の注釈のみが与えられます。
Convolutional Neural Networks (CNNs) have achieved state-of-the-art performance for medical image segmentation, yet need plenty of manual annotations for training. Semi-Supervised Learning (SSL) methods are promising to reduce the requirement of annotations, but their performance is still limited when the dataset size and the number of annotated images are small. Leveraging existing annotated datasets with similar anatomical structures to assist training has a potential for improving the model's performance. However, it is further challenged by the cross-anatomy domain shift due to the different appearance and even imaging modalities from the target structure. To solve this problem, we propose Contrastive Semi-supervised learning for Cross Anatomy Domain Adaptation (CS-CADA) that adapts a model to segment similar structures in a target domain, which requires only limited annotations in the target domain by leveraging a set of existing annotated images of similar structures in a source domain. We use Domain-Specific Batch Normalization (DSBN) to individually normalize feature maps for the two anatomical domains, and propose a cross-domain contrastive learning strategy to encourage extracting domain invariant features. They are integrated into a Self-Ensembling Mean-Teacher (SE-MT) framework to exploit unlabeled target domain images with a prediction consistency constraint. Extensive experiments show that our CS-CADA is able to solve the challenging cross-anatomy domain shift problem, achieving accurate segmentation of coronary arteries in X-ray images with the help of retinal vessel images and cardiac MR images with the help of fundus images, respectively, given only a small number of annotations in the target domain.
updated: Thu Aug 18 2022 02:54:04 GMT+0000 (UTC)
published: Thu Aug 18 2022 02:54:04 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アソシエイト