Publicação
Impact of hyper-parameter tuning on CNN accuracy in agricultural image classification
| Resumo: | This study explores the impact of hyper-parameter optimization on the performance of convolutional neural networks (CNNs) for olive cultivar classification using transfer learning. Pre-trained ImageNet models such as VGG16, InceptionV3, and ResNet50 were adapted to a proprietary dataset, with VGG16 selected for detailed evaluation. Key hyper-parameters, including layer count, neurons per layer, dropout rate, learning rate, and batch size, were tuned using random search. The best configuration achieved a validation accuracy of 87.5%, significantly outperforming the control model. Sensitivity analyses with Morris and Sobol methods identified the number of layers as the most influential factor, followed by dropout and learning rates through interaction effects. These findings demonstrate the importance of tailoring CNN architecture and regularization settings to the problem domain. These results underscore the importance of tuning architectural depth and regularization mechanisms for performance optimization. As a practical guideline, models with fewer layers and intermediate dropout levels demonstrated higher robustness and generalization, offering an effective strategy for adapting CNNs to agricultural classification tasks. |
|---|---|
| Autores principais: | Mendes, João |
| Outros Autores: | Lima, José; Costa, Lino; Hendrix, Eligius M.T.; Pereira, Ana I. |
| Assunto: | Hyper-parameter optimization Convolutional neural networks Sensitivity analysis |
| Ano: | 2025 |
| País: | Portugal |
| Tipo de documento: | artigo |
| Tipo de acesso: | acesso aberto |
| Instituição associada: | Instituto Politécnico de Bragança |
| Idioma: | inglês |
| Origem: | Biblioteca Digital do IPB |
| Resumo: | This study explores the impact of hyper-parameter optimization on the performance of convolutional neural networks (CNNs) for olive cultivar classification using transfer learning. Pre-trained ImageNet models such as VGG16, InceptionV3, and ResNet50 were adapted to a proprietary dataset, with VGG16 selected for detailed evaluation. Key hyper-parameters, including layer count, neurons per layer, dropout rate, learning rate, and batch size, were tuned using random search. The best configuration achieved a validation accuracy of 87.5%, significantly outperforming the control model. Sensitivity analyses with Morris and Sobol methods identified the number of layers as the most influential factor, followed by dropout and learning rates through interaction effects. These findings demonstrate the importance of tailoring CNN architecture and regularization settings to the problem domain. These results underscore the importance of tuning architectural depth and regularization mechanisms for performance optimization. As a practical guideline, models with fewer layers and intermediate dropout levels demonstrated higher robustness and generalization, offering an effective strategy for adapting CNNs to agricultural classification tasks. |
|---|