Back analysis of geomechanical parameters based on a data augmentation algorithm and machine learning technique

Abstract

Accurate geomechanical parameters are key factors for stability evaluation, disaster forecasting, structural design, and supporting optimization. The intelligent back analysis method based on the monitored information is widely recognized as the most efficient and cost-effective technique for inverting parameters. To address the low accuracy of measured data, and the scarcity of comprehensive datasets, this study proposes an innovative back analysis framework tailored for small sample sizes. We introduce a multi-faceted back analysis approach that combines data augmentation with advanced optimization and machine learning techniques. The auxiliary classifier generative adversarial network (ACGAN)-based data augmentation algorithm is first employed to generate synthetic yet realistic samples that adhere to the underlying probability distribution of the original data, thereby expanding the dataset and mitigating the impact of small sample sizes. Subsequently, we harness the power of optimized particle swarm optimization (OPSO) integrated with support vector machine (SVM) to mine the intricate nonlinear relationships between input and output variables. Then, relying on a case study, the validity of the augmented data and the performance of the developed OPSO-SVM algorithms based on two different sample sizes are studied. Results show that the new datasets generated by ACGAN almost coincide with the actual monitored convergences, exhibiting a correlation coefficient exceeding 0.86. Furthermore, the superiority of the OPSO-SVM algorithm is also demonstrated by comparing the displacement prediction capability of various algorithms through four indices. It is also indicated that the relative error of the predicted displacement values reduces from almost 20% to 5% for the OPSO-SVM model trained with 25 samples and that trained with 625 samples. Finally, the inversed parameters and corresponding convergences predicted by the two OPSO-SVM models trained with different samples are discussed, indicating the feasibility of the combination application of ACGAN and OPSO-SVM in back analysis of geomechanical parameters. This endeavor not only facilitates the progression of underground engineering analysis in scenarios with limited data, but also serves as a pivotal reference for both researchers and practitioners alike.

Keywords

Back analysis / Machine learning / Data augmentation / Geomechanical parameters

Data availability

The data that support the findings of this study are available from the corresponding author upon reasonable request.

CRediT authorship contribution statement

Hui Li: Writing - review & editing, Writing - original draft, Visualization, Validation, Methodology, Conceptualization. Weizhong Chen: Writing - review & editing, Validation, Supervision, Methodology, Conceptualization. Xianjun Tan: Writing - review & editing, Supervision.

Declaration of competing interest

The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.

Acknowledgement

This work is supported by the National Natural Science Foundation of China (Grant Nos. 51991392 and 51922104).

References

[1] Argilaga, A. (2023). Fractal informed generative adversarial networks (FIGAN): Application to the generation of X-ray CT images of a selfsimilar partially saturated sand. Computers and Geotechnics, 158, 105384.

[2] Bani-Hani, K. A. (2007). Vibration control of wind-induced response of tall buildings with an active tuned mass damper using neural networks. Structural Control and Health Monitoring, 14(1), 83-108.

[3] Bertuzzi, R. (2017). Back-analysing rock mass modulus from monitoring data of two tunnels in Sydney, Australia. Journal of Rock Mechanics and Geotechnical Engineering, 9(5), 877-891.

[4] Blümling, P., Bernier, F., Lebon, P., & Martin, C. D. (2007). The excavation damaged zone in clay formations time-dependent behavior and influence on performance assessment. Physics and Chemistry of the Earth, 32, 588-599.

[5] Chang, X. Y., Wang, H., & Zhang, Y. M. (2023). Back analysis of rock mass parameters in tunnel engineering using machine learning techniques. Computers and Geotechnics, 163, 105738.

[6] Chen, T., & Guestrin, C. (2016). XGBoost: A scalable tree boosting system. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. Presented at the KDD '16: The 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, ACM, San Francisco California USA (Pp. 785-794). Association for Computing Machinery.

