Work

Impact of the breathing motion prediction horizon on the performance of bidirectional classical recurrent neural and temporal Kolmogorov–Arnold networks

Year: 2025

Type: article

Abstract: Objective. For surface-based breathing motion prediction, which is essential to overcome inherent system latencies of active motion management strategies in radiotherapy, long short-term memory (LSTM) networks and related networks—bidirectional LSTMs (BiLSTMs), gated recurrent units (GRUs), and BiGRUs—have shown promise, primarily for prediction horizons around 500 ms. The novel temporal Kolmogorov–Arnold networks (TKANs), and BiTKANs remain unexplored for this application. Therefore, this study aimed at assessing the performance of the above six networks for different prediction horizons up to 1000 ms. Approach. C-Rad Sentinel (C-Rad AB, Sweden) breathing signals were retrospectively collected for 207 patients. Signals from 41 patients were used for model testing, while the remaining data sets were 9:1 split for training and validation. Pre-processing included cropping, low-pass filtering, re-sampling, and scaling. Signals were cut into 8 s segments as input for the prediction models with a corresponding true label. A grid search was performed to optimize hyperparameters for each of the six network types using a prediction horizon of 500 ms. Subsequently, each network was trained ten times with these parameters across various training:validation splits, evaluating ten prediction horizons between 100 ms and 1000 ms. Root mean square errors (RMSEs) between the predicted breathing amplitudes and the true labels as well as inference times were calculated on the test set. Main results. The hyperparameter-tuning resulted in a lowest mean RMSE of 57 ± 39 μ m for the LSTM, while a mean RMSE of 88 ± 35 μ m was obtained for the TKAN. TKANs were fastest to infer, averaging 10.2 ms per prediction. Increasing prediction horizons resulted in moderate increases of RMSEs up to 113 μ m for horizons below 600 ms. RMSEs remained below 500 μ m for horizons up to 1000 ms. Significance . The LSTM outperforms newer networks for most prediction horizons. All models showed feasibility for a wide range of prediction horizons and the effect of the prediction horizon on the RMSE was assessed.

Source: Physics in Medicine and Biology

DOI: 10.1088/1361-6560/ae2819