拡散強調MRI(DWI)での前立腺全腺と遷移ゾーンのセグメンテーションは、前立腺癌のコンピューター支援検出アルゴリズムを設計するための最初のステップです。ただし、MRI取得パラメータとスキャナーの製造方法が異なると、画像内の前立腺組織の外観が異なります。セグメンテーションを含むさまざまな医用画像分析タスクで成功することが示されている畳み込みニューラルネットワーク(CNN)は、通常、画像パラメータの変動に敏感です。この感度により、ソースコホートでトレーニングされ、別のスキャナーからのターゲットコホートでテストされたCNNのセグメンテーションパフォーマンスが低下するため、クロスコホートのトレーニングとテストへのCNNの適用性が制限されます。 DWI画像の前立腺全体と移行帯の輪郭を描くことは時間と費用がかかります。したがって、ソースドメインの画像で事前トレーニングされたCNNを有効にして、ターゲットドメインからの画像の手動セグメンテーションの最小要件でターゲットドメインの画像をセグメント化できるようにすることが重要です。この作業では、ソースデータセットで事前トレーニングされ、ターゲットデータセットでテストされたCNNを使用して、DWIの前立腺全腺と遷移ゾーンをセグメンテーションするための修正U-netアーキテクチャと損失関数に基づく転移学習法を提案します。事前にトレーニングされたCNNを微調整するために使用されるターゲットデータセットのサブセットのサイズが、全体的なセグメンテーションの精度に与える影響を調査します。私たちの結果は、ターゲットドメインからのわずか30人の患者の微調整データで、提案された転移学習ベースのアルゴリズムが、前立腺全体と遷移ゾーンセグメンテーションの両方で0.80のダイススコア係数に達することができることを示しています。ターゲットドメインからの115人の患者の微調整データを使用して、0.85と0.84のダイススコア係数が、ターゲットドメインの腺全体と遷移ゾーンのセグメンテーションでそれぞれ達成されます。
The segmentation of prostate whole gland and transition zone in Diffusion Weighted MRI (DWI) are the first step in designing computer-aided detection algorithms for prostate cancer. However, variations in MRI acquisition parameters and scanner manufacturing result in different appearances of prostate tissue in the images. Convolutional neural networks (CNNs) which have shown to be successful in various medical image analysis tasks including segmentation are typically sensitive to the variations in imaging parameters. This sensitivity leads to poor segmentation performance of CNNs trained on a source cohort and tested on a target cohort from a different scanner and hence, it limits the applicability of CNNs for cross-cohort training and testing. Contouring prostate whole gland and transition zone in DWI images are time-consuming and expensive. Thus, it is important to enable CNNs pretrained on images of source domain, to segment images of target domain with minimum requirement for manual segmentation of images from the target domain. In this work, we propose a transfer learning method based on a modified U-net architecture and loss function, for segmentation of prostate whole gland and transition zone in DWIs using a CNN pretrained on a source dataset and tested on the target dataset. We explore the effect of the size of subset of target dataset used for fine-tuning the pre-trained CNN on the overall segmentation accuracy. Our results show that with a fine-tuning data as few as 30 patients from the target domain, the proposed transfer learning-based algorithm can reach dice score coefficient of 0.80 for both prostate whole gland and transition zone segmentation. Using a fine-tuning data of 115 patients from the target domain, dice score coefficient of 0.85 and 0.84 are achieved for segmentation of whole gland and transition zone, respectively, in the target domain.