Physics·Biology·Technique
Multi-task learning-based three-dimensional dose distribution prediction for multiple organs in a single model
Futong Guo, Yongbao Li, Qiyuan Jia, Mengke Qi, Aiqian Wu, Fantu Kong, Yanhua Mai, Ting Song, Linghong Zhou
Published 2019-06-15
Cite as Chin J Radiat Oncol, 2019, 28(6): 432-437. DOI: 10.3760/cma.j.issn.1004-4221.2019.06.008
Abstract
ObjectiveTo establish a three-dimensional (3D) dose prediction model, which can predict multiple organs simultaneously in a single model and automatically learn the effect of the geometric anatomical structure on dose distribution.
MethodsClinical radiotherapy plans of patients diagnosed with the same type of tumors were collected and retrospectively analyzed. For every plan, each organs at risk (OAR) voxel was regarded as the study sample and its deposited dose was considered as the dosimetric feature. A regularized multi-task learning method than could learn the relationship among different tasks was employed to establish the relationship matrix among tasks and the correlation between geometric structure and dose distribution among organs. In this experiment, the spinal cord, brainstem and bilateral parotids involved in the intensity-modulated radiotherapy (IMRT) plan of 15 nasopharyngeal cancer patients were utilized to establish the multi-organ prediction model. The relative percentage error between the predicted dose of voxel and the clinical planning dose was calculated to assess the feasibility of the model.
ResultsTen cases receiving IMRT plans were utilized as the training data, and the remaining five cases were used as the test data. The test results demonstrated a higher prediction accuracy and less data demand. And the average voxel dose errors among the spinal cord, brainstem and the left and right parotids were (2.01±0.02)%, (2.65±0.02)%, (2.45±0.02)% and (2.55±0.02)%, respectively.
ConclusionThe proposed model can accurately predict the dose of multiple organs in a single model and avoid the establishment of multiple single-organ prediction models, laying a solid foundation for patient-specific plan quality control and knowledge-based treatment planning.
Key words:
Multi-task learning; Three-dimensional dose distribution prediction; Correlation of multi-organ in a single-model; Intensity-modulated radiation therapy
Contributor Information
Futong Guo
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Yongbao Li
Department of Radiation Oncology, Sun Yat-sen University Cancer Center, Guangzhou 510060, China
Qiyuan Jia
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Mengke Qi
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Aiqian Wu
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Fantu Kong
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Yanhua Mai
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Ting Song
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China
Linghong Zhou
Department of Biomedical Engineering, Southern Medical University, Guangzhou 510515, China