arXiv reaDer
半教師あり医療画像セグメンテーションのための自己アンサンブル対照学習
Self-Ensembling Contrastive Learning for Semi-Supervised Medical Image Segmentation
ディープラーニングは、手動ラベル付きの十分な量のトレーニングデータを使用して、医療画像セグメンテーションの大幅な改善を示しています。代表的なラベルを取得するには、専門知識と徹底的な労力が必要です。この論文では、自己アンサンブル対照学習手法を使用して、限定されたラベルを使用した医療画像セグメンテーションの半教師あり学習のパフォーマンスを向上させることを目指しています。この目的のために、少量のラベル付き画像を使用して画像レベルでエンコーダ-デコーダネットワークをトレーニングすることを提案します。さらに重要なことに、ラベルなし画像に対照的な損失を課すことにより、特徴レベルで潜在表現を直接学習します。この方法は、クラス内のコンパクト性とクラス間の分離可能性を強化して、より優れたピクセル分類器を取得します。さらに、オンライン学習用の学生エンコーダーと、教師エンコーダーと呼ばれる指数移動平均バージョンを考案して、自己アンサンブル方式でパフォーマンスを繰り返し改善します。ラベルのない画像で対照的なサンプルを構築するために、医用画像全体の構造の類似性を活用し、構築に疑似ラベルを利用する、領域認識および解剖学的認識の対照サンプリングと呼ばれる2つのサンプリング戦略を調査します。 MRIおよびCTセグメンテーションデータセットで広範な実験を実施し、限られたラベル設定で、提案された方法が最先端のパフォーマンスを達成することを実証します。さらに、疑似ラベルを使用してオンザフライで対照的なサンプルを準備する解剖学的認識戦略により、特徴表現のより優れた対照的な正則化が実現されます。
Deep learning has demonstrated significant improvements in medical image segmentation using a sufficiently large amount of training data with manual labels. Acquiring well-representative labels requires expert knowledge and exhaustive labors. In this paper, we aim to boost the performance of semi-supervised learning for medical image segmentation with limited labels using a self-ensembling contrastive learning technique. To this end, we propose to train an encoder-decoder network at image-level with small amounts of labeled images, and more importantly, we learn latent representations directly at feature-level by imposing contrastive loss on unlabeled images. This method strengthens intra-class compactness and inter-class separability, so as to get a better pixel classifier. Moreover, we devise a student encoder for online learning and an exponential moving average version of it, called teacher encoder, to improve the performance iteratively in a self-ensembling manner. To construct contrastive samples with unlabeled images, two sampling strategies that exploit structure similarity across medical images and utilize pseudo-labels for construction, termed region-aware and anatomical-aware contrastive sampling, are investigated. We conduct extensive experiments on an MRI and a CT segmentation dataset and demonstrate that in a limited label setting, the proposed method achieves state-of-the-art performance. Moreover, the anatomical-aware strategy that prepares contrastive samples on-the-fly using pseudo-labels realizes better contrastive regularization on feature representations.
updated: Thu May 27 2021 03:27:58 GMT+0000 (UTC)
published: Thu May 27 2021 03:27:58 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アソシエイト