arXiv reaDer
SAR: Scale-Aware Restoration Learning for 3D Tumor Segmentation
医用画像での自動で正確な腫瘍セグメンテーションは、医師の診断と治療を支援するために高い需要があります。ただし、手動の描写プロセスは面倒で専門知識が必要になることが多いため、ディープラーニングモデルに必要な大量の注釈付きトレーニングデータを取得することは困難です。この問題に対処するために自己教師あり学習(SSL)スキームが広く採用されていますが、ほとんどのSSL手法は、腫瘍領域の主要な特徴である局所強度変動と大きなサイズ分布を無視して、グローバル構造情報のみに焦点を当てています。本論文では、3D腫瘍セグメンテーションのためのSSL法であるScale-Aware Restoration(SAR)を提案します。具体的には、自己復元タスクと組み合わせて3Dニューラルネットワークを事前トレーニングするために、新しいプロキシタスク、つまりスケール弁別が定式化されます。したがって、事前にトレーニングされたモデルは、マルチスケール入力を通じてマルチレベルのローカル表現を学習します。さらに、複数のラベルなしソースデータセットからモダリティ不変表現を学習するために、敵対的学習モジュールがさらに導入されています。私たちは、2つの下流のタスクで私たちの方法の有効性を示しています:i)脳腫瘍のセグメンテーション、ii)膵臓腫瘍のセグメンテーション。最先端の3DSSL手法と比較して、提案されたアプローチはセグメンテーションの精度を大幅に向上させることができます。さらに、データ効率、パフォーマンス、収束速度などの複数の観点からその利点を分析します。
Automatic and accurate tumor segmentation on medical images is in high demand to assist physicians with diagnosis and treatment. However, it is difficult to obtain massive amounts of annotated training data required by the deep-learning models as the manual delineation process is often tedious and expertise required. Although self-supervised learning (SSL) scheme has been widely adopted to address this problem, most SSL methods focus only on global structure information, ignoring the key distinguishing features of tumor regions: local intensity variation and large size distribution. In this paper, we propose Scale-Aware Restoration (SAR), a SSL method for 3D tumor segmentation. Specifically, a novel proxy task, i.e. scale discrimination, is formulated to pre-train the 3D neural network combined with the self-restoration task. Thus, the pre-trained model learns multi-level local representations through multi-scale inputs. Moreover, an adversarial learning module is further introduced to learn modality invariant representations from multiple unlabeled source datasets. We demonstrate the effectiveness of our methods on two downstream tasks: i) Brain tumor segmentation, ii) Pancreas tumor segmentation. Compared with the state-of-the-art 3D SSL methods, our proposed approach can significantly improve the segmentation accuracy. Besides, we analyze its advantages from multiple perspectives such as data efficiency, performance, and convergence speed.
updated: Thu Aug 05 2021 09:25:04 GMT+0000 (UTC)
published: Tue Oct 13 2020 01:23:17 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト