arXiv reaDer
Recurrent Mask Refinement for Few-Shot Medical Image Segmentation
医療画像のセグメンテーションで大きな成功を収めましたが、深い畳み込みニューラルネットワークは通常、トレーニング用に手動の注釈を付けた大きなデータセットを必要とし、目に見えないクラスに一般化することは困難です。少数のショット学習は、いくつかのラベル付きの例から新しいクラスを学習することによって、これらの課題に対処する可能性があります。この作業では、プロトタイプネットワークに基づく数ショットの医療画像セグメンテーションのための新しいフレームワークを提案します。私たちの革新は、2つの主要なモジュールの設計にあります。1)相関を使用して前景領域と背景領域の間のローカル関係の特徴をキャプチャするコンテキスト関係エンコーダー(CRE)。 2)CREとプロトタイプネットワークを繰り返し使用して、コンテキスト関係の変化を再キャプチャし、セグメンテーションマスクを繰り返しリファインする、リカレントマスクリファインモジュール。 2つの腹部CTデータセットと腹部MRIデータセットでの実験は、提案された方法が、DSCに関してそれぞれ平均16.32%、8.45%、および6.24%、最先端の方法よりも大幅に改善されていることを示しています。コードは公開されています。
Although having achieved great success in medical image segmentation, deep convolutional neural networks usually require a large dataset with manual annotations for training and are difficult to generalize to unseen classes. Few-shot learning has the potential to address these challenges by learning new classes from only a few labeled examples. In this work, we propose a new framework for few-shot medical image segmentation based on prototypical networks. Our innovation lies in the design of two key modules: 1) a context relation encoder (CRE) that uses correlation to capture local relation features between foreground and background regions; and 2) a recurrent mask refinement module that repeatedly uses the CRE and a prototypical network to recapture the change of context relationship and refine the segmentation mask iteratively. Experiments on two abdomen CT datasets and an abdomen MRI dataset show the proposed method obtains substantial improvement over the state-of-the-art methods by an average of 16.32%, 8.45% and 6.24% in terms of DSC, respectively. Code is publicly available.
updated: Wed Aug 04 2021 04:27:27 GMT+0000 (UTC)
published: Mon Aug 02 2021 04:06:12 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト