Externally Validated Breast Ultrasound Segmentation via Multi-task Learning with BI-RADS-Consistent Morphological Priors

작성자

카테고리:

← 피드로
arXiv cs.AI · Jingru Zhang, Saed Moradi, Ashirbani Saha · 2026-08-05 AI

[Submitted on 20 Nov 2025 (v1), last revised 4 Aug 2026 (this version, v2)]

View PDF HTML (experimental)

Abstract:External validation of breast ultrasound segmentation models remains limited because internal train–test splits do not capture domain shifts across imaging systems, acquisition protocols, and patient populations. We introduce a novel multi-task framework for breast ultrasound segmentation and malignancy classification. Its central methodological advance is a differentiable morphology-to-malignancy bridge: lesion area, boundary roughness, compactness, and texture are computed from the predicted soft segmentation mask, aggregated with learned weights into a morphology-based malignancy score, and constrained to agree with the image-level classifier. To our knowledge, this is the first breast ultrasound framework to use BI-RADS-inspired morphology derived from its own soft segmentation output as an end-to-end consistency target for malignancy classification. It is also the first 2D B-mode multi-task study to report every directed external transfer among four independent datasets: training on each dataset and testing on the other three yields 12 source–target pairs, assessed with single models and five-fold ensembles. In matched comparisons across all pairs, the proposed single-model configuration outperforms dedicated single-task baselines in segmentation (DC: 0.764 vs. 0.740) and malignancy classification (AUC: 0.818 vs. 0.791). The ensemble achieves a mean external DC of 0.786 and is competitive with SAM-based segmentation methods using an EfficientNet-B7 encoder while also predicting malignancy. The learned morphology weights retain the same ordering across all four datasets, with boundary roughness receiving the greatest weight. These results establish the first complete four-dataset directed benchmark for joint 2D breast ultrasound segmentation and malignancy classification and demonstrate that clinically grounded morphological consistency improves both tasks under domain shift.

Submission history

From: Saed Moradi [view email]
[v1] Thu, 20 Nov 2025 01:45:25 UTC (1,150 KB)
[v2] Tue, 4 Aug 2026 02:45:03 UTC (6,721 KB)

원문에서 계속 ↗

추출 본문 · 출처: arxiv.org · https://arxiv.org/abs/2511.15968

코멘트

답글 남기기

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다