# SimPLe: Similarity-Aware Propagation Learning for Weakly-Supervised Breast Cancer Segmentation in DCE-MRI

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

## Key facts

- Cited by 6 papers in the map (data through 14 Oct 2026).
- Builds on 6 of its 27 references that are in the map.
- First posted on arXiv 29 Jun 2023.

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

- [3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation](https://mlsanity.com/paper/1606.06650/) (2016)
- [Going to Extremes: Weakly Supervised Medical Image Segmentation](https://mlsanity.com/paper/2009.11988/) (2020)
- [Scribble-based Domain Adaptation via Co-segmentation](https://mlsanity.com/paper/2007.03632/) (2020)
- [Inter Extreme Points Geodesics for End-to-End Weakly Supervised Image Segmentation](https://mlsanity.com/paper/2107.00583/) (2021)
- [Learning Cross-Modal Deep Representations for Multi-Modal MR Image Segmentation](https://mlsanity.com/paper/1908.01997/) (2019)
- [Weakly-Supervised 3D Medical Image Segmentation using Geometric Prior and Contrastive Similarity](https://mlsanity.com/paper/2302.02125/) (2023)

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

- [From Few to More: Scribble-based Medical Image Segmentation via Masked Context Modeling and Continuous Pseudo Labels](https://mlsanity.com/paper/2408.12814/) (2024)
- [Flip Learning: Weakly Supervised Erase to Segment Nodules in Breast Ultrasound](https://mlsanity.com/paper/2503.20685/) (2025)
- [Multi-modality transrectal ultrasound video classification for identification of clinically significant prostate cancer](https://mlsanity.com/paper/2402.08987/) (2024)
- [Towards Cross-Scale Attention and Surface Supervision for Fractured Bone Segmentation in CT](https://mlsanity.com/paper/2405.01204/) (2024)
- [Exploring Kinetic Curves Features for the Classification of Benign and Malignant Breast Lesions in DCE-MRI](https://mlsanity.com/paper/2404.13929/) (2024)
- [Contextual Embedding Learning to Enhance 2D Networks for Volumetric Image Segmentation](https://mlsanity.com/paper/2404.01723/) (2024)

## Abstract (arXiv)

Breast dynamic contrast-enhanced magnetic resonance imaging (DCE-MRI) plays an important role in the screening and prognosis assessment of high-risk breast cancer. The segmentation of cancerous regions is essential useful for the subsequent analysis of breast MRI. To alleviate the annotation effort to train the segmentation networks, we propose a weakly-supervised strategy using extreme points as annotations for breast cancer segmentation. Without using any bells and whistles, our strategy focuses on fully exploiting the learning capability of the routine training procedure, i.e., the train - fine-tune - retrain process. The network first utilizes the pseudo-masks generated using the extreme points to train itself, by minimizing a contrastive loss, which encourages the network to learn more representative features for cancerous voxels. Then the trained network fine-tunes itself by using a similarity-aware propagation learning (SimPLe) strategy, which leverages feature similarity between unlabeled and positive voxels to propagate labels. Finally the network retrains itself by employing the pseudo-masks generated using previous fine-tuned network. The proposed method is evaluated on our collected DCE-MRI dataset containing 206 patients with biopsy-proven breast cancers. Experimental results demonstrate our method effectively fine-tunes the network by using the SimPLe strategy, and achieves a mean Dice value of 81%.
