# Leveraging multi-view data without annotations for prostate MRI segmentation: A contrastive approach

arXiv: https://arxiv.org/abs/2308.06477 · HTML: https://mlsanity.com/paper/2308.06477/

## Key facts

- Cited by 3 papers in the map (data through 14 Oct 2026).
- Builds on 6 of its 25 references that are in the map.
- First posted on arXiv 12 Aug 2023.

## Builds on (most cited first, 6 of 6)

- [A Simple Framework for Contrastive Learning of Visual Representations](https://mlsanity.com/paper/2002.05709/) (2020)
- [TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation](https://mlsanity.com/paper/2102.04306/) (2021)
- [Big Self-Supervised Models Advance Medical Image Classification](https://mlsanity.com/paper/2101.05224/) (2021)
- [Improving prostate whole gland segmentation in t2-weighted MRI with synthetically generated data](https://mlsanity.com/paper/2103.14955/) (2021)
- [Anisotropic 3D Multi-Stream CNN for Accurate Prostate Segmentation from Multi-Planar MRI](https://mlsanity.com/paper/2009.11120/) (2020)
- [Prostate Age Gap (PAG): An MRI surrogate marker of aging for prostate cancer detection](https://mlsanity.com/paper/2308.05344/) (2023)

## Built on by (most cited first, 3 of 3)

- [On undesired emergent behaviors in compound prostate cancer detection systems](https://mlsanity.com/paper/2309.08381/) (2023)
- [MOSAIC: A Multi-View 2.5D Organ Slice Selector with Cross-Attentional Reasoning for Anatomically-Aware CT Localization in Medical Organ Segmentation](https://mlsanity.com/paper/2505.10672/) (2025)
- [Neuropsychiatric Deviations From Normative Profiles: An MRI-Derived Marker for Early Alzheimer's Disease Detection](https://mlsanity.com/paper/2604.00545/) (2026)

## Abstract (arXiv)

An accurate prostate delineation and volume characterization can support the clinical assessment of prostate cancer. A large amount of automatic prostate segmentation tools consider exclusively the axial MRI direction in spite of the availability as per acquisition protocols of multi-view data. Further, when multi-view data is exploited, manual annotations and availability at test time for all the views is commonly assumed. In this work, we explore a contrastive approach at training time to leverage multi-view data without annotations and provide flexibility at deployment time in the event of missing views. We propose a triplet encoder and single decoder network based on U-Net, tU-Net (triplet U-Net). Our proposed architecture is able to exploit non-annotated sagittal and coronal views via contrastive learning to improve the segmentation from a volumetric perspective. For that purpose, we introduce the concept of inter-view similarity in the latent space. To guide the training, we combine a dice score loss calculated with respect to the axial view and its manual annotations together with a multi-view contrastive loss. tU-Net shows statistical improvement in dice score coefficient (DSC) with respect to only axial view (91.25+-0.52% compared to 86.40+-1.50%,P<.001). Sensitivity analysis reveals the volumetric positive impact of the contrastive loss when paired with tU-Net (2.85+-1.34% compared to 3.81+-1.88%,P<.001). Further, our approach shows good external volumetric generalization in an in-house dataset when tested with multi-view data (2.76+-1.89% compared to 3.92+-3.31%,P=.002), showing the feasibility of exploiting non-annotated multi-view data through contrastive learning whilst providing flexibility at deployment in the event of missing views.