[7] Chen, W. Z., Tian, Y., Wang, X. H., Tian, H. M., Cao, H. X., & Xie, H. D. (2019). Squeezing prediction of tunnel in soft rocks based on modified [BQ]. Rock and Soil Mechanics, 40(8), 3215-13133 (in Chinese).

[8] Chu, Z., Wu, Z., Liu, B., & Liu, Q. (2019). Coupled analytical solutions for deep-buried circular lined tunnels considering tunnel face advancement and soft rock rheology effects. Tunnelling and Underground Space Technology, 94, 103111.

[9] Ding, H., Chen, L., Dong, L., Fu, Z., & Cui, X. (2022). Imbalanced data classification: A KNN and generative adversarial networks-based hybrid approach for intrusion detection. Future Generation Computer Systems, 131, 240-254.

[10] Ding, J., Wang, M., Ping, Z., Fu, D., & Vassiliadis, V. S. (2020). An integrated method based on relevance vector machine for short-term load forecasting. European Journal of Operational Research, 287(2), 497-510.

[11] Duan, T., Avati, A., Ding, D. Y., Thai, K. K., Basu, S., Ng, A., & Schuler, A. (2020). NGBoost: Natural gradient boosting for probabilistic prediction. In Proceedings of the 37th International Conference on Machine Learning, PMLR 119 (pp. 2690-2700). PMLR.

[12] Fahimi, F., Dosen, S., Ang, K. K., Mrachacz-Kersting, N., & Guan, C. (2020). Generative adversarial networks-based data augmentation for brain-computer interface. IEEE Transactions on Neural Networks and Learning Systems, 32(9), 4039-4051.

[13] Fukushima, K. (1980). Neocognitron: A self-organizing neural network model for a mechanism of pattern recognition unaffected by shift in position. Biological Cybernetics, 36(4), 193-202.

[14] Gao, M. Y., Zhang, N., Shen, S. L., & Zhou, A. (2020). Real-time dynamic earth-pressure regulation model for shield tunneling by integrating GRU deep learning method with GA optimization. IEEE Access, 31(8), 64310-64323.

[15] Goodfellow, I. J., Pouget-Abadie, J., Mirza, M., Xu, P., Warde-Farley, D., Ozair, S., Aaron, C., & Bengio, Y. (2014). Generative adversarial nets. In Proceedings of the 27th International Conference on Neural Information Processing Systems, Montreal, Dec. 8- 13, 2014 (pp. 26722680). Association for Computing Machinery.

[16] Hancock, J. T., & Khoshgoftaar, T. M. (2020). CatBoost for big data: An interdisciplinary review. Journal of Big Data, 7(1), 94.

[17] Iwana, B. K., & Uchida, S. (2021). An empirical survey of data augmentation for time series classification with neural networks. PLoS One1, 16(7), e0254841.

[18] Jiang, S. H., Zhu, G. Y., Wang, Z. Z., Huang, Z. T., & Huang, J. S. (2023). Data augmentation for CNN-based probabilistic slope stability analysis in spatially variable soils. Computers and Geotechnics, 160, 105501.

[19] Kashani, A. R., Raymond, C., Seyedali, M., & Gandomi, A. H. (2021). Particle swarm optimization variants for solving geotechnical problems: Review and comparative analysis. Archives of Civil and Mechanical Engineering, 28(3), 1871-1927.

[20] Ke, G. L., Meng, Q., Finley, T., Wang, T. F., Chen, W., Ma, W. D., Ye, Q. W., & Liu, T. Y. (2017). LightGBM: A highly efficient gradient boosting decision tree. In Proceedings of the 31st Conference on Neural Information Processing Systems, (NIPS 2017), Long Beach, CA, USA (pp. 3149-3157). Curran Associates Inc.

[21] Kennedy, J., & Eberhart, R. (1995). Particle swarm optimization. In Proceedings of ICNN'95 - International Conference on Neural Networks, Perth, WA, Australia (pp. 1942-1948).

