Rock and Soil Mechanics ›› 2026, Vol. 47 ›› Issue (9): 3261-3275.doi: 10.16285/j.rsm.2025.0913

• Numerical Analysis • Previous Articles     Next Articles

A physics-informed neural network model for predicting uniaxial compressive strength of rock based on while-drilling parameters

WU Jing-rong1, 2, LEI Jun-qiang1, 3, LIU Xue-wei3, LIU Bin3, ZHOU Zhe4, LUO Xu-feng4   

  1. 1. School of Civil Engineering, Architecture and Environment, Hubei University of Technology, Wuhan, Hubei 430068, China; 2. Engineering and Technology College, Hubei University of Technology, Wuhan, Hubei 430068, China; 3. State Key Laboratory of Geomechanics and Geotechnical Engineering Safety, Institute of Rock and Soil Mechanics, Chinese Academy of Sciences, Wuhan, Hubei 430071, China; 4. Guangxi China Construction Rongfu Expressway Co. Ltd., Nanning, Guangxi 530200, China
  • Received:2025-08-25 Accepted:2026-01-06 Online:2026-09-11 Published:2026-09-01
  • Supported by:
    This work was supported by the National Natural Science Foundation of China (U22A20234), the Hubei Province Key Research and Development Project (2023BCB121, 2024DJC002) and Wuhan Innovation Supporting Projects (2023020201010079).

Abstract: Accurate determination of rock mechanical parameters is crucial for engineering design. While-drilling technology has become one of the primary approaches for assessing rock strength. Existing while-drilling strength prediction models typically rely on a single approach, such as theoretical physical analysis or machine learning. As a result, their predictive performance is highly sensitive to field operating conditions and data quality. This study develops a physics-informed neural network (PINN) model to predict the uniaxial compressive strength of rock from while-drilling parameters. The model is built on a CNN-LSTM-Attention fusion architecture. It achieves accurate prediction of rock uniaxial compressive strength by integrating physical knowledge with data-driven learning. Specifically, the physical model for strength prediction based on multi-wing cutting drilling is transformed into constraint conditions within the network. Based on parameter sample construction and comparative analysis, the modified physical model improves rock strength prediction accuracy by 15%. Its correlation coefficient with the measured rock strength reaches 0.97. Compared with the 6 traditional neural network models, the determination coefficient of the strength prediction results of the PINN model on the test set is increased by at least 2%, and the mean absolute error is reduced by at least 9%. Further validation using actual while-drilling data from composite formations shows that, when the lithological sequence is granite, conglomerate, and sandstone, the average prediction errors of the PINN model are 3.63, 4.44 and 1.19 MPa, respectively. In all cases, the percentage error is less than 8%, demonstrating the model’s strong robustness in predicting the strength of formations with different lithologies. These findings provide practical guidance for tunnel while-drilling technology and the real-time automated characterization of rock mass strength.

Key words: while-drilling parameters, physics-informed neural networks (PINN), machine learning, rock strength, while-drilling detection

CLC Number: 

  • TU415
[1] ZHANG Hong-yang, LUO Chao-fan, WANG Te, HAN Li-wei, DING Ze-lin, ZHANG Xian-qi, SHI Yan-ke. A review of dam risk warning models based on intelligent optimization algorithms [J]. Rock and Soil Mechanics, 2026, 47(9): 3102-3123.
[2] ZHANG Jing, HAN Yan-song, ZHOU Zong-hong, LIU Hai, OUYANG Zhi-hua. Improvement of true triaxial Hoek-Brown criterion considering the rock critical confining pressure effect [J]. Rock and Soil Mechanics, 2026, 47(7): 2337-2346.
[3] LIU De-ren, WANG Kai-qiang, ZHANG Yan-jie, WANG Shuai-qun, WANG Yu-fei. Machine learning-based prediction and interpretability analysis of Longxi loess collapsibility coefficient [J]. Rock and Soil Mechanics, 2026, 47(7): 2235-2247.
[4] SUN Jia-hao, LI Di-yuan, XIE Lian-ku. An improved robust random forest algorithm for predicting hard rock pillar stability [J]. Rock and Soil Mechanics, 2026, 47(5): 1788-1800.
[5] JIANG Ya-long, ZHOU Ya-feng, XU Peng-chu-xuan, ZHANG Qi, QIU Si-bao. Mechanical damage and penetrability of extremely hard granite subjected to thermal-shock cycles using water cooling [J]. Rock and Soil Mechanics, 2026, 47(3): 856-868.
[6] GE Xin-bo, HUANG Jun, ZHAO Tong-bin, TAO Gang, MA Hong-ling, WANG Wei. A review of the application of artificial intelligence in underground engineering for compressed air energy storage [J]. Rock and Soil Mechanics, 2026, 47(2): 413-425.
[7] JIANG Xiao-tong, ZHANG Xi-wen, LÜ Ying-hui, LI Ren-jie, JIANG Hao, . Current applications and future prospects of machine learning in geotechnical engineering [J]. Rock and Soil Mechanics, 2025, 46(S1): 419-436.
[8] CAI Qi-hang, DONG Xue-chao, GUO Ming-wei, LU Zheng, XU An, JIANG Fan, . Intelligent prediction of sinking of super-large anchorage caisson foundation based on soil pressure at cutting edges [J]. Rock and Soil Mechanics, 2025, 46(S1): 377-388.
[9] ZHEN Jia-jie, LAI Feng-wen, HUANG Ming, LIAO Qing-xiang, LI Shuang, DUAN Yue-qiang. Intelligent geological condition recognition in shield tunneling via time-series clustering and online learning [J]. Rock and Soil Mechanics, 2025, 46(11): 3615-3625.
[10] HE Long-ping, YAO Nan, WANG Qi-hu, YE Yi-cheng, LING Ji-suo, . Rock burst intensity grading prediction model based on automatic machine learning [J]. Rock and Soil Mechanics, 2024, 45(9): 2839-2848.
[11] LONG Xiao, SUN Rui, ZHENG Tong, . Convolutional neural network-based liquefaction prediction model and interpretability analysis [J]. Rock and Soil Mechanics, 2024, 45(9): 2741-2753.
[12] YANG Yang, WEI Yi-tong. A new method of liquefaction probability level evaluation based on classification tree [J]. Rock and Soil Mechanics, 2024, 45(7): 2175-2186.
[13] DENG Zhi-xing, XIE Kang, LI Tai-feng, WANG Wu-bin, HAO Zhe-rui, LI Jia-shen, . A novel method for quality control of vibratory compaction in high-speed railway graded aggregates based on the embedded locking point of coarse particles [J]. Rock and Soil Mechanics, 2024, 45(6): 1835-1849.
[14] PAN Qiu-jing, WU Hong-tao, ZHANG Zi-long, SONG Ke-zhi, . Prediction of tunneling-induced ground surface settlement within composite strata using multi-physics-informed neural network [J]. Rock and Soil Mechanics, 2024, 45(2): 539-551.
[15] JIANG Ming-jing, ZHANG Lu-feng, HAN Liang, JIANG Peng-ming, . Damage law of structured sand using symbolic regression algorithm [J]. Rock and Soil Mechanics, 2024, 45(12): 3768-3778.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!