# AI-based prostate volume estimation from multi-planar MRI under variable acquisition protocols

## TL;DR

This study introduces a deep learning model that accurately estimates prostate volume from MRI scans, even when only limited axial-plane data is available.

## Contribution

The novel knowledge-based contrastive loss enables accurate prostate volume estimation under abbreviated MRI protocols.

## Key findings

- The KB model achieved a DSC of 0.91 and RVD of 2.1% with full multi-planar input.
- Under axial-only conditions, the KB model maintained a DSC of 0.88 and RVD of 3.4%.
- The model showed excellent agreement with reference volumes (ICC > 0.90) across conditions.

## Abstract

Prostate MRI protocols vary across institutions, with abbreviated protocols increasingly limited to axial plane acquisitions. Conventional deep learning (DL) models for prostate volume (PV) estimation typically require complete availability of annotated full imaging protocols during training and inference, limiting their adaptability in real-world clinical workflows. This study aimed to develop and evaluate a knowledge-based (KB) DL segmentation model that adapts to variable MRI acquisition protocols, including axial-only abbreviated protocols.

This retrospective study included 629 multiparametric 3-Tesla prostate MRI exams (66.60 ± 7.50 years) from biopsy-confirmed patients. Manual segmentations by expert radiologists and ellipsoid-derived volumes per PI-RADS 2.1 (PVref) served as reference standards. A 2D nnU-Net–based DL model with a KB contrastive loss was trained using only axial segmentations while incorporating unannotated orthogonal views (PVKB). Performance was compared to a fully supervised nnU-Net-based DL model trained with full multi-planar annotations and data (PVDL). Evaluation metrics included Dice Score Coefficient (DSC), Relative Volume Difference (RVD), Bland-Altman analysis, and intraclass correlation coefficient (ICC). Experiments simulated both full and abbreviated protocols (axial-only). Wilcoxon signed-rank tests were used to evaluate statistical differences in performance across configurations. Statistical significance was set at p < 0.05.

With full multi-planar input, the KB model achieved a DSC of 0.91 ± 0.03 and RVD of 2.1 ± 6.4%, comparable to the fully supervised PVDLmodel. Under axial-only conditions, the KB model maintained high performance (DSC:0.88 ± 0.04,RVD:3.4 ± 7.1%). PV agreement with PVrefremained excellent across conditions (ICC>0.90).

The proposed KB DL model enables accurate and flexible PV assessment under variable MRI protocols without requiring segmentation masks beyond the axial plane.

•The proposed knowledge-based contrastive loss allows to adapt to abbreviated MRI prostate protocols.•The proposed model achieves high accuracy in prostate volume estimation, even under axial-only abbreviated protocols.•Agreement with ellipsoid-based reference volumes was high even across heterogeneous input conditions.

The proposed knowledge-based contrastive loss allows to adapt to abbreviated MRI prostate protocols.

The proposed model achieves high accuracy in prostate volume estimation, even under axial-only abbreviated protocols.

Agreement with ellipsoid-based reference volumes was high even across heterogeneous input conditions.