arrow
返回

Multi-stage framework with difficulty-aware learning for progressive dose prediction

delete2023-04-01
delete5
PRE
AI
F
Fan Li
S
Sijie Niu *
Y
Yingying Han
Y
Yan Zhang
Z
Zihao Dong
J
Jian Zhu
DOI:10.1016/j.bspc.2022.104541delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Purpose: In this study, we are concerned with overcoming the limitation that most existing methods learn the dose prediction distribution without taking into account hard-to-predict regions. Methods: This paper presents a novel multi-stage dose prediction framework based on difficulty-aware learning to progressively predict 3D radiation dose distribution. We first stack multiple sub-models employing an encoder-decoder architecture to construct a multi-stage framework that decomposes the challenging prediction task into a few easy-to-learn sub-tasks for improving dose prediction. Moreover, a difficulty-aware learning strategy with voxel-level attention is developed to effectively handle hard-to-predict voxels or regions. Finally, the performance of our framework is compared to that of existing related models while being evaluated by voxel-based mean absolute error (MAE) and clinical-related dosimetric metrics. Results: We evaluated the proposed method on the head-and-neck datasets from the 2020 AAPM OpenKBP challenge. Our framework achieved a dose score of 2.367 and a DVH score of 1.378, which outperformed the other existing single-stage and cascade dose prediction models. The average prediction error of the dose CI and dose HI in the test dataset of 56 Gy, 63 Gy, and 70 Gy were quite similar to the ground-truth plans (ranging from 0.01 to 0.02). For the OARs, the average error in predicting the Dmax value was 2.66 +/- 2.88 Gy (brainstem), 1.94 +/- 1.43 Gy (spinal cord), and 1.39 +/- 1.67 Gy (mandible), whereas the values for the Dmean value were 1.10 +/- 1.07 Gy (left parotid), 1.13 +/- 1.03 Gy (right parotid), 1.81 +/- 1.77 Gy (esophagus), and 1.49 +/- 1.66 Gy (larynx).Conclusion: Both quantitative and qualitative results demonstrated the potential of our framework and its superior performance compared to the state-of-the-art methods in the dose prediction task, which thus proved to be a promising tool for predicting the dose distribution in computer-aided radiotherapy planning.
Keyword:
Radiation therapy
Dose prediction
Convolutional neural network
Difficulty-aware learning

期刊

Biomedical Signal Processing and Control 封面图
Biomedical Signal Processing and Control
IF:
4.9
论文数:
1.0W
被引数:
2.4W

机构

U
University of Jinan
学者数:
1.6W
论文数: 1.1W
被引数: 1.4W
N
Nanjing Medical University
学者数:
4.7W
论文数: 2.8W
被引数: 3.5K
引用论文

引用论文

Dose prediction with deep learning for prostate cancer radiation therapy: Model adaptation to different treatment planning practices前列腺癌放射治疗的深度学习剂量预测: 模型适应不同的治疗计划实践
err2020-12-01
err58
errOAAI
errKandalan, Roya Norouzi; Nguyen, Dan; Rezaeian, Nima Hassan; Barragan-Montero, Ana M.; Breedveld, Sebastiaan; Namuduri, Kamesh; Jiang, Steve; Lin, Mu-Han
err分享
err收藏
Multi-constraint generative adversarial network for dose prediction in radiotherapy多约束生成对抗网络在放射治疗剂量预测中的应用
err2022-04-01
err70
PREAI
errZhan, Bo; Xiao, Jianghong; Cao, Chongyang; Peng, Xingchen; Zu, Chen; Zhou, Jiliu; Wang, Yan
err分享
err收藏
Dose prediction via distance-guided deep learning: Initial development for nasopharyngeal carcinoma radiotherapy
err2022-05-01
err24
PREAI
errYue, Meiyan; Xue, Xiaoguang; Wang, Zhanyu; Lambo, Ricardo Lewis; Zhao, Wei; Xie, Yaoqin; Cai, Jing; Qin, Wenjian
err分享
err收藏
A deep learning method for prediction of three-dimensional dose distribution of helical tomotherapy
err2019-03-30
err100
PREAI
errLiu, Zhiqiang; Fan, Jiawei; Li, Minghui; Yan, Hui; Hu, Zhihui; Huang, Peng; Tian, Yuan; Miao, Junjie; Dai, Jianrong
err分享
err收藏
Difficulty-aware hierarchical convolutional neural networks for deformable registration of brain MR images
err2021-01-01
err22
errOAAI
errHuang, Yunzhi; Ahmad, Sahar; Fan, Jingfan; Shen, Dinggang; Yap, Pew-Thian
err分享
err收藏
Dose prediction using a deep neural network for accelerated planning of rectal cancer radiotherapy
err2020-08-01
err67
PREAI
errSong, Ying; Hu, Junjie; Liu, Yang; Hu, Haiyun; Huang, Yang; Bai, Sen; Yi, Zhang
err分享
err收藏
err分享
err收藏
学者 查看更多内容