Deep learning for classification of multi?sequence MR images of the prostate / 中华放射学杂志

ABSTRACT

Objective To develop a convolution neural network (CNN) model to classify multi?sequence MR images of the prostate. Methods ResNet18 convolution neural network (CNN) model was developed to classify multi?sequence MR images of the prostate. A deep residual network was used to improve training accuracy and test accuracy. The dataset used in this experiment included 19 146 7?sequence prostate MR images (transverse T1WI, transverse T2WI, coronal T2WI, sagittal T2WI, transverse DWI, transverse ADC, transverse PWI), from which a total of 2 800 7?sequence MR images was selected as a training set. Three hundred and eighty eight 7?sequence MR images were selected as test sets. Accuracy was used to evaluate the effectiveness of ResNet18 CNN model. Results The classification accuracy of the model for transverse DWI, sagittal T2WI, transverse ADC, transverse T1WI, and transverse T2WI was as high as 100.0% (44/44,52/52), and the accuracy for transverse PWI was also as high as 96.7% (116/120). The accuracy for coronal T2WI was 77.5% (31/40). 0.8% (1/120) of transverse PWI was incorrectly assigned to transverse T2WI, and 2.5% (3/120) incorrectly assigned to sagittal T2WI. 15.0% (6/40) of coronal T2WI was incorrectly assigned to transverse T2WI, and 7.5% (3/40) to sagittal T2WI. Conclusion The experimental results show the effectiveness of our deep learning method regarding accuracy in the prostate multi?sequence MR images detection.