Synthetic Ultrasound Image Generation for Breast Cancer Diagnosis Using cVAE-WGAN Models: An Approach Based on Generative Artificial Intelligence
Mondillo, G.; Masino, M.; Colosimo, S.; Perrotta, A.; Frattolillo, V.; Abbate, F. G.
Show abstract
The scarcity and imbalance of medical image datasets hinder the development of robust computer-aided diagnosis (CAD) systems for breast cancer. This study explores the application of advanced generative models, based on generative artificial intelligence (GenAI), for the synthesis of digital breast ultrasound images. Using a hybrid Conditional Variational Autoencoder-Wasserstein Generative Adversarial Network (CVAE-WGAN) architecture, we developed a system to generate high-quality synthetic images conditioned on the class (malignant vs. normal/benign). These synthetic images, generated from the low-resolution BreastMNIST dataset and filtered for quality, were systematically integrated with real training data at different mixing ratios (W). The performance of a CNN classifier trained on these mixed datasets was evaluated against a baseline model trained only on real data balanced with SMOTE. The optimal integration (mixing weight W=0.25) produced a significant performance increase on the real test set: +8.17% in macro-average F1-score and +4.58% in accuracy compared to using real data alone. Analysis confirmed the originality of the generated samples. This approach offers a promising solution for overcoming data limitations in image-based breast cancer diagnostics, potentially improving the capabilities of CAD systems.
Matching journals
The top 10 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Probabilistic Brain MR Image Transformation Using Generative Models 95%
- Incremental Learning Approach for Semantic Segmentation of Skin Histology Images 95%
- Segmentation of Pancreatic Ductal Adenocarcinoma (PDAC) and surrounding vessels in CT images using deep convolutional neural networks and Texture Descriptors 94%
Similar papers in this journal
- Enhancing Breast Ultrasound Segmentation through Fine-tuning and Optimization Techniques: Sharp Attention UNet 97%
- Fibre tracing in biomedical images: An objective comparison between seven algorithms 96%
- Automated Detection of COVID-19 through Convolutional Neural Network using Chest x-ray images 95%
Similar papers in this journal
- Model uncertainty estimates for deep learning mammographic density prediction using ordinal and classification approaches 92%
- Breast density prediction from low and standard dose mammograms using deep learning: effect of image resolution and model training approach on prediction quality 92%
- Recent advances in the clinical applications of machine learning in proton therapy 89%
Similar papers in this journal
- SCU-Net: A deep learning method for segmentation and quantification of breast arterial calcifications on mammograms 96%
- Phase Recognition in Contrast-Enhanced CT Scans based on Deep Learning and Random Sampling 94%
- Fully Automated Explainable Abdominal CT Contrast Media Phase Classification Using Organ Segmentation and Machine Learning 94%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.