コンボリューショナルニューラルネットワーク(CNN)は前立腺癌の自動検出に使用されており、受信者動作特性(ROC)曲線(AUC)がパフォーマンスメトリックとして通常使用されます。 AUCが微分可能でない場合、一般的な方法は、クロスエントロピーやAUCの監視などの他のパフォーマンスメトリックに基づいて損失関数を使用してCNNをトレーニングし、最適なモデルを選択することです。この作業では、より高いAUCを達成するために、遺伝的アルゴリズムを使用して、前立腺癌の検出用に訓練されたCNNを微調整することを提案します。データセットには、前立腺の6チャンネル拡散強調MRIスライスが含まれていました。 2,955のトレーニング、1,417の検証、および1,334のテストスライスのコホートで、テストAUC 0.773に達しました。基本CNNモデルと比較して9.3%の改善。
Convolutional Neural Networks (CNNs) have been used for automated detection of prostate cancer where Area Under Receiver Operating Characteristic (ROC) curve (AUC) is usually used as the performance metric. Given that AUC is not differentiable, common practice is to train the CNN using a loss functions based on other performance metrics such as cross entropy and monitoring AUC to select the best model. In this work, we propose to fine-tune a trained CNN for prostate cancer detection using a Genetic Algorithm to achieve a higher AUC. Our dataset contained 6-channel Diffusion-Weighted MRI slices of prostate. On a cohort of 2,955 training, 1,417 validation, and 1,334 test slices, we reached test AUC of 0.773; a 9.3% improvement compared to the base CNN model.