Performance improvement of Deep Learning Models using image augmentation techniques

doi:10.1007/s11042-021-11869-x

2022-01-24

PRE

M. Nagaraju

Priyanka Chawla *

Neeraj Kumar

DOI：10.1007/s11042-021-11869-x

原文链接

原文求助

摘要

En 中文

The major barrier while using deep learning models is lack of large number of images in the training dataset. In fact, there is a need of thousands of images in each image categories based on the complexity of problem. Prior studies have shown that picture augmentation techniques can be used to enhance the number of images in a training dataset artificially. These techniques can aid in improving the overall learning process and performance of a deep learning model. Hence, to address this problem we have proposed three algorithms. Firstly, two image acquisition algorithms have been proposed to systematically obtain real field images for testing and images from public datasets for training a model. Secondly, an algorithm is proposed to describe the procedure how the augmentations can be applied to enhance the datasets. During this study, we have investigated 52 augmentations that can allow enhancing the size of input dataset by improving the quantity of images. To perform the classification process of four maize crop diseases, a new convolutional neural network model is developed and several experiments have been performed to prove its effectiveness. Firstly, two tests were carried out using the original dataset from Kaggle public repository and the augmented dataset. When compared with the original dataset, the model improved by 5.14% with the augmented dataset. Secondly, three experiments carried out to evaluate the performance of proposed augmentation method. Experimental results demonstrated that the proposed approach outperforms the existing three approaches by 27.38%, 3.14%, and 1.34% during the classification process. The proposed IPA augmentation method has been compared with six existing methods: Full Stage Data Augmentation Framework, LeafGAN, Novel Augmentation method based on GAN, Wasserstein Generative Adversarial Network (WGAN), Activation Reconstruction-GAN, and Step-by-Step Data Augmentation Method and experimental results show that performance is better than existing methods by 28.31%, 19.76%, 20.18%, 13.75%, 2.42%, and 12.68% respectively.

Keyword:

Artificial intelligence

Augmentation techniques

CNN Models

Deep learning

Image acquisition

Image augmentations