Xiayi Yao

Water and mud inrush hazards pose significant threats to the safety, environmental stability, and resource efficiency of tunnel construction, representing a critical barrier to the development of sustainable transportation infrastructure. Misjudgment—especially missed detections of severe hazards—can lead to extensive geological disturbance, excessive energy consumption, and severe socio-environmental impacts. However, pre-trained large-scale models still face two major challenges when applied to tunnel hazard classification: limited labeled samples and the high cost associated with misclassifying severe hazards. This study proposes a sustainability-oriented intelligent classification framework that integrates a large-scale pre-trained model with multi-strategy data augmentation to accurately identify hazard levels during tunnel excavation. First, a Synthetic Minority Over-Sampling Technique (SMOTE)-based multi-strategy augmentation method is introduced to expand the training set, mitigate class imbalance, and enhance the model’s ability to recognize rare but critical hazard categories. Second, a deep feature extraction architecture built on the robustly optimized BERT pretraining approach (RoBERTa) is designed to strengthen semantic representation under small-sample conditions. Moreover, a hierarchical weighting mechanism is incorporated into the weighted cross-entropy loss to emphasize the identification of severe hazard levels, thereby ensuring zero missed detections. Experimental results demonstrate that the proposed method achieves an accuracy of 99.26%, representing a 27.96% improvement over the traditional SVM baseline. Importantly, the recall for severe hazards (Levels III and IV) reaches 100%, ensuring zero misjudgment of major hazards. By effectively reducing safety risks, minimizing environmental disruptions, and promoting resilient tunnel construction, this method provides strong support for sustainable and low-impact underground engineering practices.