Joint Weakly and Semi-Supervised Deep Learning for Localization and Classification of Masses in Breast Ultrasound Images

by   Seung Yeon Shin, et al.

We propose a framework for localization and classification of masses in breast ultrasound (BUS) images. In particular, we simultaneously use a weakly annotated dataset and a relatively small strongly annotated dataset to train a convolutional neural network detector. We have experimentally found that mass detectors trained with small, strongly annotated datasets are easily overfitted, whereas those trained with large, weakly annotated datasets present a non-trivial problem. To overcome these problems, we jointly use datasets with different characteristics in a hybrid manner. Consequently, a sophisticated weakly and semi-supervised training scenario is introduced with appropriate training loss selection. Experimental results show that the proposed method successfully localizes and classifies masses while requiring less effort in annotation work. The influences of each component in the proposed framework are also validated by conducting an ablative analysis. Although the proposed method is intended for masses in BUS images, it can also be applied as a general framework to train computer-aided detection and diagnosis systems for a wide variety of image modalities, target organs, and diseases.


page 1

page 2

page 3

page 8

page 9


Gradually Applying Weakly Supervised and Active Learning for Mass Detection in Breast Ultrasound Images

We propose a method for effectively utilizing weakly annotated image dat...

Weakly Supervised Lesion Detection and Diagnosis for Breast Cancers with Partially Annotated Ultrasound Images

Deep learning (DL) has proven highly effective for ultrasound-based comp...

Joint localization and classification of breast tumors on ultrasound images using a novel auxiliary attention-based framework

Automatic breast lesion detection and classification is an important tas...

WSSOD: A New Pipeline for Weakly- and Semi-Supervised Object Detection

The performance of object detection, to a great extent, depends on the a...

Rapid Training Data Creation by Synthesizing Medical Images for Classification and Localization

While the use of artificial intelligence (AI) for medical image analysis...

Virtual Adversarial Training for Semi-supervised Breast Mass Classification

This study aims to develop a novel computer-aided diagnosis (CAD) scheme...

Please sign up or login with your details

Forgot password? Click here to reset