A Green Solution for Breast Region Segmentation Using Deep Active Learning

TL;DR

This paper introduces a deep active learning approach with a novel sample selection strategy based on breast anatomy geometry to efficiently segment breast regions in MRI images, reducing training data and computational costs.

Contribution

It proposes a new sample selection method using breast anatomy geometry analysis within deep active learning for improved segmentation efficiency.

Findings

Nearest Point strategy achieved lowest carbon footprint at higher data proportions.

Increasing training data improved segmentation performance across strategies.

Combining Nearest Point with 30% data offers optimal balance of accuracy and efficiency.

Abstract

Purpose: Annotation of medical breast images is an essential step toward better diagnostic but a time consuming task. This research aims to focus on different selecting sample strategies within deep active learning on Breast Region Segmentation (BRS) to lessen computational cost of training and effective use of resources. Methods: The Stavanger breast MRI dataset containing 59 patients was used in this study, with FCN-ResNet50 adopted as a sustainable deep learning (DL) model. A novel sample selection approach based on Breast Anatomy Geometry (BAG) analysis was introduced to group data with similar informative features for DL. Patient positioning and Breast Size were considered the key selection criteria in this process. Four selection strategies including Random Selection, Nearest Point, Breast Size, and a hybrid of all three strategies were evaluated using an active learning framework. Four training data proportions of 10%, 20%, 30%, and 40% were used for model training, with the remaining data reserved for testing. Model performance was assessed using Dice score, Intersection over Union, precision, and recall, along with 5-fold cross-validation to enhance generalizability. Results: Increasing the training data proportion from 10% to 40% improved segmentation performance for nearly all strategies, except for Random Selection. The Nearest Point strategy consistently achieved the lowest carbon footprint at 30% and 40% data proportions. Overall, combining the Nearest Point strategy with 30% of the training data provided the best balance between segmentation performance, efficiency, and environmental sustainability. Keywords: Deep Active Learning, Breast Region Segmentation, Human-center analysis