TransQA: deep hybrid transformer network for measurement-guided volumetric dose prediction of pre-treatment patient-specific quality assurance

Abstract

Objective. Performing pre-treatment patient-specific quality assurance (prePSQA) is considered an essential, time-consuming, and resource-intensive task for volumetric modulated arc radiotherapy (VMAT) which confirms the dose accuracy and ensure patient safety. Most current machine learning and deep learning approaches stack excessive convolutional/pooling operations (CPs) to predict prePSQA with two-dimensional or one-dimensional information input. However, these models generally present limitations in explicitly modeling long-range dependency for volumetric dose prediction due to the loss of spatial dose features and the inherent locality of CPs. The purpose of this work is to construct a deep hybrid network by combining the self-attention mechanism-based Transformer with modified U-Net for predicting measurement-guided volumetric dose (MDose) of prePSQA. Approach. The enrolled 307 cancer patients underwent VMAT were randomly divided into 246 and 61 cases for training and testing the model. The input data included computed tomography images, radiotherapy dose images exported from the treatment planning system, as well as the MDose distribution from the verification system. The output was the predicted high-quality voxel-wise prePSQA dose distribution. Main results: qualitative and quantitative experimental results show that the proposed prediction method could achieve comparable or better performance on MDose prediction over other approaches in terms of spatial dose distribution, dose-volume histogram metrics, gamma passing rates, mean absolute error, root mean square error, and structural similarity. Significance. The preliminary results on multiple cancer sites show that our approach can be taken as a clinical guidance tool and help medical physicists to reduce the measurement work of prePSQA.