Als: Active Learning-Based Image Segmentation Model for Skin Lesion

Xiu Shu,Yunyun Yang,Ruicheng Xie,Jun Liu,Xiaojun Chang,Boying Wu
DOI: https://doi.org/10.2139/ssrn.4141765
2022-01-01
SSRN Electronic Journal
Abstract:Medical image segmentation requires experienced doctors to provide accurate segmentation results as ground truth, since it involves plentiful medical knowledge. The deep learning-based model requires a large number of medical images and ground truth, during the training stage, which brings a heavy workload to the doctor. Therefore, based on the differences between medical images, we designed an active learning model to select more representative images for doctors to obtain the ground truth, which saves their time in segmenting similar medical images. Eventually, these representative images will serve as training sets and test sets, training a more accurate segmentation model. In this paper, we propose the active learning-based segmentation (ALS) model for medical images, in which we use the mean structure similarity index measure to measure how similar the two images are. Then we select some highly similar images from the entire training set. To our knowledge, we are the first to apply the active learning strategy based on the mean structure similarity index measure to medical image segmentation. Finally, we reduce the amount of data to 75%, 50%, 25%, and 12.5% to validate our ALS model. We use the selected images after active learning training and the medical semantic segmentation model to segment medical images, and the active learning model can reduce the cost of labeling and obtain accurate segmentation results through experiments, and even better than the segmentation results with a large number of highly similar data.
What problem does this paper attempt to address?