正確な診断と縦断的研究には、3D医用画像で関心のある臓器のセグメンテーションが必要です。ディープラーニングを使用した最近の進歩は多くのセグメンテーションタスクで成功を示していますが、高性能には大規模なデータセットが必要であり、注釈プロセスは時間と労力の両方を伴います。この論文では、ターゲット臓器アノテーションの限られたトレーニングサンプルを使用して正確な臓器セグメンテーションのための3D数ショットセグメンテーションフレームワークを提案します。これを実現するために、U-Netのようなネットワークは、サポートデータの2Dスライスと、隣接するスライス間のエンコードされた特徴の一貫性を学習する双方向ゲート付き回帰ユニット(GRU)を含むクエリ画像との関係を学習することにより、セグメンテーションを予測するように設計されています。また、任意のサポートでテストする前にモデルを更新し、サポートデータからサンプリングしたクエリデータを使用して、ターゲット画像と臓器の特性を適応させる転移学習法を紹介します。異なる臓器の注釈付きの3つの3DCTデータセットを使用して、提案されたモデルを評価します。私たちのモデルは、最先端の数ショットセグメンテーションモデルよりも大幅に改善されたパフォーマンスをもたらし、より多くのターゲットトレーニングデータでトレーニングされた完全に監視されたモデルに匹敵しました。
Segmentation of organs of interest in 3D medical images is necessary for accurate diagnosis and longitudinal studies. Though recent advances using deep learning have shown success for many segmentation tasks, large datasets are required for high performance and the annotation process is both time consuming and labor intensive. In this paper, we propose a 3D few shot segmentation framework for accurate organ segmentation using limited training samples of the target organ annotation. To achieve this, a U-Net like network is designed to predict segmentation by learning the relationship between 2D slices of support data and a query image, including a bidirectional gated recurrent unit (GRU) that learns consistency of encoded features between adjacent slices. Also, we introduce a transfer learning method to adapt the characteristics of the target image and organ by updating the model before testing with arbitrary support and query data sampled from the support data. We evaluate our proposed model using three 3D CT datasets with annotations of different organs. Our model yielded significantly improved performance over state-of-the-art few shot segmentation models and was comparable to a fully supervised model trained with more target training data.