Development of a transferring GAN based CAD scheme for breast mass classification: an initial study

Xuxin Chen; Theresa C. Thai; Camille C. Gunderson; Katheleen Moore; Robert S. Mannel; Hong Liu; Bin Zheng; Yuchen Qiu

doi:10.1117/12.2583246

5 March 2021 Development of a transferring GAN based CAD scheme for breast mass classification: an initial study

Xuxin Chen, Theresa C. Thai, Camille C. Gunderson, Katheleen Moore, Robert S. Mannel, Hong Liu, Bin Zheng, Yuchen Qiu

Proceedings Volume 11643, Biophotonics and Immune Responses XVI; 116430H (2021) https://doi.org/10.1117/12.2583246
Event: SPIE BiOS, 2021, Online Only

Abstract

The purpose of this study is to develop a novel computer-aided diagnosis (CAD) scheme to facilitate breast mass classification, which is based on the latest transferring generative adversarial networks (GAN) technology. Although GAN is one of the most popular techniques for image augmentation, it requires a relatively large original dataset to achieve satisfactory results, which may not be available for most of the medical imaging tasks. To address this challenge, we developed a novel transferring GAN, which was built based on the deep convolutional generative adversarial networks (DCGAN). This novel model was first pre-trained on a dataset of non-mass mammogram patches. Then the generator and the discriminator were fine-tuned on the mass dataset. A supervised loss was integrated with the discriminator, such that it can be used to directly classify the benign/malignant masses. We retrospectively assembled a total of 25,000 non-mass patches and 1024 mass images to assess this model, using classification accuracy and receiver operating characteristic (ROC) curve. The results demonstrated that our proposed approach improved the accuracy and area under the ROC curve (AUC) by 6.0% and 3.5% respectively, when compared with the classifiers trained without conventional data augmentation. This investigation may provide a new perspective for researchers to effectively train the GAN models on medical imaging tasks with limited datasets.

Conference Presentation

Citation Download Citation

Xuxin Chen, Theresa C. Thai, Camille C. Gunderson, Katheleen Moore, Robert S. Mannel, Hong Liu, Bin Zheng, and Yuchen Qiu "Development of a transferring GAN based CAD scheme for breast mass classification: an initial study", Proc. SPIE 11643, Biophotonics and Immune Responses XVI, 116430H (5 March 2021); https://doi.org/10.1117/12.2583246

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available