Yıl: 2020 Cilt: 8 Sayı: 2 Sayfa Aralığı: 116 - 120 Metin Dili: İngilizce İndeks Tarihi: 31-10-2020

Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem

Öz:
In the last decade, deep learning methods have become the key solution for various machine learning problems. One major drawback of deep learning methods is that they require large datasets to have a good generalization performance. Researchers propose data augmentation techniques for generating synthetic data to overcome this problem. Traditional methods, such as flipping, rotation etc., which are referred as transformation based methods in this study are commonly used for obtaining synthetic data in the literature. These methods take as input an image and process that image to obtain a new one. On the other hand, generative models such as generative adversarial networks, auto-encoders, after trained with aset of image learn to generatesyntheticdata. Recently generative models are commonly used for data augmentation in various domains. In this study, we evaluate the effectiveness of a generative model, variational autoencoders (VAE), on the image classification problem. For this purpose, we train a VAE using CIFAR-10 dataset and generate synthetic samples with this model. We evaluate the classification performance using various sized datasets and compare the classification performances on four datasets; dataset without augmentation, dataset augmented with VAE and two datasets augmented with transformation based methods. We observe that the contribution of data augmentation is sensitive to the size of the dataset and VAE augmentation is as effective as the transformation based augmentation methods.
Anahtar Kelime:

Belge Türü: Makale Makale Türü: Araştırma Makalesi Erişim Türü: Erişime Açık
  • [1]M. Frid-Adar, E. Klang, M. Amitai, J. Goldberger and H. Greenspan, "Synthetic data augmentation using GAN for improved liver lesion classification," 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018), Washington.
  • [2]Y. Xu, Y. Zhang, H.Wang and X. Liu, "Underwater image classification using deep convolutional neural networks and data augmentation," 2017 IEEE International Conference on Signal Processing, Communications and Computing (ICSPCC), Xiamen, 2017, pp. 1-5.
  • [3]M. Zhang, Z. Cui, X. Wang and Z. Cao, "Data Augmentation Method of SAR Image Dataset,"IGARSS 2018 -2018 IEEE International Geoscience and Remote Sensing Symposium, Valencia, 2018, pp. 5292-5295.
  • [4]K. Chen, X. Zhou, Q. Zhou and H. Xu, "Adversarial Learning-based Data Augmentation for Rotation-robust Human Tracking,"ICASSP 2019 -2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Brighton, United Kingdom, 2019, pp. 1942-1946
  • [5]Shorten, C., Khoshgoftaar, T.M. A survey on Image Data Augmentation for Deep Learning.J Big Data6,60 (2019). https://doi.org/10.1186/s40537-019-0197-0
  • [6]J. Shijie, W. Ping, J. Peiyi and H. Siping, "Research on data augmentation for image classification based on convolution neural networks," 2017 Chinese Automation Congress (CAC), Jinan, 2017, pp. 4165-4170.
  • [7]C. Ge, I. Y. Gu, A. Store Jakola and J. Yang, "Cross-Modality Augmentation of Brain Mr Images Using a Novel Pairwise Generative Adversarial Network for Enhanced Glioma Classification,"2019 IEEE International Conference on Image Processing (ICIP), Taipei, Taiwan, 2019, pp. 559-563.
  • [8]A. H. Ornek and M. Ceylan, "Comparison of Traditional Transformations for Data Augmentation in Deep Learning of Medical Thermography,"2019 42nd International Conference on Telecommunications and Signal Processing (TSP), Budapest, Hungary, 2019, pp. 191-194.
  • [9]Kingma, D.P., Welling, M. (2014). "Auto-Encoding Variational Bayes", CoRR, abs/1312.6114.
  • [10]Jorge, J., Vieco, J., Paredes, R., Snchez, J., Bened, J. (2018). Empirical evaluation of variational autoencoders for data augmentation. VISIGRAPP.
  • [11]Perez, L., Wang, J. (2017). The effectiveness of data augmentation in image classification using deep learning. CoRR, abs/1712.04621.
  • [12]Z. Zhong, L. Zheng, G. Kang, S. Li, Y. Yang, ‘Random Erasing Data Augmentation’. ArXiv e-prints 2017.[13]Terrance V, Graham WT. Improved regularization of convolutional neural networks with cutout. arXiv preprint. 2017.
  • [14]Alex Krizhevsky, "Learning Multiple Layers of Features from Tiny Images", Technical Report, 2009.
  • [15]Krizhevsky, Alex; Sutskever, Ilya; Hinton, Geofrey E. (2017-05-24). "ImageNet classification with deep convolutional neural networks" (PDF). Communications of the ACM. 60 (6): 8490.
  • [16]Y. Lecun, L. Bottou, Y. Bengio and P. Ha_ner, "Gradient-based learning applied to document recognition," in Proceedings of the IEEE, vol. 86, no. 11, pp. 2278-2324, Nov. 1998.
  • [17]J. Shijie, W. Ping, J. Peiyi and H. Siping, "Research on data augmentation for image classification based on convolution neural networks,"2017 Chinese Automation Congress (CAC), Jinan, 2017, pp. 4165-4170.
  • [18]A. Merchant, T. Syed, B. Khan and R. Munir, "Appearance-based data augmentation for image datasets using contrast preserving sampling,"2018 24th International Conference on Pattern Recognition (ICPR), Beijing, 2018, pp. 1235-1240.
  • [19]X. Liuet al., "Data Augmentation via Latent Space Interpolation for Image Classification,"2018 24th International Conference on Pattern Recognition (ICPR), Beijing, 2018, pp. 728-733.
  • [20]E. D. Cubuk, B. Zoph, D. Mané, V. Vasudevan and Q. V. Le, "AutoAugment: Learning Augmentation Strategies From Data,"2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 2019, pp. 113-123.
  • [21]R. Takahashi, T. Matsubara and K. Uehara, "Data Augmentation using Random Image Cropping and Patching for Deep CNNs," inIEEE Transactions on Circuits and Systems for Video Technology.
  • [22]https://imgaug.readthedocs.io/, last accessed in March 2020.
