Automated Terrain Classification with a Bayesian Hyperparameter Optimized Deep Supervised Autoencoder Model
- 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.
DOI: https://doi.org/10.3844/jcssp.2023.1073.1086
Copyright: © 2023 Tiny Du Toit, Hennie Kruger and Annette Van Der Merwe. This is an open access article distributed under the terms of the
Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
- 1,019 Views
- 592 Downloads
- 0 Citations
Download
Keywords
- Bayesian Optimization
- Hyperparameter Optimization
- Supervised Autoencoder
- Terrain Classification