Leveraging graph neural networks and gate recurrent units for accurate and transparent prediction of baseball pitching speed.
Journal:
Scientific reports
PMID:
40044722
Abstract
Long short-term memory (LSTM) networks are widely used in biomechanical data analysis but have the significant limitations in interpretability and decision transparency. Combining graph neural networks (GNN) with gate recurrent units (GRU) may offer a better solution. This study proposes and validates a hybrid GNN-GRU model for predicting baseball pitching speed and enhancing its interpretability using layer-wise relevance propagation (LRP). C3D data from 53 baseball athletes were downloaded from a public dataset. Kinematic features of 9 joints and pitching speed during the pitching process were calculated using Visual3D, resulting in a total of 208 valid pitches. The feature data were input into both LSTM and GNN-GRU hybrid models, with hyperparameters tuned using particle swarm optimization. LRP was employed to obtain the contribution rate changes of kinematic features to the prediction results throughout the pitching cycle. The prediction accuracy of the models was evaluated using mean absolute error (MAE), mean squared error (MSE), and R-squared (R). The results showed that there were the significant statistical differences in the MAE and R metrics between the LSTM model and the GNN-GRU model in predicting pitching speed on the test set. The MAE (P = 0.000, Z = - 5.170, Cohen's d = 1.514) and R (P = 0.000, Z = - 2.981, Cohen's d = 2.314) of the LSTM model were significantly lower than those of the GNN-GRU model. Compared to LSTM, the GNN-GRU model achieved better prediction accuracy but was potentially more susceptible to the influence of data variability. Moreover, the GNN-GRU-based model demonstrated the better interpretability and decision transparency.