arXiv reaDer
3D Semi-Supervised Learning with Uncertainty-Aware Multi-View Co-Training
 ディープニューラルネットワークは、さまざまな分野で多大な影響を与えますが、通常、多くのアプリケーション(特に医療分野)で収集するには高価なトレーニング用のラベル付きデータを大量に必要とします。一方、ラベルなしのデータははるかに豊富です。共同トレーニングなどの半教師付き学習技術は、ラベルのないデータを活用するための強力なツールを提供できます。本論文では、医用画像からの体積データなどの3Dデータの半教師付き学習に対処するために、新しいフレームワークである不確実性認識マルチビューコトレーニング(UMCT)を提案します。私たちの仕事では、3Dデータの多視点の一貫性を活用することで、共同トレーニングを実現しています。 3Dデータを回転または置換することでさまざまなビューを生成し、非対称3Dカーネルを利用して、さまざまなサブネットワークの多様な機能を促進します。さらに、ベイジアンディープラーニングを使用して各ビューの予測の信頼性を推定するために、不確実性加重ラベル融合メカニズムを提案します。あるビューでは他のビューからの協調トレーニングが必要なので、自己適応型アプローチでは、信頼性の高い擬似ラベルを割り当てるために、各ラベルなしサンプルの予測の信頼性スコアを計算します。したがって、このアプローチでは、トレーニング中にラベルのないデータを活用できます。医療画像セグメンテーションタスク(NIH膵臓&LiTS肝腫瘍データセット)からのいくつかのパブリックデータセットで提案された半教師付き方法の有効性を示します。一方、私たちのアプローチに基づく完全に監督された方法は、LiTS肝腫瘍セグメンテーションとMedical Segmentation Decathlon(MSD)チャレンジの両方で最先端のパフォーマンスを達成し、完全に監督された場合でも、フレームワークの堅牢性と価値を実証しましたトレーニングが可能です。
While making a tremendous impact in various fields, deep neural networks usually require large amounts of labeled data for training which are expensive to collect in many applications, especially in the medical domain. Unlabeled data, on the other hand, is much more abundant. Semi-supervised learning techniques, such as co-training, could provide a powerful tool to leverage unlabeled data. In this paper, we propose a novel framework, uncertainty-aware multi-view co-training (UMCT), to address semi-supervised learning on 3D data, such as volumetric data from medical imaging. In our work, co-training is achieved by exploiting multi-viewpoint consistency of 3D data. We generate different views by rotating or permuting the 3D data and utilize asymmetrical 3D kernels to encourage diversified features in different sub-networks. In addition, we propose an uncertainty-weighted label fusion mechanism to estimate the reliability of each view's prediction with Bayesian deep learning. As one view requires the supervision from other views in co-training, our self-adaptive approach computes a confidence score for the prediction of each unlabeled sample in order to assign a reliable pseudo label. Thus, our approach can take advantage of unlabeled data during training. We show the effectiveness of our proposed semi-supervised method on several public datasets from medical image segmentation tasks (NIH pancreas & LiTS liver tumor dataset). Meanwhile, a fully-supervised method based on our approach achieved state-of-the-art performances on both the LiTS liver tumor segmentation and the Medical Segmentation Decathlon (MSD) challenge, demonstrating the robustness and value of our framework, even when fully supervised training is feasible.
updated: Mon Feb 24 2020 01:50:28 GMT+0000 (UTC)
published: Thu Nov 29 2018 21:58:53 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト