Therapeutic dose prediction using score-based diffusion model for pretreatment patient-specific quality assurance

Objectives: Implementing pre-treatment patient-specific quality assurance (prePSQA) for cancer patients is a necessary but time-consuming task, imposing a significant workload on medical physicists. Currently, the prediction methods used for prePSQA fall under the category of supervised learning, limiting their generalization ability and resulting in poor performance on new data. In the context of this work, the limitation of traditional supervised models was broken by proposing a conditional generation method utilizing unsupervised diffusion model.

Methods: A conditional generation method base on the score-based diffusion model was proposed, which employed diffusion model for the first time to predict the predict patients' therapeutic doses (TherapDose). The proposed diffusion model TherapDose prediction method (DMTP) learns the data distribution of dose images. The data distribution contains the quantitative relationship between the radiotherapy dose (RTDose) derived from the VMAT plan files of the Treatment Planning System (TPS) and the measured Dose (MDose, i.e., TherapDose) obtained from the Dolphin Compass physical system. By sampling from the learnt distribution, efficient prediction of TherapDose was achieved. The training dataset comprises RTDose, and the MDose. The three-dimensional information of dose slice was utilized to predict TherapDose, aiming to enhance the accuracy and efficiency of TherapDose prediction. Root mean square error (RMSE), mean absolute error (MAE), and structural similarity (SSIM) metrics were leveraged to validate the effectiveness of the proposed method. Meanwhile, CT images were further added to test the impacts of CT images on the prediction effect of MDose.

Results: The DMTP method has demonstrated superior performance in predicting TherapDose within key anatomical regions including the head and neck, chest, and abdomen, outperforming existing state-of-the-art methods by achieving high-quality predictions as measured across different evaluation metrics. It indicates that the proposed method is highly effective and accurate in its dose prediction capabilities.

Conclusions: The proposed method has proven to be highly effective, consistently outperforming state-of-the-art techniques in MDose prediction across multiple anatomical regions and evaluation metrics. This method can serve as a clinical aid to assist medical physicists in diminishing the measurement workload associated with prePSQA.