Mammographic Lesion Segmentation with Lightweight Models: A Comparative Study
For developers of computer-aided diagnosis systems in mammography, this work provides a benchmark of efficient models, though the performance is modest and incremental over existing methods.
This study compares lightweight deep learning models for mammographic lesion segmentation, finding that MobileNetV2 with Squeeze-and-Excitation achieves a Dice score of 0.5766 with 75% fewer parameters than U-Net, offering a practical trade-off for resource-constrained environments.
Breast cancer is a leading cause of cancer-related mortality among women worldwide, with mammography as the primary screening tool. While deep learning models have shown strong performance in lesion segmentation, most rely on computationally intensive architectures that limit their use in resource-constrained environments. This study evaluates the performance and efficiency of lightweight models for mammographic lesion segmentation. Architectures including MobileNetV2, EfficientNet Lite, ENet, and Fast-SCNN were compared against a U-Net baseline using the INbreast dataset with 5-fold cross-validation. Performance was assessed using Dice score, Intersection over Union (IoU), and Recall, alongside model complexity. MobileNetV2 with Squeeze-and-Excitation (SCSE) achieved the best performance, with a Dice score of 0.5766 while using approximately 75\% fewer parameters than U-Net. Cross-dataset evaluation on the DMID dataset showed reduced accuracy due to domain shift but preserved recall. These results demonstrate that lightweight architectures offer a practical balance between performance and efficiency for deployable CAD systems.