Probabilistic feature extraction, dose statistic prediction and dose mimicking for automated radiation therapy treatment planning

Abstract

Purpose: We propose a general framework for quantifying predictive uncertainties of dose-related quantities and leveraging this information in a dose mimicking problem in the context of automated radiation therapy treatment planning. Methods: A three-step pipeline, comprising feature extraction, dose statistic prediction and dose mimicking, is employed. In particular, the features are produced by a convolutional variational autoencoder and used as inputs in a previously developed nonparametric Bayesian statistical method, estimating the multivariate predictive distribution of a collection of predefined dose statistics. Specially developed objective functions are then used to construct a probabilistic dose mimicking problem based on the produced distributions, creating deliverable treatment plans. Results: The numerical experiments are performed using a dataset of 94 retrospective treatment plans of prostate cancer patients. We show that the features extracted by the variational autoencoder capture geometric information of substantial relevance to the dose statistic prediction problem and are related to dose statistics in a more regularized fashion than hand-crafted features. The estimated predictive distributions are reasonable and outperforms a non-input-dependent benchmark method, and the deliverable plans produced by the probabilistic dose mimicking agree better with their clinical counterparts than for a non-probabilistic formulation. Conclusions: We demonstrate that prediction of dose-related quantities may be extended to include uncertainty estimation and that such probabilistic information may be leveraged in a dose mimicking problem. The treatment plans produced by the proposed pipeline resemble their original counterparts well, illustrating the merits of a holistic approach to automated planning based on probabilistic modeling.

References in corpus (11)

• Adam: A Method for Stochastic Optimization
• Auto-Encoding Variational Bayes
• Variational Inference: A Review for Statisticians
• Deep Learning in Medical Image Registration: A Review
• Fully Automated Treatment Planning for Head and Neck Radiotherapy using a Voxel-Based Dose Prediction and Dose Mimicking Method
• Incorporating human and learned domain knowledge into training deep neural networks: A differentiable dose volume histogram and adversarial inspired framework for generating Pareto optimal dose distributions in radiation therapy
• Neural network dose models for knowledge-based planning in pancreatic SBRT
• Probabilistic dose prediction using mixture density networks for automated radiation therapy treatment planning
• Direct optimization of dose-volume histogram metrics in radiation therapy treatment planning
• A comparison of Monte Carlo dropout and bootstrap aggregation on the performance and uncertainty estimation in radiation therapy dose prediction with deep learning neural networks
• A similarity-based Bayesian mixture-of-experts model

Cited by in corpus (4)

• Robust automated radiation therapy treatment planning using scenario-specific dose prediction and robust dose mimicking
• Probabilistic Pareto plan generation for semiautomated multicriteria radiation therapy treatment planning
• Deep Evidential Learning for Radiotherapy Dose Prediction
• OpenKBP-Opt: An international and reproducible evaluation of 76 knowledge-based planning pipelines