Study on radiation dose distribution based on generative adversarial network
Liao Wentao1, Pu Yuehu2
1University of Chinese Academy of Sciences/Shanghai Institute of Applied Physics,Chinese Academy of Sciences,Shanghai 201800,China; 2Shanghai Advanced Research Institute, Chinese Academy of Sciences, Shanghai 201800,China
Abstract:Objective To investigate whether the combination of the advantages of deep learining in image processing and radiotherapy will make the radiotherapy process more intelligent. Methods The generative adversarial network (GAN) is a generation model using neural network. High-quality dose distribution images can be generated by inputting relevant features. Firstly, random unconditional GAN was utilized to verify the ideal data, then conditional GAN (cGAN) was employed to train DICOMRT data of tumor patients, and the target contour information was used to directly generate dose distribution images. Results For the verification of ideal data, the generation of ideal distribution yielded good effect. By extracting target contour and real dose slice data and using cGAN training, the dose distribution maps of planning target volume (PTV) and organs at risk (OAR) of tumor patients could be obtained. The absolute error evaluation of the maximum and average values between the predicted value and the real dose was shown as[3.57%, 3.37%](PTV),[2.63%, 2.87%](brain),[1.50%, 2.70%](CTV),[3.87%, 1.79%](GTV),[3.60%, 3.23%](OAR1) and[4.40%, 3.13%](OAR2), respectively. Conclusions GAN model can be used to generate ideal dose distribution data, and cGAN model with prior knowledge can be employed to establish the relationship between target information and dose distribution. Directly generating the corresponding dose distribution image by inputting the target contour information is an effective attempt for dose prediction.
Liao Wentao,Pu Yuehu. Study on radiation dose distribution based on generative adversarial network[J]. Chinese Journal of Radiation Oncology, 2021, 30(4): 376-381.
[1] Mcintosh C, Welch M, Mcniven A, et al. Fully automated treatment planning for head and neck radiotherapy using a voxel-based dose prediction and dose mimicking method[J]. Phys Med Biol, 2017, 62(15):5926-5944. DOI:10.1088/1361-6560/AA71F8. [2] Zarepisheh M, Long T, Li N, et al. A DVH-guided IMRT optimization algorithm for automatic treatment planning and adaptive radiotherapy replanning[J]. Med Phys, 2014, 41(6):061711. DOI:10.1118/1.4875700. [3] Krizhevsky A, Sutskever H, Hinton GE. ImageNet classification with deep convolutional neural networks[J]. Commun ACM, 2017, 60(6):84-90. DOI:10.1145/3065386. [4] Goodfellow I, Pouget-Abadie J, Mirza M, et al. Generative adversarial nets[C]. Advances in Neural Information Processing Systems, 2014, 2672-2680. [5] Radford A, Metz L, Chintala S. Unsupervised representation learning with deep convolutional generative adversarial networks[C]. International Conference on Learning Representations, 2015. [6] Mirza M, Simon O. Conditional generative adversarial nets[DB/OL][2019-06-10]. https://arxiv.org/abs/1411.1784. [7] Isola P. Image-to-image translation with conditional adversarial networks[C]. IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017, 5967-5976. DOI:10.1109/CVPR.2017.632. [8] Zhu JY. Unpaired image-to-image translation using cycle-consistent adversarial networks[C]. IEEE International Conference on Computer Vision (ICCV), 2017, 2242-2251. DOI:10.1109/ICCV.2017.244. [9] Ronneberger O, Fischer P, Brox T. U-net:convolutional networks for biomedical image segmentation[C]. International Conference on Medical Image Computing and Computer-Assisted Intervention, 2015, 234-241. DOI:10.1007/978-3-662-54345-0_3. [10] Ioffe S, Christian S. Batch normalization:accelerating deep network training by reducing internal covariate shift[C]. Proceedings of the 32nd International Conference on Machine Learning, 2015, 448-456. [11] Kingma D, Ba JL. Adam:a method for stochastic optimization[C]. ICLR, 2015. [12] Kikinis R, Pieper SD, Vosburgh KG. 3D slicer:a platform for subject-specific image analysis, visualization, and clinical support// Jolesz FA. Intraoperative imaging and image-guided therapy[M]. New York:Springer, 2014. [13] Chen XY, Men K, Li YX, et al. A feasibility study on an automated method to generate patient-specific dose distributions for radiotherapy using deep learning[J]. Med Phys, 2019, 46(1):56-64. DOI:10.1002/mp.13262.