Research Article Open Access

Automated Terrain Classification with a Bayesian Hyperparameter Optimized Deep Supervised Autoencoder Model

Tiny Du Toit1, Hennie Kruger2 and Annette Van Der Merwe2
  • 1 School of Computer Science and Information Systems, Faculty of Natural and Agricultural Sciences, North-West University, Private Bag X6001, Potchefstroom, 2520, South Africa
  • 2 School of Computer Science and Information Systems, Faculty of Natural and Agricultural Sciences, North-West University, Private Bag X6001, Potchefstroom, 2520, South Africa

Abstract

Terrain classification according to specific terrain attributes, has become increasingly important in certain decision-making scenarios. Automated robots are often utilized to traverse a specific surface to collect data that can be used in classification models to identify a specific terrain. In this study, a supervised autoencoder model (i.e., an autoencoder combined with a supervised learner such as a multilayer perceptron) is proposed to perform the classification of different terrains. Furthermore, a Bayes hyperparameter optimization approach is employed to determine optimum hyperparameter values. The dataset used for model building and training was obtained by driving a Lego Mindstorm EV3 mobile robot, fitted with a Raspberry Pi computer and a Sense HAT inertial measurement unit over six different terrain surfaces, i.e., asphalt, dirt, epoxy, grass, paving, and stone surfaces. The final dataset contains 281 232 data points which were used for model building. The results of the proposed supervised autoencoder were compared and contextualized with three other models, i.e., an SVM model, a logistic regression model, and an XGBoost model. Results indicate that it is not only feasible but also desirable to consider the use of a supervised autoencoder model when there is a need for terrain classifications.

Journal of Computer Science
Volume 19 No. 9, 2023, 1073-1086

DOI: https://doi.org/10.3844/jcssp.2023.1073.1086

Submitted On: 22 April 2023 Published On: 31 August 2023

How to Cite: Toit, T. D., Kruger, H. & Merwe, A. V. D. (2023). Automated Terrain Classification with a Bayesian Hyperparameter Optimized Deep Supervised Autoencoder Model. Journal of Computer Science, 19(9), 1073-1086. https://doi.org/10.3844/jcssp.2023.1073.1086

  • 1,275 Views
  • 747 Downloads
  • 0 Citations

Download

Keywords

  • Bayesian Optimization
  • Hyperparameter Optimization
  • Supervised Autoencoder
  • Terrain Classification