Patient-Specific Autoregressive Models for Organ Motion Prediction in Radiotherapy
Journal:
arXiv
Published Date:
May 17, 2025
Abstract
Radiotherapy often involves a prolonged treatment period. During this time,
patients may experience organ motion due to breathing and other physiological
factors. Predicting and modeling this motion before treatment is crucial for
ensuring precise radiation delivery. However, existing pre-treatment organ
motion prediction methods primarily rely on deformation analysis using
principal component analysis (PCA), which is highly dependent on registration
quality and struggles to capture periodic temporal dynamics for motion
modeling.In this paper, we observe that organ motion prediction closely
resembles an autoregressive process, a technique widely used in natural
language processing (NLP). Autoregressive models predict the next token based
on previous inputs, naturally aligning with our objective of predicting future
organ motion phases. Building on this insight, we reformulate organ motion
prediction as an autoregressive process to better capture patient-specific
motion patterns. Specifically, we acquire 4D CT scans for each patient before
treatment, with each sequence comprising multiple 3D CT phases. These phases
are fed into the autoregressive model to predict future phases based on prior
phase motion patterns. We evaluate our method on a real-world test set of 4D CT
scans from 50 patients who underwent radiotherapy at our institution and a
public dataset containing 4D CT scans from 20 patients (some with multiple
scans), totaling over 1,300 3D CT phases. The performance in predicting the
motion of the lung and heart surpasses existing benchmarks, demonstrating its
effectiveness in capturing motion dynamics from CT images. These results
highlight the potential of our method to improve pre-treatment planning in
radiotherapy, enabling more precise and adaptive radiation delivery.