arXiv reaDer
ハイブリッド制約付き半教師あり学習による3DUSでの医療機器セグメンテーション
Medical Instrument Segmentation in 3D US by Hybrid Constrained Semi-Supervised Learning
3D超音波での医療機器のセグメンテーションは、画像誘導介入に不可欠です。ただし、機器のセグメンテーションのために成功するディープニューラルネットワークをトレーニングするには、多数のラベル付き画像が必要であり、取得するのに費用と時間がかかります。この記事では、3D USでの機器セグメンテーションのための半教師あり学習(SSL)フレームワークを提案します。これは、既存の方法よりもはるかに少ない注釈作業で済みます。 SSL学習を実現するために、機器をセグメント化するためのDual-UNetが提案されています。 Dual-UNetは、不確実性とコンテキスト制約で構成される新しいハイブリッド損失関数を使用して、ラベルのないデータを活用します。具体的には、不確実性制約はUNetの予測の不確実性推定を活用するため、SSLトレーニングのラベルなし情報を改善します。さらに、コンテキスト制約は、ボクセルごとの不確実性推定の補足情報として使用されるトレーニング画像のコンテキスト情報を活用します。複数のex-vivoおよびin-vivoデータセットでの広範な実験は、提案された方法が約68.6%-69.1%のダイススコアと約1秒の推論時間を達成することを示しています。ボリュームあたり。これらの結果は、最先端のSSLメソッドよりも優れており、推論時間は教師ありアプローチに匹敵します。
Medical instrument segmentation in 3D ultrasound is essential for image-guided intervention. However, to train a successful deep neural network for instrument segmentation, a large number of labeled images are required, which is expensive and time-consuming to obtain. In this article, we propose a semi-supervised learning (SSL) framework for instrument segmentation in 3D US, which requires much less annotation effort than the existing methods. To achieve the SSL learning, a Dual-UNet is proposed to segment the instrument. The Dual-UNet leverages unlabeled data using a novel hybrid loss function, consisting of uncertainty and contextual constraints. Specifically, the uncertainty constraints leverage the uncertainty estimation of the predictions of the UNet, and therefore improve the unlabeled information for SSL training. In addition, contextual constraints exploit the contextual information of the training images, which are used as the complementary information for voxel-wise uncertainty estimation. Extensive experiments on multiple ex-vivo and in-vivo datasets show that our proposed method achieves Dice score of about 68.6%-69.1% and the inference time of about 1 sec. per volume. These results are better than the state-of-the-art SSL methods and the inference time is comparable to the supervised approaches.
updated: Fri Jul 30 2021 07:59:45 GMT+0000 (UTC)
published: Fri Jul 30 2021 07:59:45 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アソシエイト