Aiming at the lack of accuracy and effectiveness of the current shield machine speed prediction method, the study proposes to improve the residual network and combine this improved algorithm with the surrounding rock category prediction model to construct the underground shield machine digging speed prediction model. With an average accuracy of 87.4%, an F1 value of 0.86, and an accuracy of 0.84, the study's prediction model of surrounding rock categories was determined to be valid and superior to the other compared models. The effectiveness of the improved residual algorithm constructed by the study was verified, and it was found to have a better fit to the actual values, with a maximum deviation error value of 4.6 mm/min and a root mean square error of 1.835, which was lower than the other comparative algorithms. The empirical analysis of the underground shield machine digging speed prediction model constructed by the study revealed that the area under the line of the work characteristic curve of the subjects was 0.74, and the F1 value was 0.35, and the accuracy was as high as 84.6%, which was significantly better than that of other comparative models. The shield machine digging speed prediction model, which is based on an enhanced residual network built in the study, performs better than other comparison models, according to the results, which can serve as a theoretical guide for the digital management of coal mine output.