ラベルなしデータの大規模なデータセットで対照学習を使用して認識モデルを事前トレーニングすることで、画像分類などのダウンストリームタスクのパフォーマンスを向上させる大きな可能性が示されています。ただし、医用画像などの分野では、ラベルのないデータを収集することは困難で費用がかかる可能性があります。この作業では、追加のラベルなしデータが利用できない場合でも、医療画像セグメンテーションでのモデルのパフォーマンスを向上させるために、メタラベル注釈を使用するように対照学習を適応させることを提案します。 3D MRIスキャンでの2Dスライスの位置や使用されるデバイスのタイプなどのメタラベルは、多くの場合、取得プロセス中に無料で提供されます。メタラベルを使用して、画像エンコーダーを事前トレーニングするだけでなく、半教師ありトレーニングを正規化します。このトレーニングでは、注釈付きデータの削減されたセットがトレーニングに使用されます。最後に、弱い注釈を完全に活用するために、自己ペース学習アプローチを使用して、学習を支援し、有用なラベルをノイズから区別します。 3つの異なる医療画像セグメンテーションデータセットの結果は、私たちのアプローチが次のことを示しています:i)数回のスキャンでトレーニングされたモデルのパフォーマンスを大幅に向上させ、ii)以前の対照的および半教師ありアプローチを上回り、iii)モデルのパフォーマンスに近づきます完全なデータでトレーニングされています。
Pre-training a recognition model with contrastive learning on a large dataset of unlabeled data has shown great potential to boost the performance of a downstream task, e.g., image classification. However, in domains such as medical imaging, collecting unlabeled data can be challenging and expensive. In this work, we propose to adapt contrastive learning to work with meta-label annotations, for improving the model's performance in medical image segmentation even when no additional unlabeled data is available. Meta-labels such as the location of a 2D slice in a 3D MRI scan or the type of device used, often come for free during the acquisition process. We use the meta-labels for pre-training the image encoder as well as to regularize a semi-supervised training, in which a reduced set of annotated data is used for training. Finally, to fully exploit the weak annotations, a self-paced learning approach is used to help the learning and discriminate useful labels from noise. Results on three different medical image segmentation datasets show that our approach: i) highly boosts the performance of a model trained on a few scans, ii) outperforms previous contrastive and semi-supervised approaches, and iii) reaches close to the performance of a model trained on the full data.