Preprint Article Version 1 This version is not peer-reviewed

Data Augmentation of Breast Ultrasound Images Using Wasserstein Generative Adversarial Networks

Version 1 : Received: 26 October 2024 / Approved: 28 October 2024 / Online: 28 October 2024 (14:54:23 CET)

How to cite: Diyasa, I. G. S. M.; Humairah, S.; Puspaningrum, E. Y.; Durry, F. D.; Lestari, W. D.; Caesarendra, W. Data Augmentation of Breast Ultrasound Images Using Wasserstein Generative Adversarial Networks. Preprints 2024, 2024102205. https://doi.org/10.20944/preprints202410.2205.v1 Diyasa, I. G. S. M.; Humairah, S.; Puspaningrum, E. Y.; Durry, F. D.; Lestari, W. D.; Caesarendra, W. Data Augmentation of Breast Ultrasound Images Using Wasserstein Generative Adversarial Networks. Preprints 2024, 2024102205. https://doi.org/10.20944/preprints202410.2205.v1

Abstract

Breast cancer is among the most prevalent cancers in Indonesia. One of the methods for early detection of breast cancer is utilizing ultrasound images to classify breast conditions into normal, benign tumors, or cancer. The advent of deep learning technology facilitates the image analysis process, such as the use of Transfer Learning Convolutional Neural Networks (CNNs). Generally, CNN models require large and balanced datasets to perform well in classification tasks. However, medical datasets like breast ultrasound images tend to be limited and imbalanced. The Wasserstein GAN (WGAN) is a generative data augmentation method capable of producing synthetic images by learning the pattern of the distribution of real image data. The implementation of the Wasserstein distance results in the training process of WGAN demonstrating stability from epochs 3000, 2500, and 3000 out of a total of 5000 epochs. The quality of the synthetic images improves with an increasing number of training iterations. By using WGAN, all of the evaluation metrics of each classifier are increasing, with the best accuracy score is achieved by VGG16 model with 83.33% accuracy

Keywords

Wasserstein GAN; Breast Cancer; Ultrasound Image; Augmentation

Subject

Computer Science and Mathematics, Artificial Intelligence and Machine Learning

Comments (0)

We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.

Leave a public comment
Send a private comment to the author(s)
* All users must log in before leaving a comment
Views 0
Downloads 0
Comments 0


×
Alerts
Notify me about updates to this article or when a peer-reviewed version is published.
We use cookies on our website to ensure you get the best experience.
Read more about our cookies here.