Finding the Most Effective Data Augmentation Techniques on Brain MRI Data Using Deep Networks
Keywords:
Data Augmentation, Brain MRI, Deep NetworksAbstract
In 2020, over 250,000 people died from brain and Central Nervous System (CNS) tumors. Brain tumors account for around 85% or more of these. As of 2021, over 16 million people in the United States have been diagnosed with some type of cognitive impairment. The goal of this paper is to find the most effective set of data augmentations to correctly classify cognitive diseases with deep networks, using structural Magnetic Resonance Imaging (MRI) data. This paper demonstrates a Greedy optimization technique to find the most effective sequence of data augmentations out of blurring, distortion, position, and red noise (an overlay augmentation displaying random clouds of noise on the images). We sought to classify three tumors: glioma tumors, pituitary tumors, and meningioma tumors, as well as detect if there was no tumor at all. We also classified three stages of Alzheimer’s disease: not demented, very mildly demented, and mildly demented, to further demonstrate the effectiveness of the data augmentation sequence.
References or Bibliography
[Ama93] Shun-ichi Amari. Backpropagation and stochastic gradient descent method. Neurocom-
puting, 5(4-5):185–196, 1993.
[BIK+20] Alexander Buslaev, Vladimir I Iglovikov, Eugene Khvedchenya, Alex Parinov, Mikhail
Druzhinin, and Alexandr A Kalinin. Albumentations: fast and flexible image augmen-
tations. Information, 11(2):125, 2020.
[BRH19] Marcus D Bloice, Peter M Roth, and Andreas Holzinger. Biomedical image augmenta-
tion using augmentor. Bioinformatics, 35(21):4522–4524, 2019.
[BSH17] Marcus D Bloice, Christof Stocker, and Andreas Holzinger. Augmentor: an image
augmentation library for machine learning. arXiv preprint arXiv:1708.04680, 2017.
[can22] Brain tumor - statistics, Feb 2022.
[cdc22] Faststats - leading causes of death, Jan 2022.
[HZRS16] Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning
for image recognition. In Proceedings of the IEEE conference on computer vision and
pattern recognition, pages 770–778, 2016.
[joh] Brain tumors and brain cancer.
[KB14] Diederik P Kingma and Jimmy Ba. Adam: A method for stochastic optimization. arXiv
preprint arXiv:1412.6980, 2014.
[nat20] Magnetic resonance imaging (mri), 2020.
[Nic21] Msoud Nickparvar. Brain tumor mri dataset, 2021.
[Nus81] Henri J Nussbaumer. The fast fourier transform. In Fast Fourier Transform and Convo-
lution Algorithms, pages 80–111. Springer, 1981.
[Pin22] Marco Pinamonti. Alzheimer mri 4 classes dataset, Jan 2022.
[PPLF17] Justin S Paul, Andrew J Plassard, Bennett A Landman, and Daniel Fabbri. Deep learning
for brain tumor classification. In Medical Imaging 2017: Biomedical Applications in
Molecular, Structural, and Functional Imaging, volume 10137, pages 253–268. SPIE,
[RDS+15] Olga Russakovsky, Jia Deng, Hao Su, Jonathan Krause, Sanjeev Satheesh, Sean Ma,
Zhiheng Huang, Andrej Karpathy, Aditya Khosla, Michael Bernstein, et al. Imagenet
large scale visual recognition challenge. International journal of computer vision,
(3):211–252, 2015.
[SAZ20] Muhammad Farhan Safdar, Shayma Saad Alkobaisi, and Fatima Tuz Zahra. A compar-
ative analysis of data augmentation approaches for magnetic resonance imaging (mri)
scan images of brain tumor. Acta informatica medica, 28(1):29, 2020.
[ten] Module: Tf.keras nbsp;: nbsp; tensorflow v2.10.0.
[TL19] Mingxing Tan and Quoc Le. Efficientnet: Rethinking model scaling for convolutional
neural networks. In International conference on machine learning, pages 6105–6114.
PMLR, 2019.
Downloads
Posted
License
Copyright (c) 2023 Nishank Raisinghani
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
The copyright holder for this article has granted JSR.org a license to display the article in perpetuity.
