MassSeg-Framework: A Breast Mass Detection and Segmentation Framework Based on Deep Learning and an Active Contour Model
Open Access
Abstract
This work introduces the MassSeg-Framework, a fully automatic two-stage pipeline for breast mass analysis in mammography that integrates YOLOv11-based detection with Chan–Vese ACM refinement to achieve accurate mass localization and segmentation with a lightweight computational footprint. The framework was trained and evaluated on two publicly available datasets using consistent experimental protocols. In the detection stage, YOLOv11-nano was the most effective architecture, with a confidence threshold of 0.4, achieving statistically significant mAP50 values of 0.862 and 0.709 on the dINbreast and dCBIS datasets, respectively. These results confirm that a moderate threshold preserves clinically relevant true-positive candidates, which is particularly important for screening-oriented settings where missed lesions are costly. In the segmentation stage, the proposed framework achieved mean DICE scores of 0.721 and 0.700 on the test sets of the same datasets, demonstrating consistent overlap with expert annotations. Compared with state-of-the-art approaches that commonly assume lesion-centered ROIs or rely on heavier backbones, the proposed pipeline addresses a more realistic scenario by performing automatic detection followed by segmentation while maintaining substantially lower computational requirements. This balance between performance and efficiency makes the MassSeg-Framework a promising tool for scalable mammography analysis, particularly in resource-constrained environments or high-throughput screening workflows that require rapid processing.Keywords
Funding Information
- the Universidad San Francisco de Quito USFQ for financial support through the Poli-Grants Program (17981)
- the Universidad San Francisco de Quito USFQ for financial support through the Poli-Grants Program (19031)
- the Universidad San Francisco de Quito USFQ for financial support through the Poli-Grants Program (24276)
This publication has 63 references indexed in Scilit:
- Features for Active Contour and Surface Segmentation: A ReviewArchives of Computational Methods in Engineering, 2025
- Transformer-based mass detection in digital mammogramsJournal of Ambient Intelligence and Humanized Computing, 2023
- Breast cancer: One-stage automated detection, segmentation, and classification of digital mammograms using UNet model based-semantic segmentationBiomedical Signal Processing and Control, 2021
- Image segmentation evaluation: a survey of methodsArtificial Intelligence Review, 2020
- An improved random forest-based rule extraction method for breast cancer diagnosisApplied Soft Computing, 2020
- Lightweight Deep Learning Pipeline for Detection, Segmentation and Classification of Breast Cancer Anomalies2019
- INbreastAcademic Radiology, 2012
- The Elements of Statistical Learning2001
- Breast cancer screening with mammography: overview of Swedish randomised trialsThe Lancet, 1993
- Snakes: Active contour modelsInternational Journal of Computer Vision, 1988