Tuning of data augmentation hyperparameters in deep learning to building construction image classification with small datasets

Int J Mach Learn Cybern. 2023;14(1):171-186. doi: 10.1007/s13042-022-01555-1. Epub 2022 Apr 13.

Abstract

Deep Learning methods have important applications in the building construction image classification field. One challenge of this application is Convolutional Neural Networks adoption in a small datasets. This paper proposes a rigorous methodology for tuning of Data Augmentation hyperparameters in Deep Learning to building construction image classification, especially to vegetation recognition in facades and roofs structure analysis. In order to do that, Logistic Regression models were used to analyze the performance of Convolutional Neural Networks trained from 128 combinations of transformations in the images. Experiments were carried out with three architectures of Deep Learning from the literature using the Keras library. The results show that the recommended configuration (Height Shift Range = 0.2; Width Shift Range = 0.2; Zoom Range =0.2) reached an accuracy of 95.6 % in the test step of first case study. In addition, the hyperparameters recommended by proposed method also achieved the best test results for second case study: 93.3 % .

Keywords: Building construction image classification; Convolutional neural networks; Data augmentation; Deep learning; Hyperparameter tuning.