22 February 2024 Conditional generative adversarial network based multi-task learning framework for breast lesion segmentation of ultrasound images
Satyajeet Datta, Mohammad Ariful Haque
Author Affiliations +
Abstract

The conditional generative adversarial network (cGAN) is widely used in image-to-image translation for natural images by implementing supervised learning loss in the GAN framework. It has also been extended to biomedical images for image segmentation. However, during the training phase, it lacks proper guidance for smooth convergence, and the model requires a large dataset, which is often unavailable for biomedical images. To address this limitation, we integrated a multi-task learning technique using a cGAN-based adversarial framework for lesion segmentation in ultrasound (US) breast images. Our proposition defines an auxiliary task of finding the mask similarity score between the input US image and the ground-truth segmentation mask, which is trained on features extracted from the available data for the segmentation task. This learning methodology is imposed within a cGAN framework to boost the original segmentation task of the generator and guide it through the auxiliary task. The proposed method shows improved convergence and stability compared to the core adversarial network, and its performance has been assessed on different datasets and model architectures, demonstrating significant improvement over the baseline.

© 2024 SPIE and IS&T
Satyajeet Datta and Mohammad Ariful Haque "Conditional generative adversarial network based multi-task learning framework for breast lesion segmentation of ultrasound images," Journal of Electronic Imaging 33(1), 013048 (22 February 2024). https://doi.org/10.1117/1.JEI.33.1.013048
Received: 24 June 2023; Accepted: 30 January 2024; Published: 22 February 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image segmentation

Gallium nitride

Network architectures

Data modeling

Education and training

Biomedical optics

Breast

Back to Top