Improving Brain Tumor Classification with Deep Learning UsingSyntheticData |
| |
| Authors: | Muhammed Mutlu Yapici Rukiye Karakis Kali Gurkahraman |
| |
| Affiliation: | 1.Department of Computer Science and Engineering, Lovely Professional University, Punjab, India2 Department of Artificial Intelligence and Big Data, Woosong University, Daejeon, 34606, Korea3 Center for ICT & Automotive Convergence, Kyungpook National University, Daegu, 41566, Korea4 School of Electronics Engineering, Kyungpook National University, Daegu, 41566, Korea |
| |
| Abstract: | Deep learning (DL) techniques, which do not need complex pre-processing and feature analysis, are used in many areas of medicine and achieve promising results. On the other hand, in medical studies, a limited dataset decreases the abstraction ability of the DL model. In this context, we aimed to produce synthetic brain images including three tumor types (glioma, meningioma, and pituitary), unlike traditional data augmentation methods, and classify them with DL. This study proposes a tumor classification model consisting of a Dense Convolutional Network (DenseNet121)-based DL model to prevent forgetting problems in deep networks and delay information flow between layers. By comparing models trained on two different datasets, we demonstrated the effect of synthetic images generated by Cycle Generative Adversarial Network (CycleGAN) on the generalization of DL. One model is trained only on the original dataset, while the other is trained on the combined dataset of synthetic and original images. Synthetic data generated by CycleGAN improved the best accuracy values for glioma, meningioma, and pituitary tumor classes from 0.9633, 0.9569, and 0.9904 to 0.9968, 0.9920, and 0.9952, respectively. The developed model using synthetic data obtained a higher accuracy value than the related studies in the literature. Additionally, except for pixel-level and affine transform data augmentation, synthetic data has been generated in the figshare brain dataset for the first time. |
| |
| Keywords: | Brain tumor classification deep learning cycle generative adversarial network data augmentation |
|
| 点击此处可从《计算机、材料和连续体（英文）》浏览原始摘要信息 |
|
点击此处可从《计算机、材料和连续体（英文）》下载全文 |
|
