arXiv reaDer
ノイズの多いラベルを使用した堅牢な医療画像セグメンテーションのための効果的な監視の蒸留
Distilling effective supervision for robust medical image segmentation with noisy labels
医療画像セグメンテーションタスクでの深層学習手法の成功にもかかわらず、人間レベルのパフォーマンスは、収集に費用と時間がかかる高品質の注釈を含む大量のトレーニングデータに依存しています。事実、ラベルノイズのある低品質の注釈が存在するため、学習したモデルのパフォーマンスが最適ではなくなります。ノイズの多いラベルを使用したセグメンテーション学習の2つの重要な方向性には、ピクセル単位のノイズロバストトレーニングと画像レベルのノイズロバストトレーニングがあります。この作業では、ピクセルレベルと画像レベルの両方から効果的な監視情報を抽出することにより、ノイズの多いラベルを使用したセグメンテーションに対処するための新しいフレームワークを提案します。特に、ピクセル単位のノイズ推定としてすべてのピクセルの不確実性を明示的に推定し、元のラベルと疑似ラベルの両方を使用してピクセル単位のロバストな学習を提案します。さらに、ピクセルレベルの学習を補完するものとして、より多くの情報に対応するための画像レベルのロバストな学習方法を紹介します。シミュレートされたデータセットと実際のノイズの多いデータセットの両方で広範な実験を行います。結果は、ノイズの多いラベルを使用した医療画像セグメンテーションの最先端のベースラインと比較して、私たちの方法の有利なパフォーマンスを示しています。
Despite the success of deep learning methods in medical image segmentation tasks, the human-level performance relies on massive training data with high-quality annotations, which are expensive and time-consuming to collect. The fact is that there exist low-quality annotations with label noise, which leads to suboptimal performance of learned models. Two prominent directions for segmentation learning with noisy labels include pixel-wise noise robust training and image-level noise robust training. In this work, we propose a novel framework to address segmenting with noisy labels by distilling effective supervision information from both pixel and image levels. In particular, we explicitly estimate the uncertainty of every pixel as pixel-wise noise estimation, and propose pixel-wise robust learning by using both the original labels and pseudo labels. Furthermore, we present an image-level robust learning method to accommodate more information as the complements to pixel-level learning. We conduct extensive experiments on both simulated and real-world noisy datasets. The results demonstrate the advantageous performance of our method compared to state-of-the-art baselines for medical image segmentation with noisy labels.
updated: Mon Jun 21 2021 13:33:38 GMT+0000 (UTC)
published: Mon Jun 21 2021 13:33:38 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アソシエイト