AI-based rock strength assessment from tunnel face images using hybrid neural networks

Abstract

In geological engineering and related fields, accurately and quickly identifying lithology and assessing rock strength are crucial for ensuring structural safety and optimizing design. Traditional rock strength assessment methods mainly rely on field sampling and laboratory tests, such as uniaxial compressive strength (UCS) tests and velocity tests. Although these methods provide relatively accurate rock strength data, they are complex, time-consuming, and unable to reflect real-time changes in field conditions. Therefore, this study proposes a new method based on artificial intelligence and neural networks to improve the efficiency and accuracy of rock strength assessments. This research utilizes a Transformer + UNet hybrid model for lithology identification and an optimized ResNet-18 model for determining rock weathering degrees, thereby correcting the strength of the tunnel face surrounding rock. Experimental results show that the Transformer + UNet hybrid model achieves an accuracy of 95.57% in lithology identification tasks, while the optimized ResNet model achieves an accuracy of 96.13% in rock weathering degree determination. Additionally, the average relative error in tunnel face strength detection results is only 9.33%, validating the feasibility and effectiveness of this method in practical engineering applications. The multi-model neural network system developed in this study significantly enhances prediction accuracy and efficiency, providing robust scientific decision support for tunnel construction, thereby improving construction safety and economy.