Effect of hyper-parameters on the performance of ConvLSTM based deep neural network in crop classification

PLoS One. 2023 Feb 9;18(2):e0275653. doi: 10.1371/journal.pone.0275653. eCollection 2023.

Abstract

Deep learning based data driven methods with multi-sensors spectro-temporal data are widely used for pattern identification and land-cover classification in remote sensing domain. However, adjusting the right tuning for the deep learning models is extremely important as different parameter setting can alter the performance of the model. In our research work, we have evaluated the performance of Convolutional Long Short-Term Memory (ConvLSTM) and deep learning techniques, over various hyper-parameters setting for an imbalanced dataset and the one with highest performance is utilized for land-cover classification. The parameters that are considered for experimentation are; Batch size, Number of Layers in ConvLSTM model, and No of filters in each layer of the ConvLSTM are the parameters that will be considered for our experimentation. Experiments also have been conducted on LSTM model for comparison using the same hyper-parameters. It has been found that the two layered ConvLSTM model having 16-filters and a batch size of 128 outperforms other setting scenarios, with an overall validation accuracy of 97.71%. The accuracy achieved for the LSTM is 93.9% for training and 92.7% for testing.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Memory, Long-Term*
  • Neural Networks, Computer*

Grants and funding

This article is funded by the projects SP2022/18 and SP2022/34, assigned to VSB-Technical University of Ostrava, the Ministry of Education, Youth and Sports in Czech Republic. The funders have role in the formal analysis, investigation, review editing and validation of the manuscript, along with funding acquisition from their relevant institutions.