Deep learning-based voxel sampling for particle therapy treatment planning

Objective.Scanned particle therapy often requires complex treatment plans, robust optimization, as well as treatment adaptation. Plan optimization is especially complicated for heavy ions due to the variable relative biological effectiveness. We present a novel deep-learning model to select a subset of voxels in the planning process thus reducing the planning problem size for improved computational efficiency.Approach.Using only a subset of the voxels in target and organs at risk (OARs) we produced high-quality treatment plans, but heuristic selection strategies require manual input. We designed a deep-learning model based onP-Net to obtain an optimal voxel sampling without relying on patient-specific user input. A cohort of 70 head and neck patients that received carbon ion therapy was used for model training (50), validation (10) and testing (10). For training, a total of 12 500 carbon ion plans were optimized, using a highly efficient artificial intelligence (AI) infrastructure implemented into a research treatment planning platform. A custom loss function increased sampling density in underdosed regions, while aiming to reduce the total number of voxels.Main results.On the test dataset, the number of voxels in the optimization could be reduced by 84.8% (median) at <1% median loss in plan quality. When the model was trained to reduce sampling in the target only while keeping all voxels in OARs, a median reduction up to 71.6% was achieved, with 0.5% loss in the plan quality. The optimization time was reduced by a factor of 7.5 for the total AI selection model and a factor of 3.7 for the model with only target selection.Significance.The novel deep-learning voxel sampling technique achieves a significant reduction in computational time with a negligible loss in the plan quality. The reduction in optimization time can be especially useful for future real-time adaptation strategies.