ProstAttention-Net: A deep attention model for prostate cancer segmentation by aggressiveness in MRI scans

Audrey Duran1, Gaspard Dussert1, Olivier Rouvière2

  • 1Univ Lyon, CNRS, Inserm, INSA Lyon, UCBL, CREATIS, UMR5220, U1206, F-69621, Villeurbanne, France.

Medical Image Analysis
|January 27, 2022
PubMed

Insights

This study introduces ProstAttention-Net, an AI model for prostate cancer detection and grading using MRI. It accurately segments prostate glands and cancer lesions, improving diagnostic capabilities beyond current clinical practice.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Oncology

Background:

  • Multiparametric MRI (mp-MRI) shows promise for prostate cancer (PCa) detection.
  • Characterizing PCa aggressiveness and Gleason score (GS) from mp-MRI is challenging in clinical practice.
  • Biopsy remains the gold standard for determining PCa aggressiveness.

Purpose of the Study:

  • To develop a novel end-to-end multi-class network for joint prostate gland segmentation and PCa lesion detection with Gleason score group grading.
  • To improve the characterization of prostate cancer aggressiveness using AI-driven mp-MRI analysis.

Main Methods:

  • An end-to-end multi-class network (ProstAttention-Net) was designed with two branches: one for prostate segmentation and another for lesion detection/grading using zonal priors as an attention gate.
  • The model was trained and validated using 5-fold cross-validation on 219 MRI exams from three scanners.
  • Performance was evaluated using Free-Response Receiver Operating Characteristic (FROC) analysis and Cohen's quadratic weighted kappa (κ) for lesion detection and GS grading.

Main Results:

  • ProstAttention-Net achieved 69.0%±14.5% sensitivity for clinically significant lesion detection (GS > 6) across the whole prostate and 70.8%±14.4% in the peripheral zone.
  • The model obtained a Cohen's kappa of 0.418±0.138 for automatic GS group grading, outperforming previous lesion-wise GS segmentation.
  • State-of-the-art prostate segmentation (Dice of 0.875±0.013) and improved performance over U-Net, DeepLabv3+, and E-Net were demonstrated.

Conclusions:

  • ProstAttention-Net shows strong potential for improving prostate cancer detection, segmentation, and Gleason score grading from mp-MRI.
  • The proposed attention mechanism enhances diagnostic accuracy and generalization capabilities.
  • This AI approach offers a promising non-invasive method to characterize prostate cancer aggressiveness, potentially reducing the need for extensive biopsies.

Related Concept Videos