arXiv reaDer
半教師あり医療画像セグメンテーションのためのクロスレベルの対照学習と一貫性の制約
Cross-level Contrastive Learning and Consistency Constraint for Semi-supervised Medical Image Segmentation
少数のラベル付き画像と多数のラベルなし画像をネットワークトレーニングに活用することを目的とした半教師あり学習(SSL)は、医療画像セグメンテーションにおけるデータ注釈の負担を軽減するのに役立ちます。医用画像の専門家の経験によると、テクスチャ、光沢、滑らかさなどの局所的な属性は、医用画像の病変やポリープなどのターゲットオブジェクトを識別するための非常に重要な要素です。これに動機付けられて、半教師あり医療画像セグメンテーションにおける局所特徴の表現能力を強化するために、クロスレベルの対照学習スキームを提案します。既存の画像ワイズ、パッチワイズ、ポイントワイズの対照学習アルゴリズムと比較して、私たちの考案した方法は、より複雑な類似性の手がかり、すなわちグローバルとローカルのパッチワイズ表現間の関係特性を探索することができます。さらに、クロスレベルのセマンティック関係を十分に活用するために、パッチの予測を完全な画像の予測と比較する新しい整合性制約を考案します。クロスレベルの対照的な学習と一貫性の制約の助けを借りて、ラベルのないデータを効果的に探索して、ポリープと皮膚病変のセグメンテーションの2つの医療画像データセットのセグメンテーションパフォーマンスをそれぞれ向上させることができます。私たちのアプローチのコードが利用可能です。
Semi-supervised learning (SSL), which aims at leveraging a few labeled images and a large number of unlabeled images for network training, is beneficial for relieving the burden of data annotation in medical image segmentation. According to the experience of medical imaging experts, local attributes such as texture, luster and smoothness are very important factors for identifying target objects like lesions and polyps in medical images. Motivated by this, we propose a cross-level contrastive learning scheme to enhance representation capacity for local features in semi-supervised medical image segmentation. Compared to existing image-wise, patch-wise and point-wise contrastive learning algorithms, our devised method is capable of exploring more complex similarity cues, namely the relational characteristics between global and local patch-wise representations. Additionally, for fully making use of cross-level semantic relations, we devise a novel consistency constraint that compares the predictions of patches against those of the full image. With the help of the cross-level contrastive learning and consistency constraint, the unlabelled data can be effectively explored to improve segmentation performance on two medical image datasets for polyp and skin lesion segmentation respectively. Code of our approach is available.
updated: Sun Feb 13 2022 21:31:13 GMT+0000 (UTC)
published: Tue Feb 08 2022 15:12:11 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アソシエイト