APA OZTIMUR KARADAG O, Erdas Cicek O (2020). Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. , 116 - 120.
Chicago OZTIMUR KARADAG OZGE,Erdas Cicek Ozlem Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. (2020): 116 - 120.
MLA OZTIMUR KARADAG OZGE,Erdas Cicek Ozlem Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. , 2020, ss.116 - 120.
AMA OZTIMUR KARADAG O,Erdas Cicek O Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. . 2020; 116 - 120.
Vancouver OZTIMUR KARADAG O,Erdas Cicek O Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. . 2020; 116 - 120.
IEEE OZTIMUR KARADAG O,Erdas Cicek O "Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem." , ss.116 - 120, 2020.
ISNAD OZTIMUR KARADAG, OZGE - Erdas Cicek, Ozlem. "Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem". (2020), 116-120.
APA OZTIMUR KARADAG O, Erdas Cicek O (2020). Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. International Journal of Intelligent Systems and Applications in Engineering, 8(2), 116 - 120.
Chicago OZTIMUR KARADAG OZGE,Erdas Cicek Ozlem Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. International Journal of Intelligent Systems and Applications in Engineering 8, no.2 (2020): 116 - 120.
MLA OZTIMUR KARADAG OZGE,Erdas Cicek Ozlem Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. International Journal of Intelligent Systems and Applications in Engineering, vol.8, no.2, 2020, ss.116 - 120.
AMA OZTIMUR KARADAG O,Erdas Cicek O Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. International Journal of Intelligent Systems and Applications in Engineering. 2020; 8(2): 116 - 120.
Vancouver OZTIMUR KARADAG O,Erdas Cicek O Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem. International Journal of Intelligent Systems and Applications in Engineering. 2020; 8(2): 116 - 120.
IEEE OZTIMUR KARADAG O,Erdas Cicek O "Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem." International Journal of Intelligent Systems and Applications in Engineering, 8, ss.116 - 120, 2020.
ISNAD OZTIMUR KARADAG, OZGE - Erdas Cicek, Ozlem. "Empirical Evaluation of the Effectiveness of Variational Autoencoders on Data Augmentation for the Image Classification Problem". International Journal of Intelligent Systems and Applications in Engineering 8/2 (2020), 116-120.