This page documents the reference implementation used in the following paper on online training of vanilla RNNs using UORO for medical time-series forecasting.
We provide an implementation of UORO using closed-form simplifications for quantities appearing in the loss gradient estimation for vanilla RNNs.
We experimentally compare UORO with several baselines, including Real-Time Recurrent Learning (RTRL), in a respiratory motion-trace forecasting problem relevant to radiotherapy.
We take into account hyperparameter tuning, inference time, prediction oscillations, and the influence of the prediction horizon h and ground-truth signal irregularity.
UORO yielded lower horizon-averaged RMSE, nRMSE, and maximum error than the other algorithms, and was generally the most accurate method for medium-to-high values of h.
It also exhibited satisfactory robustness to input-signal unsteadiness, limited step-to-step prediction fluctuations, and relatively low processing time.
1@article{pohl2022prediction,
2 title={Prediction of the position of external markers using a recurrent neural network trained with unbiased online recurrent optimization for safe lung cancer radiotherapy},
3 author={Pohl, Michel and Uesaka, Mitsuru and Takahashi, Hiroyuki and Demachi, Kazuyuki and Chhatkuli, Ritu Bhusal},
4 journal={Computer Methods and Programs in Biomedicine},
5 volume={222},
6 pages={106908},
7 year={2022},
8 publisher={Elsevier}
9}