Weakly Supervised MRI Slice‐Level Deep Learning Classification of Prostate Cancer Approximates Full Voxel‐ and Slice‐Level Annotation: Effect of Increasing Training Set Size

Background Weakly supervised learning promises reduced annotation effort while maintaining performance. Purpose To compare weakly supervised training with full slice‐wise annotated training of a deep convolutional classification network (CNN) for prostate cancer (PC). Study Type Retrospective. Subje...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Journal of magnetic resonance imaging 2024-04, Vol.59 (4), p.1409-1422
Hauptverfasser: Weißer, Cedric, Netzer, Nils, Görtz, Magdalena, Schütz, Viktoria, Hielscher, Thomas, Schwab, Constantin, Hohenfellner, Markus, Schlemmer, Heinz‐Peter, Maier‐Hein, Klaus H., Bonekamp, David
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Background Weakly supervised learning promises reduced annotation effort while maintaining performance. Purpose To compare weakly supervised training with full slice‐wise annotated training of a deep convolutional classification network (CNN) for prostate cancer (PC). Study Type Retrospective. Subjects One thousand four hundred eighty‐nine consecutive institutional prostate MRI examinations from men with suspicion for PC (65 ± 8 years) between January 2015 and November 2020 were split into training (N = 794, enriched with 204 PROSTATEx examinations) and test set (N = 695). Field Strength/Sequence 1.5 and 3T, T2‐weighted turbo‐spin‐echo and diffusion‐weighted echo‐planar imaging. Assessment Histopathological ground truth was provided by targeted and extended systematic biopsy. Reference training was performed using slice‐level annotation (SLA) and compared to iterative training utilizing patient‐level annotations (PLAs) with supervised feedback of CNN estimates into the next training iteration at three incremental training set sizes (N = 200, 500, 998). Model performance was assessed by comparing specificity at fixed sensitivity of 0.97 [254/262] emulating PI‐RADS ≥ 3, and 0.88–0.90 [231–236/262] emulating PI‐RADS ≥ 4 decisions. Statistical Tests Receiver operating characteristic (ROC) and area under the curve (AUC) was compared using DeLong and Obuchowski test. Sensitivity and specificity were compared using McNemar test. Statistical significance threshold was P = 0.05. Results Test set (N = 695) ROC‐AUC performance of SLA (trained with 200/500/998 exams) was 0.75/0.80/0.83, respectively. PLA achieved lower ROC‐AUC of 0.64/0.72/0.78. Both increased performance significantly with increasing training set size. ROC‐AUC for SLA at 500 exams was comparable to PLA at 998 exams (P = 0.28). ROC‐AUC was significantly different between SLA and PLA at same training set sizes, however the ROC‐AUC difference decreased significantly from 200 to 998 training exams. Emulating PI‐RADS ≥ 3 decisions, difference between PLA specificity of 0.12 [51/433] and SLA specificity of 0.13 [55/433] became undetectable (P = 1.0) at 998 exams. Emulating PI‐RADS ≥ 4 decisions, at 998 exams, SLA specificity of 0.51 [221/433] remained higher than PLA specificity at 0.39 [170/433]. However, PLA specificity at 998 exams became comparable to SLA specificity of 0.37 [159/433] at 200 exams (P = 0.70). Data Conclusion Weakly supervised training of a classification CNN using patient‐level‐only an
ISSN:1053-1807
1522-2586
1522-2586
DOI:10.1002/jmri.28891