Machine Intelligence Laboratory, College of Computer Science, Sichuan University, Chengdu 610065, PR China.
Department of Radiation Oncology, West China Hospital, Sichuan University, Chengdu 610041, PR China.
Med Image Anal. 2021 Jan;67:101886. doi: 10.1016/j.media.2020.101886. Epub 2020 Oct 26.
As the main treatment for cancer patients, radiotherapy has achieved enormous advancement over recent decades. However, these achievements have come at the cost of increased treatment plan complexity, necessitating high levels of expertise experience and effort. The accurate prediction of dose distribution would alleviate the above issues. Deep convolutional neural networks are known to be effective models for such prediction tasks. Most studies on dose prediction have attempted to modify the network architecture to accommodate the requirement of different diseases. In this paper, we focus on the input and output of dose prediction model, rather than the network architecture. Regarding the input, the non-modulated dose distribution, which is the initial quantity in the inverse optimization of the treatment plan, is used to provide auxiliary information for the prediction task. Regarding the output, a historical sub-optimal ensemble (HSE) method is proposed, which leverages the sub-optimal models during the training phase to improve the prediction results. The proposed HSE is a general method that does not require any modification of the learning algorithm and does not incur additional computational cost during the training phase. Multiple experiments, including the dose prediction, segmentation, and classification tasks, demonstrate the effectiveness of the strategies applied to the input and output parts.
作为癌症患者的主要治疗方法,放射治疗在最近几十年取得了巨大的进展。然而,这些成就的取得是以增加治疗计划的复杂性为代价的,这需要高水平的专业知识和经验。准确预测剂量分布将缓解上述问题。众所周知,深度卷积神经网络是用于此类预测任务的有效模型。大多数关于剂量预测的研究都试图修改网络架构以适应不同疾病的要求。在本文中,我们专注于剂量预测模型的输入和输出,而不是网络架构。关于输入,使用非调制剂量分布(治疗计划逆优化中的初始量)为预测任务提供辅助信息。关于输出,提出了一种历史次优集合(HSE)方法,该方法利用训练阶段的次优模型来提高预测结果。所提出的 HSE 是一种通用方法,不需要修改学习算法,并且在训练阶段不会增加额外的计算成本。多个实验,包括剂量预测、分割和分类任务,证明了应用于输入和输出部分的策略的有效性。