Classification Task Assisted Segmentation Network for Breast Tumor Segmentation in Ultrasound Images

• Published in: 2023 IEEE International Conference on Image Processing (ICIP) pp. 3294 - 3298
• Main Authors: Zhang, Kunkun, Wang, Bin
• Format: Conference Proceeding
• Language: English
• Published: IEEE 08.10.2023
• Subjects: Auxiliary-task learning; breast ultrasound imaging; Image segmentation; Location awareness; Measurement; multi-task learning; semantic segmentation; Semantics; Shape; Speckle; Ultrasonic imaging
• DOI: 10.1109/ICIP49359.2023.10222770

Accurate and effective segmentation of breast masses plays an important role in the early stages of breast cancer treatment. However, the irregular shape of mass, the blurred mass boundary and speckle noise in breast ultrasound (BUS) images make automatic segmentation still challenging. In this paper, we propose a classification task assist (CTA) module for boosting the performance of the commonly used deep segmentation models on BUS images. This module can be easily inserted into representative segmentation models to focus on a learning task of BUS image-level classification. The incorporation of this classification learning task enable the segmentation model achieve additional supervision information to enhance the extraction of semantic context information and the localization of segmentation object. We added the CTA module to four state-of-the-art segmentation models, including codec and non-codec structures, and the experimental results show that the insertion of the CTA module into the original model results in the lowest improvement of 1.48% and the highest improvement of 4.29% in the intersection over union (IoU) metric.