arXiv reaDer
Scribble-based Hierarchical Weakly Supervised Learning for Brain Tumor Segmentation
 最近の最先端の深層学習法により、脳腫瘍のセグメンテーションが大幅に改善されました。ただし、完全に監督されたトレーニングには大量の手動ラベル付きマスクが必要であり、非常に時間がかかり、専門知識が必要です。スクリブルを使用した弱い教師付き学習は、モデルの精度と手作業によるラベリングの労力との適切なトレードオフを提供します。ただし、階層的な脳腫瘍構造をセグメント化するために、各下位構造の落書きを手動でラベル付けすることは依然として困難です。このホワイトペーパーでは、2種類の弱いラベル、つまり腫瘍全体と健康な脳組織の落書き、および各下位構造の存在に対するグローバルラベルのみを使用して、深層学習モデルをトレーニングし、すべての下位領域をセグメント化します。具体的には、2つのネットワークを2つのフェーズでトレーニングします。最初に、全体の腫瘍(WT)セグメンテーションネットワークをトレーニングするために、腫瘍全体のスクリブルのみを使用します。次に、グローバルラベルのガイドを使用してWT領域をクラスター化します。クラスタリングからの大まかな部分構造セグメンテーションは、2番目のネットワークをトレーニングするための弱いラベルとして使用されます。密なCRF損失は、弱く監視されたセグメンテーションを改善するために使用されます。 BraTS2017データセットでアプローチを評価し、競合注釈付きWTダイススコアと、完全注釈付きマスクでトレーニングされた場合の上限と比較した部分構造セグメンテーションの同等のスコアを達成します。
The recent state-of-the-art deep learning methods have significantly improved brain tumor segmentation. However, fully supervised training requires a large amount of manually labeled masks, which is highly time-consuming and needs domain expertise. Weakly supervised learning with scribbles provides a good trade-off between model accuracy and the effort of manual labeling. However, for segmenting the hierarchical brain tumor structures, manually labeling scribbles for each substructure could still be demanding. In this paper, we use only two kinds of weak labels, i.e., scribbles on whole tumor and healthy brain tissue, and global labels for the presence of each substructure, to train a deep learning model to segment all the sub-regions. Specifically, we train two networks in two phases: first, we only use whole tumor scribbles to train a whole tumor (WT) segmentation network, which roughly recovers the WT mask of training data; then we cluster the WT region with the guide of global labels. The rough substructure segmentation from clustering is used as weak labels to train the second network. The dense CRF loss is used to refine the weakly supervised segmentation. We evaluate our approach on the BraTS2017 dataset and achieve competitive WT dice score as well as comparable scores on substructure segmentation compared to an upper bound when trained with fully annotated masks.
updated: Tue Nov 05 2019 16:56:35 GMT+0000 (UTC)
published: Tue Nov 05 2019 16:56:35 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト