Hyperparameter tuning of convolutional neural networks for building construction image classification

被引:6
|
作者
Carvalho Ottoni, Andre Luiz [1 ]
Novo, Marcela Silva [2 ]
Costa, Dayana Bastos [3 ]
机构
[1] Univ Fed Reconcavo Bahia, Technol & Exact Ctr, Cruz Das Almas, Brazil
[2] Univ Fed Bahia, Dept Elect & Comp Engn, Salvador, BA, Brazil
[3] Univ Fed Bahia, Dept Struct & Construct Engn, Salvador, BA, Brazil
来源
VISUAL COMPUTER | 2023年 / 39卷 / 03期
关键词
Deep learning; Convolutional neural networks; Hyperparameter tuning; Scott-Knott method; Building construction image classification; LEARNING APPROACH; DEEP; SYSTEM;
D O I
10.1007/s00371-021-02350-9
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
Deep Learning models have important applications in image processing. However, one of the challenges in this field is the definition of hyperparameters. Thus, the objective of this work is to propose a rigorous methodology for hyperparameter tuning of Convolutional Neural Network for building construction image classification, especially in roofs structure analysis. For this, the HyperTuningSK algorithm was developed, intended to create recommendation rankings for two hyperparameters: learning rate and optimizer. The approach uses concepts from the statistical design of experiments, such as Analysis of Variance and the Scott-Knott clustering algorithm. In addition, the adopted database includes images of inspections on buildings roofs made with unmanned aerial vehicles. The images are divided into two classes: (i) roofs with clean gutters and (ii) roofs with dirty gutters. The methods recommended by the HyperTuningSK algorithm achieved good results in comparison to the hyperparameters adopted in the literature. In this respect, adagrad015 achieved the highest average values of accuracy in the validation (100%) and testing steps (90%) for Convolutional Neural Network architecture with 12 layers. In addition, the hyperparameters recommended by the HyperTuningSK algorithm achieved the best test results for other two literature architectures: Densenet121 (85.7%) and VGG16 (84.4%).
引用
收藏
页码:847 / 861
页数:15
相关论文
共 50 条
  • [1] Hyperparameter tuning of convolutional neural networks for building construction image classification
    André Luiz Carvalho Ottoni
    Marcela Silva Novo
    Dayana Bastos Costa
    [J]. The Visual Computer, 2023, 39 : 847 - 861
  • [2] Hyperparameter tuning in convolutional neural networks for domain adaptation in sentiment classification (HTCNN-DASC)
    Krishnakumari, K.
    Sivasankar, E.
    Radhakrishnan, Sam
    [J]. SOFT COMPUTING, 2020, 24 (05) : 3511 - 3527
  • [3] Hyperparameter tuning in convolutional neural networks for domain adaptation in sentiment classification (HTCNN-DASC)
    K. Krishnakumari
    E. Sivasankar
    Sam Radhakrishnan
    [J]. Soft Computing, 2020, 24 : 3511 - 3527
  • [4] Hyperparameter Importance for Image Classification by Residual Neural Networks
    Sharma, Abhinav
    van Rijn, Jan N.
    Hutter, Frank
    Muller, Andreas
    [J]. DISCOVERY SCIENCE (DS 2019), 2019, 11828 : 112 - 126
  • [5] Convolutional Neural Networks for image classification
    Jmour, Nadia
    Zayen, Sehla
    Abdelkrim, Afef
    [J]. 2018 INTERNATIONAL CONFERENCE ON ADVANCED SYSTEMS AND ELECTRICAL TECHNOLOGIES (IC_ASET), 2017, : 397 - 402
  • [6] Convolutional Neural Networks for Multi-scale Lung Nodule Classification in CT: Influence of Hyperparameter Tuning on Performance
    Hernandez-Rodriguez, Jorge
    Cabrero-Fraile, Francisco-Javier
    Rodriguez-Conde, Maria-Jose
    [J]. TEM JOURNAL-TECHNOLOGY EDUCATION MANAGEMENT INFORMATICS, 2022, 11 (01): : 297 - 306
  • [7] CONVOLUTIONAL NEURAL NETWORKS IN THE TASK OF IMAGE CLASSIFICATION
    Zelenina, Larisa
    Khaimina, Liudmila
    Khaimin, Evgenii
    Khripunov, D.
    Zashikhina, Inga
    [J]. MATHEMATICS AND INFORMATICS, 2022, 65 (01): : 19 - 29
  • [8] Convolutional neural networks for hyperspectral image classification
    Yu, Shiqi
    Jia, Sen
    Xu, Chunyan
    [J]. NEUROCOMPUTING, 2017, 219 : 88 - 98
  • [9] Convolutional Neural Networks for Document Image Classification
    Kang, Le
    Kumar, Jayant
    Ye, Peng
    Li, Yi
    Doermann, David
    [J]. 2014 22ND INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2014, : 3168 - 3172
  • [10] Hyperspectral Image Classification with Convolutional Neural Networks
    Slavkovikj, Viktor
    Verstockt, Steven
    De Neve, Wesley
    Van Hoecke, Sofie
    Van de Walle, Rik
    [J]. MM'15: PROCEEDINGS OF THE 2015 ACM MULTIMEDIA CONFERENCE, 2015, : 1159 - 1162