Surrounding Rock Squeezing Classification in Underground Engineering Using a Hybrid Paradigm of Generative Artificial Intelligence and Deep Ensemble Learning

Surrounding rock squeezing is a common geological disaster in underground excavation projects (e.g., TBM tunneling and deep mining), which has adverse effects on construction safety, schedule, and property. To predict the squeezing of the surrounding rock accurately and quickly, this study proposes a hybrid machine learning paradigm that integrates generative artificial intelligence and deep ensemble learning. Specifically, conditional tabular generative adversarial network is devised to solve the problems of data shortage and class imbalance for data augmentation at the data level, and the deep random forest is built based on the augmented data for subsequent squeezing classification. A total of 139 historical squeezing cases are collected worldwide to validate the efficacy of the proposed modeling paradigm. The results reveal that this paradigm achieves a prediction accuracy of 92.86% and a macro F 1 -score of 0.9292. In particular, the individual F 1 -scores on strong squeezing and extremely strong squeezing are more than 0.9, with excellent prediction reliability for high-intensity squeezing. Finally, a comparative analysis with traditional machine learning techniques is conducted and the superiority of this paradigm is further verified. This study provides a valuable reference for surrounding rock squeezing classification under a limited data environment.