A hybrid SVR-BO model for predicting the soil thermal conductivity with uncertainty

Kai-Qi LI,Zhen-Yu Yin,Yong Liu
DOI: https://doi.org/10.1139/cgj-2023-0105
IF: 3.6
2023-06-06
Canadian Geotechnical Journal
Abstract:This study proposes a generalised framework for developing a hybrid machine learning (ML) model that combines support vector regression (SVR) with hyperparameter optimisation to predict thermal conductivity (k) with uncertainty. The framework contains four phases: data pre-processing, determining the best-performing hybrid model, selecting the optimal input combination, and uncertainty implementation. A database containing 2197 data points is first compiled to train the ML model. Three hyperparameter optimisation algorithms are adopted to tune hyperparameters, and their performance is evaluated by model evaluation metrics. Results show that SVR with Bayesian optimisation (SVR-BO) is the best-performing model since it produces more accurate predictions for k than do models that employ grid and random searches. Given the sample insufficiency issue encountered in practice, the SVR-BO models with 144 input combinations are analysed. The compassion among models under various input combinations indicates that incorporating temperature as an additional input can provide moderate improvement in the accuracy and generalisability of the hybrid model. Based on the comparison, a 5-input model is selected as the best candidate to implement the uncertainty evaluation for k. Results demonstrate that the predicted k possesses higher reliability for denser datasets and shows promising potential for applications in k with uncertainty assessments.
geosciences, multidisciplinary,engineering, geological
What problem does this paper attempt to address?