Volume 32, Issue 4 pp. 1209-1220
RESEARCH ARTICLE

CR-SSL: A closely related self-supervised learning based approach for improving breast ultrasound tumor segmentation

Arnab Kumar Mishra

Corresponding Author

Arnab Kumar Mishra

Department of CSE, National Institute of Technology Silchar, Silchar, India

Correspondence

Arnab Kumar Mishra, Department of CSE, National Institute of Technology Silchar, Silchar, Assam 788010, India.

Email: [email protected]

Search for more papers by this author
Pinki Roy

Pinki Roy

Department of CSE, National Institute of Technology Silchar, Silchar, India

Search for more papers by this author
Sivaji Bandyopadhyay

Sivaji Bandyopadhyay

Department of CSE, National Institute of Technology Silchar, Silchar, India

Search for more papers by this author
Sujit Kumar Das

Sujit Kumar Das

Department of CSE, National Institute of Technology Silchar, Silchar, India

Search for more papers by this author
First published: 27 December 2021
Citations: 1

Abstract

Breast ultrasound (BUS) tumor segmentation can help with the early detection of breast cancer; however, the lack of human-labeled training data is a big problem in this area. To deal with this, we propose a novel self-supervised learning approach (CR-SSL), where several related pretext tasks like unsupervised segmentation and edge detection are firstly learned, followed by the target tumor segmentation based fine-tuning. Learning such related pretext tasks ensures better representation learning. Experimental study shows that CR-SSL can improve the mean Dice and Jaccard scores by more than 4–5%, with ≥0.6 scores while having access to only 20–50 human-labeled training samples. The best Dice scores of 0.7303 and 0.8207, and Jaccard scores of 0.7082 and 0.8015 are obtained by CR-SSL on the test datasets. Compared with the No-SSL baseline, CR-SSL can achieve (10–20)% improvements in segmentation quality while working in small-sized training dataset scenarios, suggesting its high potential practical utility.

CONFLICT OF INTEREST

The authors declare no conflicts of interest.

DATA AVAILABILITY STATEMENT

Two publicly available Breast Ultrasound Image datasets: BUSI and UDIAT are used in this work, both of which can be accessed freely by following appropriate procedures as mentioned in References 28 and 29, respectively.

The full text of this article hosted at iucr.org is unavailable due to technical difficulties.