Automatic delineation of craniospinal clinical target volume based on hybrid attention U-net
10.3760/cma.j.cn113030-20210201-00053
- VernacularTitle:基于混合注意力U-net全脑全脊髓临床靶区自动勾画
- Author:
Hongwei LI
1
;
Chunxia NI
;
Shu CHEN
;
Ge MENG
;
Xiaoyang HU
;
Yang WANG
Author Information
1. 上海伽玛医院放疗科,上海 200235
- Keywords:
Deep learning;
Convolutional neural networks;
Automatic segmentation;
Craniospinal clinical target volume
- From:
Chinese Journal of Radiation Oncology
2022;31(3):266-271
- CountryChina
- Language:Chinese
-
Abstract:
Objective:Hybrid attention U-net (HA-U-net) neural network was designed based on U-net for automatic delineation of craniospinal clinical target volume (CTV) and the segmentation results were compared with those of U-net automatic segmentation model.Methods:The data of 110 craniospinal patients were reviewed, Among them, 80 cases were selected for the training set, 10 cases for the validation set and 20 cases for the test set. HA-U-net took U-net as the basic network architecture, double attention module was added at the input of U-net network, and attention gate module was combined in skip-connection to establish the craniospinal automatic delineation network model. The evaluation parameters included Dice similarity coefficient (DSC), Hausdorff distance (HD) and precision.Results:The DSC, HD and precision of HA-U-net network were 0.901±0.041, 2.77±0.29 mm and 0.903±0.038, respectively, which were better than those of U-net (all P<0.05). Conclusion:The results show that HA-U-net convolutional neural network can effectively improve the accuracy of automatic segmentation of craniospinal CTV, and help doctors to improve the work efficiency and the consistent delineation of CTV.