TransAnaNet: Transformer-based Anatomy Change Prediction Network for Head and Neck Cancer Patient Radiotherapy
Meixu Chen1, Kai Wang1,2, Michael Dohopolski1
1Medical Artificial Intelligence and Automation (MAIA) Lab, Department of Radiation Oncology, UT Southwestern Medical Center, Dallas, TX, 75235, USA.
Background:
Adaptive radiotherapy (ART) can compensate for the dosimetric impact of anatomic change during radiotherapy of head neck cancer (HNC) patients. However, implementing ART universally poses challenges in clinical workflow and resource allocation, given the variability in patient response and the constraints of available resources. Therefore, early identification of head and neck cancer (HNC) patients who would experience significant anatomical change during radiotherapy (RT) is of importance to optimize patient clinical benefit and treatment resources.
Purpose:
The purpose of this study is to assess the feasibility of using a vision-transformer (ViT) based neural network to predict radiotherapy induced anatomic change of HNC patients.
Methods:
We retrospectively included 121 HNC patients treated with definitive RT/CRT. We collected the planning CT (pCT), planned dose, CBCTs acquired at the initial treatment (CBCT01) and fraction 21 (CBCT21), and primary tumor volume (GTVp) and involved nodal volume (GTVn) delineated on both pCT and CBCTs for model construction and evaluation. A UNet-style ViT network was designed to learn the spatial correspondence and contextual information from embedded image patches of CT, dose, CBCT01, GTVp, and GTVn. The deformation vector field between CBCT01 and CBCT21 was estimated by the model as the prediction of anatomic change, and deformed CBCT01 was used as the prediction of CBCT21. We also generated binary masks of GTVp, GTVn and patient body for volumetric change evaluation. We used data from 100 patients for training and validation, and the remaining 21 patients for testing. Image and volumetric similarity metrics including mean square error (MSE), structural similarity index (SSIM), dice coefficient, and average surface distance were used to measure the similarity between the target image and predicted CBCT.
Results:
The predicted image from the proposed method yielded the best similarity to the real image (CBCT21) over pCT, CBCT01, and predicted CBCTs from other comparison models. The average MSE and SSIM between the normalized predicted CBCT to CBCT21 are 0.009 and 0.933, while the average dice coefficient between body mask, GTVp mask, and GTVn mask are 0.972, 0.792, and 0.821 respectively.
Conclusions:
The proposed method showed promising performance for predicting radiotherapy induced anatomic change, which has the potential to assist in the decision making of HNC Adaptive RT.