[22] Khatti, H., Grover, K. S., Kim, H. J., Mawuntu, K. B., & Park, T. W. (2024). Prediction of ultimate bearing capacity of shallow foundations on cohesionless soil using hybrid LSTM and RVM approaches: An extended investigation of multicollinearity. Computers and Geotechnics, 165, 105912.

[23] Kim, Y., Lim, S. Y., Kim, K. Y., & Yun, T. S. (2023). Prediction of compressional wave velocity of cement-reinforced soil from core images using a convolutional neural network regression model. Computers and Geotechnics, 153, 105067.

[24] LeCun, Y., Bengio, Y., & Hinton, G. (2015). Deep learning. Nature, 521 (7553), 436-444.

[25] LeCun, Y., Boser, B., Denker, J. S., Henderson, D., Howard, R. E., Hubbard, W., & Jackel, L. D. (1989). Backpropagation applied to handwritten zip code recognition. Neural Computation, 1(4), 541-551.

[26] LeCun, Y., & Bottou, L. (1998). Gradient-based learning applied to document recognition. Proceedings of the IEEE, 86(11), 2278-2324.

[27] Lee, M. B., Kim, Y. H., & Park, K. R. (2019). Conditional generative adversarial network-based data augmentation for enhancement of iris recognition accuracy. IEEE Access, 7, 122134-122152.

[28] Li, H., Chen, W. Z., & Tan, X. J. (2023a). Displacement-based back analysis of mitigating the effects of displacement loss in underground engineering. Journal of Rock Mechanics and Geotechnical Engineering, 15(10), 2626-2638.

[29] Li, H., Chen, W. Z., Tan, X. Y., & Tan, X. J. (2023b). Back analysis of geomechanical parameters for rock mass under complex geological conditions using a novel algorithm. Tunnelling and Underground Space Technology, 136, 105099.

[30] Li, W., Zhong, X., Shao, H., Cai, B., & Yang, X. (2022). Multi-mode data augmentation and fault diagnosis of rotating machinery using modified ACGAN designed with new framework. Advanced Engineering Informatics, 52, 101552.

[31] Liang, W. Z., Luo, S. Z., Zhao, G. Y., & Wu, H. (2020). Predicting hard rock pillar stability using GBDT, XGBoost, and LightGBM algorithms. Mathematics, 8(5), 765.

[32] Liu, Q. S., Lei, Y. M., Yin, X., Lei, J. S., Pan, Y. C., & Sun, L. (2023). Development and application of a novel probabilistic back-analysis framework for geotechnical parameters in shield tunneling based on the surrogate model and Bayesian theory. Acta Geotechnica, 18(9), 4899-4921.

[33] Luo, Y., Chen, J., Chen, Y., Diao, P., & Qiao, X. (2018). Longitudinal deformation profile of a tunnel in weak rock mass by using the back analysis method. Tunnelling and Underground Space Technology, 71, 478-493.

[34] Mahmoodzadeh, A., Mohammadi, M., Abdulhamid, S. N., Ibrahim, H. H., Ali, H. F. H., & Salim, S. G. (2021). Dynamic reduction of time and cost uncertainties in tunneling projects. Tunnelling and Underground Space Technology, 109, 103774.

[35] Manzanal, D., Drempetic, V., Haddad, B., Pastor, M., Martin, S. M., & Mira, P. (2016). Application of a new rheological model to rock avalanches: An SPH approach. Rock Mechanics and Rock Engineering, 49(6), 2353-2372.

[36] Mao, J., Wang, H., & Spencer, B. F. (2021). Toward data anomaly detection for automated structural health monitoring: Exploiting generative adversarial nets and autoencoders. Structural Control and Health Monitoring, 20(4), 1609-1626.

[37] McCulloch, W. S., & Pitts, W. (1943). A logical calculus of the ideas immanent in nervous activity. The Bulletin of Mathematical Biophysics, 5(4), 115-133.

[38] McKay, M. D., Beckman, R. J., & Conover, W. J. (1979). A comparison of three methods for selecting values of input variables in the analysis of output from a computer code. Technometrics, 21(2), 239-245.

