| Literature DB >> 32210512 |
Muhammad Farhan Safdar1, Shayma Saad Alkobaisi2, Fatima Tuz Zahra3.
Abstract
INTRODUCTION: Machine Learning (ML) is a rapidly growing subfield of Artificial Intelligence (AI). It is used for different purposes in our daily life such as face recognition, speech recognition, text translation in different languages, weather prediction, and business prediction. In parallel, ML also plays an important role in the medical domain such as in medical imaging. ML has various algorithms that need to be trained with large volumes of data to produce a well-trained model for prediction. AIM: The aim of this study is to highlight the most suitable Data Augmentation (DA) technique(s) for medical imaging based on their results.Entities:
Keywords: Data Augmentation; Machine Learning; Medical Imaging
Year: 2020 PMID: 32210512 PMCID: PMC7085309 DOI: 10.5455/aim.2020.28.29-36
Source DB: PubMed Journal: Acta Inform Med ISSN: 0353-8109
Figure 1.Tumor on different regions of brain; (A) Tumor on Parietal lobe of the brain; (B) Tumor on Temporal lobe of the brain; (C) Tumor on Frontal lobe of the brain; (D) Tumor on Occipital lobe of the brain (13)
Figure 2.Architecture of YOLO v3 Model
Summary of Experiments
| Sr. # | Dataset | Avg. training loss from epochs 1-15 | Avg. evaluation loss from epochs | Training loss at epoch 15th | Evaluation loss at epoch 15th | IOU | Test Accuracy |
|---|---|---|---|---|---|---|---|
| 1 | Rotate 180o | 0.46 | 0.90 | 0.09 | 0.86 | 0.80 | 96% |
| 2 | Rotate 90o | 0.43 | 0.94 | 0.06 | 0.87 | 0.76 | 92% |
| 3 | Crop & Scale | 0.56 | 0.77 | 0.06 | 0.60 | 0.84 | 83% |
| 4 | Horizontal Flip | 0.46 | 0.96 | 0.07 | 0.84 | 0.80 | 72% |
| 5 | Vertical Flip | 0.43 | 0.96 | 0.08 | 1.33 | 0.80 | 70% |
| 6 | Original | 0.59 | 1.00 | 0.25 | 1.23 | 0.72 | 68% |
| 7 | Shear | 0.54 | 0.80 | 0.06 | 0.68 | 0.78 | 68% |
| 8 | Gaussian blur | 0.44 | 0.72 | 0.11 | 0.46 | 0.72 | 66% |
| 9 | Noise | 0.48 | 0.94 | 0.11 | 1.80 | 0.77 | 60% |
Comparison of results with prior studies
| Authors | Classification Problem | Methods | Data Augmentation | Accuracy |
|---|---|---|---|---|
| Shanchen Pang et al. ( | Cholilithiasis and gallstone from CT scan images | You Only Look Once (YOLO) | No | 92.7% |
| Joseph Redmon et al. ( | Natural objects | You Only Look Once (YOLO) | No | 65.5% |
| Data Augmentation & YOLO v3 | Low-grade glioma Brain tumor from MRI scan images | You Only Look Once (YOLO) | Yes | 96% |