Prompt-Free SAM-Based Multi-Task Framework for Breast Ultrasound Lesion Segmentation and Classification

Samuel E. Johnny, Bernes L. Atabonfack, Israel Alagbe, Assane Gueye
1/9/2026
cs.CVcs.AIcs.LG

No official code found. Best implementation: caumente/multi_task_breast_cancer (16 stars, Apache-2.0, last commit 2026-03-20).

Abstract

Accurate tumor segmentation and classification in breast ultrasound (BUS) imaging remain challenging due to low contrast, speckle noise, and diverse lesion morphology. This study presents a multi-task deep learning framework that jointly performs lesion segmentation and diagnostic classification using embeddings from the Segment Anything Model (SAM) vision encoder. Unlike prompt-based SAM variants, our approach employs a prompt-free, fully supervised adaptation where high-dimensional SAM features are decoded through either a lightweight convolutional head or a UNet-inspired decoder for pixel-wise segmentation. The classification branch is enhanced via mask-guided attention, allowing the model to focus on lesion-relevant features while suppressing background artifacts. Experiments on the PRECISE 2025 breast ultrasound dataset, split per class into 80 percent training and 20 percent testing, show that the proposed method achieves a Dice Similarity Coefficient (DSC) of 0.887 and an accuracy of 92.3 percent, ranking among the top entries on the PRECISE challenge leaderboard. These results demonstrate that SAM-based representations, when coupled with segmentation-guided learning, significantly improve both lesion delineation and diagnostic prediction in breast ultrasound imaging.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

Multi-task framework for breast cancer segmentation and classification

164PythonJul 17, 20247 months agoApache-2.0
breast-cancerclassificationcomputer-visiondata-curationdeep-learning+2 more

Cite this paper

@article{johnny2026promptfree,
  title  = {Prompt-Free SAM-Based Multi-Task Framework for Breast Ultrasound Lesion Segmentation and Classification},
  author = {Samuel E. Johnny and Bernes L. Atabonfack and Israel Alagbe and Assane Gueye},
  year   = {2026},
  eprint = {2601.05498},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.05498v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.05498.svg)](https://codeofpaper.com/paper/2601.05498)

Discussion