Highly accurate and effective deep neural networks in pathological diagnosis of prostate cancer

Abstract

Purpose: To established an AI system to make the pathological diagnosis of prostate cancer.

Methods: Prostate histopathological whole mount (WM) sections from patients underwent robot-assisted laparoscopic prostatectomy were prepared. All the prostate WM pathological sections were converted to digital image data and marked with different colors on the basis of the ISUP Gleason grade group. The image was then fed into a segmentation algorithm. We chose modified U-Net as our fundamental network architecture.

Results: 172 patients were involved in this study. 896 pieces of prostate WM pathological sections from 160 patients, in which 826 pieces of WM sections from 148 patients were assigned to the training set randomly. After image segmentation there were totally 2,138,895 patches, of which 1,646,535 patches were valid for training. The other WM section was arranged for testing. Based on the whole image testing, AI and pathologists presented the same answers among 21 of 22 pieces of sections. To evaluate the diagnostic results at the pixel level, we anticipated correct cancer or non-cancer diagnose from this AI system. The area under the ROC curve as 96.8%. The value of pixel accuracy of three methods (binary analysis, clinically oriented analysis and analysis for different ISUP Gleason grade) were 96.93%, 95.43% and 93.88%, respectively. The value of frequency weighted IoU were 94.32%, 92.13% and 90.21%, respectively.

Conclusions: This AI system is able to assist pathologists to make a final diagnosis, indicating the great potential and a wide-range of applications of AI in the medical field.

Keywords: Artificial intelligence; Deep neural network; Image segmentation; Prostate cancer; Whole mount section.