- Research Article
- 10.1080/12269328.2026.2667190
Integrating log data and advanced machine learning for one-dimensional geomechanical modeling with PSO and Transformer mechanisms
- May 09, 2026
- Geosystem Engineering
- Farhad Mollaei + 2 more +2
ABSTRACT In one-dimensional (1D) geomechanical modeling, precise predictions are essential for ensuring wellbore stability, defining safe mud weight windows, and determining hydraulic fracturing pressure ranges. However, accurately estimating geomechanical parameters, especially in data-limited environments, remains a significant challenge. This study addresses these challenges by integrating advanced machine learning algorithms with well log data from two wells in one of Iran’s hydrocarbon fields. The dataset includes shear wave velocity (Vs) logs, static Young’s modulus (Est) samples, uniaxial compressive strength (UCS) samples, internal friction angle ( φ ) measurements, and pore pressure (PP) samples. To tackle the challenge of predicting Vs with high precision, three sophisticated algorithms were employed: Bidirectional Long Short-Term Memory (BiLSTM), BiLSTM+Transformer, and a hybrid model (CBiT) combining 1D Convolutional Neural Network (CNN), BiLSTM, and Transformer mechanisms. Feature selection using Pearson correlation and Mutual Information and SHapley Additive exPlanations (SHAP) analysis identified the most influential parameters for each algorithm. The models were evaluated against traditional empirical methods using statistical metrics, such as mean square error (MSE), root mean square error (RMSE), and coefficient of determination (R2). The results revealed that the machine learning models significantly outperformed the empirical methods, with the hybrid CBiT model showing superior accuracy, robustness, and generalization capability, establishing a new benchmark in predictive performance for shear wave velocity due to its ability to capture complex nonlinear relationships and long-term dependencies. Particle Swarm Optimization (PSO) was applied to further optimize model hyperparameters, achieving R2 values of 0.9989 (training) and 0.9985 (testing), with RMSE for unseen data reduced to 0.0269 and R2 reaching 0.9973. Dynamic elastic parameters, including Young’s modulus, bulk modulus, and shear modulus, were calculated using the optimized model. Strong correlations between dynamic and static properties, such as dynamic Young’s modulus and UCS, were extrapolated across the entire depth range of both wells. The study also introduced a novel approach to estimate internal friction angle using Plumb’s correlation and laboratory data, alongside improved pore pressure predictions based on Eaton’s empirical method. Finally, vertical and horizontal stresses were calculated using poroelastic equations for comprehensive stress modeling. This study demonstrates how the integration of hybrid machine learning algorithms, advanced optimization techniques, and rigorous data analysis can significantly improve the accuracy, reliability, and practical applicability of 1D geomechanical models, providing a robust and efficient framework for subsurface analysis and offering substantial advantages over conventional approaches.
Read more