A comparative study of soil classification machine learning models for construction management

Abstract

Prior to the construction of buildings and roads, soil classification plays a vital role to ensure proper project management and durability of the structure. Civil engineers need to possess knowledge about soil behavior and properties of soil classification to construct reliable and long-lasting infrastructures. However, traditional soil classification methods are both expensive and time-consuming. Recently, machine learning (ML) has become increasingly popular in solving complex problems in Geotechnical Engineering, leading to novel approaches for automating soil classification. This research evaluates the effectiveness of various machine learning algorithms, including Multinomial Logistic Regression (MLR), Gaussian Naive Bayes (GNB), Extreme Gradient Boosting (XGBoost), Random Forest, and Artificial Neural Network-Multilayer Layer Perceptron (ANN-MLP), in classifying soils. The paper also implemented Hard and Soft Voting Ensemble models. Each model was quantitatively evaluated and compared using various metrics. The models achieved high scores in multiple performance metrics across the soil classes with accuracies ranging between 87% and 93%, except the Naive Bayes model with a balanced accuracy of 77.6% while the hard-voting model outperformed the others with a balanced accuracy of 93%. The study’s results benefit industry practitioners and academic researchers for integrating and assessing ML algorithms in soil classification.