Prediction of Horizontal in Situ Stress in Shale Reservoirs Based on Machine Learning Models

To address the limitations of traditional methods in modeling complex nonlinear relationships in horizontal in situ stress prediction for shale reservoirs, this study proposes an integrated framework that combines well logging interpretation with machine learning to accurately predict horizontal in situ stress in shale reservoirs. Based on the logging data from five wells in the Luzhou Block of the Sichuan Basin (16,000 samples), Recursive Feature Elimination (RF-RFE) was used to identify nine key factors, including Stoneley wave slowness and caliper, from 30 feature parameters. Bayesian optimization was employed to fine-tune the hyperparameters of the XGBoost model globally. Results indicate that the XGBoost model performs optimally in predicting maximum horizontal principal stress (SHmax) and minimum horizontal principal stress (SHmin). It achieves R2 values of 0.978 and 0.959, respectively, on the test set. The error metrics (MAE, MSE, RMSE) of the XGBoost model are significantly lower than those of SVM and Random Forest, demonstrating its precise capture of the nonlinear relationships between logging parameters and in situ stress. This framework enhances the model’s adaptability to complex geological conditions through multi-well data training and eliminating redundant features, providing a reliable tool for hydraulic fracturing design and wellbore stability assessment in shale gas development.