DACov: A Deeper Analysis of Data Augmentation on the Computed Tomography Segmentation Problem

Abstract

Due to the COVID-19 global pandemic, computer-assisted diagnoses of medical images have gained much attention, and robust methods of semantic segmentation of Computed Tomography (CT) images have become highly desirable. In this work, we present a deeper analysis of how data augmentation techniques improve segmentation performance on this problem. We evaluate 20 traditional augmentation techniques on five public datasets. Six different probabilities of applying each augmentation technique on an image were evaluated. We also assess a different training methodology where the training subsets are combined into a single larger set. All networks were evaluated through a 5-fold cross-validation strategy, resulting in over 4,600 experiments. We also propose a novel data augmentation technique based on Generative Adversarial Networks (GANs) to create new healthy and unhealthy lung CT images, evaluating four variations of our approach with the same six probabilities of the traditional methods. Our findings show that GAN-based techniques and spatial-level transformations are the most promising for improving the learning of deep models on this problem, with the StarGANv2 + F with a probability of 0.3 achieving the highest F-score value on the Ricord1a dataset in the unified training strategy. Our code is publicly available at https://github.com/VRI-UFPR/DACov2022

References in corpus (15)

• Conditional Generative Adversarial Nets
• Albumentations: fast and flexible image augmentations
• Automatic Detection of Coronavirus Disease (COVID-19) Using X-ray Images and Deep Convolutional Neural Networks
• Review of Artificial Intelligence Techniques in Imaging Data Acquisition, Segmentation and Diagnosis for COVID-19
• Training Generative Adversarial Networks with Limited Data
• GridMask Data Augmentation
• Residual Attention U-Net for Automated Multi-Class Segmentation of COVID-19 Chest CT Images
• On the Cross-dataset Generalization in License Plate Recognition
• Towards Image-based Automatic Meter Reading in Unconstrained Scenarios: A Robust and Efficient Approach
• CovTANet: A Hybrid Tri-level Attention Based Network for Lesion Segmentation, Diagnosis, and Severity Prediction of COVID-19 Chest CT Scans
• MosMedData: Chest CT Scans With COVID-19 Related Findings Dataset
• A First Look at Dataset Bias in License Plate Recognition
• GASNet: Weakly-supervised Framework for COVID-19 Lesion Segmentation
• IDA: Improved Data Augmentation Applied to Salient Object Detection
• D2A U-Net: Automatic Segmentation of COVID-19 Lesions from CT Slices with Dilated Convolution and Dual Attention Mechanism

Cited by in corpus (1)

• Generative AI for Synthetic Data Across Multiple Medical Modalities: A Systematic Review of Recent Developments and Challenges