arXiv reaDer
体積医療画像セグメンテーションのための位置対照学習
Positional Contrastive Learning for Volumetric Medical Image Segmentation
ディープラーニングの成功は、ラベル付きの大規模なトレーニングセットの可用性に大きく依存します。ただし、厳格なプライバシーの懸念と費用のかかるラベル付け作業のため、医用画像ドメインで大きなラベル付きデータセットを取得することは困難です。教師なし学習手法である対照学習は、ラベルのないデータから画像レベルの表現を学習するのに強力であることが証明されています。次に、学習したエンコーダーを転送または微調整して、ラベルが制限されたダウンストリームタスクのパフォーマンスを向上させることができます。対照学習の重要なステップは、対照データペアの生成です。これは、自然画像の分類には比較的簡単ですが、データセット全体に同じ組織または臓器が存在するため、医療画像のセグメンテーションには非常に困難です。その結果、医療画像のセグメンテーションに適用すると、ほとんどの最先端の対照的な学習フレームワークは必然的に多くの偽陰性のペアを導入し、セグメンテーションの品質を低下させます。この問題に対処するために、ボリューム医用画像の位置情報を活用して対照的なデータペアを生成する新しい位置対照学習(PCL)フレームワークを提案します。 CTおよびMRIデータセットの実験結果は、提案されたPCL法が、半教師あり設定と転移学習設定の両方で既存の方法と比較してセグメンテーションパフォーマンスを大幅に改善できることを示しています。
The success of deep learning heavily depends on the availability of large labeled training sets. However, it is hard to get large labeled datasets in medical image domain because of the strict privacy concern and costly labeling efforts. Contrastive learning, an unsupervised learning technique, has been proved powerful in learning image-level representations from unlabeled data. The learned encoder can then be transferred or fine-tuned to improve the performance of downstream tasks with limited labels. A critical step in contrastive learning is the generation of contrastive data pairs, which is relatively simple for natural image classification but quite challenging for medical image segmentation due to the existence of the same tissue or organ across the dataset. As a result, when applied to medical image segmentation, most state-of-the-art contrastive learning frameworks inevitably introduce a lot of false-negative pairs and result in degraded segmentation quality. To address this issue, we propose a novel positional contrastive learning (PCL) framework to generate contrastive data pairs by leveraging the position information in volumetric medical images. Experimental results on CT and MRI datasets demonstrate that the proposed PCL method can substantially improve the segmentation performance compared to existing methods in both semi-supervised setting and transfer learning setting.
updated: Fri Jun 18 2021 03:49:32 GMT+0000 (UTC)
published: Wed Jun 16 2021 22:15:28 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アソシエイト