Diffusion-based Radiotherapy Dose Prediction Guided by Inter-slice Aware Structure Encoding

Zhenghao Feng,Lu Wen,Jianghong Xiao,Yuanyuan Xu,Xi Wu,Jiliu Zhou,Xingchen Peng,Yan Wang
2024-07-09
Abstract:Deep learning (DL) has successfully automated dose distribution prediction in radiotherapy planning, enhancing both efficiency and quality. However, existing methods suffer from the over-smoothing problem for their commonly used L1 or L2 loss with posterior average calculations. To alleviate this limitation, we propose a diffusion model-based method (DiffDose) for predicting the radiotherapy dose distribution of cancer patients. Specifically, the DiffDose model contains a forward process and a reverse process. In the forward process, DiffDose transforms dose distribution maps into pure Gaussian noise by gradually adding small noise and a noise predictor is simultaneously trained to estimate the noise added at each timestep. In the reverse process, it removes the noise from the pure Gaussian noise in multiple steps with the well-trained noise predictor and finally outputs the predicted dose distribution maps...
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?