[39] Mirza, M., & Osindero, S. (2014). Conditional generative adversarial net. Computer Science, 2672-2680.

[40] Montgomery, D. C., Peck, E. A., & Vining, G. G. (2012). Introduction to linear regression analysis. John Wiley & Sons.

[41] Ni, Y., & Li, M. (2016). Wind pressure data reconstruction using neural network techniques: A comparison between BPNN and GRNN. Measurement, 88, 468-476.

[42] Odena, A., Olah, C., & Shlens, J. (2017). Conditional image synthesis with auxiliary classifier GANs. In Proceedings of the 34th International Conference on Machine Learning, Sydney NSW Australia, PMLR 70 (pp. 2642-2651). PMLR.

[43] Oh, B. K., Glisic, B., Kim, Y., & Park, H. S. (2020). Convolutional neural network-based data recovery method for structure health monitoring. Structural Control and Health Monitoring, 19(6), 1821.

[44] Qi, C., & Fourie, A. (2018). A real-time back-analysis technique to infer rheological parameters from field monitoring. Rock Mechanics and Rock Engineering, 51(10), 3029-3043.

[45] Radford, A., Metz, L., & Chintala, S. (2015). Unsupervised representation learning with deep convolutional generative adversarial networks. arXiv:1511.06434.

[46] Sun, J. L., Wu, S. C., Wang, H., Wang, T., Geng, X. J., & Zhang, Y. J. (2023). Inversion of surrounding rock mechanical parameters in a soft rock tunnel based on a hybrid model EO-LightGBM. Rock Mechanics and Rock Engineering, 56(9), 6691-6707.

[47] Xiong, Q., Li, Z., Luo, H., & Zhao, Z. (2019). Wind tunnel test study on wind load coefficients variation law of heliostat based on uniform design method. Solar Energy, 184, 209-229.

[48] Xu, G. W., He, C., Wang, J., & Zhang, J. B. (2020). Study on the damage evolution of secondary tunnel lining in layered rock stratum. Bulletin of Engineering Geology and the Environment, 79(7), 1697041.

[49] Zhang, F., Jiang, A., Guo, X., & Yang, X. (2021). Creep parameter inversion and long-term stability analysis of tunnel based on GP-DE intelligent algorithm. In Advances in Materials Science and Engineering (pp.3769474).

[50] Zhang, Y. Q., Tang, Z. Y., & Yang, R. J. (2022). Data anomaly detection for structural health monitoring by multi-view representation based on local binary patterns. Measurement, 202, 111804.

[51] Zhang, Y., Miyamori, Y., Mikami, S., & Saito, T. (2019). Vibration-based structural state identification by a 1 -dimensional convolutional neural network. Computer-Aided Civil and Infrastructure Engineering, 34(9), 822-839.

[52] Zhao, B. Y., Wang, X. P., Zhang, C., Li, W. C., Abbassi, R., & Chen, K. (2020). Structural integrity assessment of shield tunnel crossing of a railway bridge using orthogonal experimental design. Engineering Failure Analysis, 114, 104594.

[53] Zhu, J. Y, Park, T., Isola, P., & Efros, A. A. (2017). Unpaired image-toimage translation using cycle- consistent adversarial networks. In Proceedings of International Conference on Computer Vision, Venice, Oct 22-29, 2017 (pp. 2242-2251). IEEE.

[54] Zhu, M. L., Peng, H., Liang, M., Song, G. X., Huang, N. H., Xie, W. W., & Han, Y. (2024). RC-XGBoost based mechanical parameters back analysis of rock mass in heavily fractured tunnel: A case in Yunnan, China. Rock Mechanics and Rock Engineering, 57(4), 2997-3019.

[55] Zou, L., Zhang, H., Wang, C., Wu, F., & Gu, F. (2020). MW-ACGAN: Generating multiscale high-resolution SAR images for ship detection. Sensors, 20(22), 6673.