A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy

Purpose: Deep learning (DL) is widely used in dose prediction for radiation oncology, multiple DL techniques comparison is often lacking in the literature. To compare the performance of 4 state-of-the-art DL models in predicting the voxel-level dose distribution for cervical cancer volumetric modula...

Full description

Bibliographic Details
Main Authors: Zhe Wu PhD, Mujun Liu PhD, Ya Pang BS, Lihua Deng PhD, Yi Yang MS, Yi Wu PhD
Format: Article
Language:English
Published: SAGE Publishing 2024-04-01
Series:Technology in Cancer Research & Treatment
Online Access:https://doi.org/10.1177/15330338241242654
_version_ 1797215537955078144
author Zhe Wu PhD
Mujun Liu PhD
Ya Pang BS
Lihua Deng PhD
Yi Yang MS
Yi Wu PhD
author_facet Zhe Wu PhD
Mujun Liu PhD
Ya Pang BS
Lihua Deng PhD
Yi Yang MS
Yi Wu PhD
author_sort Zhe Wu PhD
collection DOAJ
description Purpose: Deep learning (DL) is widely used in dose prediction for radiation oncology, multiple DL techniques comparison is often lacking in the literature. To compare the performance of 4 state-of-the-art DL models in predicting the voxel-level dose distribution for cervical cancer volumetric modulated arc therapy (VMAT). Methods and Materials: A total of 261 patients’ plans for cervical cancer were retrieved in this retrospective study. A three-channel feature map, consisting of a planning target volume (PTV) mask, organs at risk (OARs) mask, and CT image was fed into the three-dimensional (3D) U-Net and its 3 variants models. The data set was randomly divided into 80% as training-validation and 20% as testing set, respectively. The model performance was evaluated on the 52 testing patients by comparing the generated dose distributions against the clinical approved ground truth (GT) using mean absolute error (MAE), dose map difference (GT-predicted), clinical dosimetric indices, and dice similarity coefficients (DSC). Results: The 3D U-Net and its 3 variants DL models exhibited promising performance with a maximum MAE within the PTV 0.83% ± 0.67% in the UNETR model. The maximum MAE among the OARs is the left femoral head, which reached 6.95% ± 6.55%. For the body, the maximum MAE was observed in UNETR, which is 1.19 ± 0.86%, and the minimum MAE was 0.94 ± 0.85% for 3D U-Net. The average error of the Dmean difference for different OARs is within 2.5 Gy. The average error of V40 difference for the bladder and rectum is about 5%. The mean DSC under different isodose volumes was above 90%. Conclusions: DL models can predict the voxel-level dose distribution accurately for cervical cancer VMAT treatment plans. All models demonstrated almost analogous performance for voxel-wise dose prediction maps. Considering all voxels within the body, 3D U-Net showed the best performance. The state-of-the-art DL models are of great significance for further clinical applications of cervical cancer VMAT.
first_indexed 2024-04-24T11:31:39Z
format Article
id doaj.art-0e99510d4f8745cb9f763b2033f748fd
institution Directory Open Access Journal
issn 1533-0338
language English
last_indexed 2024-04-24T11:31:39Z
publishDate 2024-04-01
publisher SAGE Publishing
record_format Article
series Technology in Cancer Research & Treatment
spelling doaj.art-0e99510d4f8745cb9f763b2033f748fd2024-04-10T09:05:05ZengSAGE PublishingTechnology in Cancer Research & Treatment1533-03382024-04-012310.1177/15330338241242654A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc TherapyZhe Wu PhD0Mujun Liu PhD1Ya Pang BS2Lihua Deng PhD3Yi Yang MS4Yi Wu PhD5 Department of Radiation Oncology, , Zigong First People's Hospital, Zigong, Sichuan, China Department of Digital Medicine, School of Biomedical Engineering and Medical Imaging, , Chongqing, China Department of Radiation Oncology, , Zigong First People's Hospital, Zigong, Sichuan, China Department of Radiology, The First Affiliated Hospital of the , Chongqing, China Department of Digital Medicine, School of Biomedical Engineering and Medical Imaging, , Chongqing, China Department of Digital Medicine, School of Biomedical Engineering and Medical Imaging, , Chongqing, ChinaPurpose: Deep learning (DL) is widely used in dose prediction for radiation oncology, multiple DL techniques comparison is often lacking in the literature. To compare the performance of 4 state-of-the-art DL models in predicting the voxel-level dose distribution for cervical cancer volumetric modulated arc therapy (VMAT). Methods and Materials: A total of 261 patients’ plans for cervical cancer were retrieved in this retrospective study. A three-channel feature map, consisting of a planning target volume (PTV) mask, organs at risk (OARs) mask, and CT image was fed into the three-dimensional (3D) U-Net and its 3 variants models. The data set was randomly divided into 80% as training-validation and 20% as testing set, respectively. The model performance was evaluated on the 52 testing patients by comparing the generated dose distributions against the clinical approved ground truth (GT) using mean absolute error (MAE), dose map difference (GT-predicted), clinical dosimetric indices, and dice similarity coefficients (DSC). Results: The 3D U-Net and its 3 variants DL models exhibited promising performance with a maximum MAE within the PTV 0.83% ± 0.67% in the UNETR model. The maximum MAE among the OARs is the left femoral head, which reached 6.95% ± 6.55%. For the body, the maximum MAE was observed in UNETR, which is 1.19 ± 0.86%, and the minimum MAE was 0.94 ± 0.85% for 3D U-Net. The average error of the Dmean difference for different OARs is within 2.5 Gy. The average error of V40 difference for the bladder and rectum is about 5%. The mean DSC under different isodose volumes was above 90%. Conclusions: DL models can predict the voxel-level dose distribution accurately for cervical cancer VMAT treatment plans. All models demonstrated almost analogous performance for voxel-wise dose prediction maps. Considering all voxels within the body, 3D U-Net showed the best performance. The state-of-the-art DL models are of great significance for further clinical applications of cervical cancer VMAT.https://doi.org/10.1177/15330338241242654
spellingShingle Zhe Wu PhD
Mujun Liu PhD
Ya Pang BS
Lihua Deng PhD
Yi Yang MS
Yi Wu PhD
A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
Technology in Cancer Research & Treatment
title A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
title_full A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
title_fullStr A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
title_full_unstemmed A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
title_short A Comparative Study of Deep Learning Dose Prediction Models for Cervical Cancer Volumetric Modulated Arc Therapy
title_sort comparative study of deep learning dose prediction models for cervical cancer volumetric modulated arc therapy
url https://doi.org/10.1177/15330338241242654
work_keys_str_mv AT zhewuphd acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT mujunliuphd acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yapangbs acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT lihuadengphd acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yiyangms acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yiwuphd acomparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT zhewuphd comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT mujunliuphd comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yapangbs comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT lihuadengphd comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yiyangms comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy
AT yiwuphd comparativestudyofdeeplearningdosepredictionmodelsforcervicalcancervolumetricmodulatedarctherapy