Abstract:Abdominal multi-organ segmentation of medical images is essential for clinical applications such as surgical treatment planning and assisted diagnosis. Most published medical image datasets label partial organs only, which is difficult for accurate multiple organs segmentation of medical images, thus the segmentation model developed by this approach is not generally applicable. In this paper, we proposed a joint-dataset-based multi-organ segmentation abdominal network: C2F-MSNet, which contained coarse segment and fine segment. During coarse segmentation, the explicit conditional control module was employed for the training of the network on multiple partially labeled datasets, while the self-attention module and the deep supervision strategy were implemented. During the fine segmentation, the fine segmentation area was indexed by rough segmentation results, and the fine segmentation was guided and the multi-scale fine segmentation network was constructed. Experiments were performed on 663 CT data obtained from KiTS, Decathlon-liver, Decathlon-spleen and Decathlon-pancreas datasets, evaluated by dice similarity coefficient (DSC) and Hausdorff distance (HD). The results of DSC reached 0.967, 0.964, 0.956, and 0.838 for kidney, liver, spleen, and pancreas, respectively, and that of HD reached 12.51, 25.02, 6.68, and 12.58, respectively. Experiment results showed that the C2F-MSNet effectively solved the training problem of multiple partially labeled datasets and achieved accurate multi-organ segmentation in the joint datasets.
[1] 胡玉进,雷柏英,郭力宝,等. 基于BiSeNet的小儿超声心动图左心分割方法[J]. 中国生物医学工程学报, 2019, 38(5): 533-539. [2] 韩阳,宋金淼,薛安懿,等. 基于三重注意力的脑肿瘤图像分割网络[J]. 中国生物医学工程学报, 2022, 41(1): 57-63. [3] Chu Chengwen, Oda M, Kitasaka T, et al. Multi-organ segmentation based on spatially-divided probabilistic atlas from 3D abdominal CT images[C]//Proceedings of 16th International Conference on Medical Image Computing and Computer-assisted Intervention. Nagoya: Springer, 2013: 165-172. [4] Wolz R, Chu Chengwen, Misawa K, et al. Automated abdominal multi-organ segmentation with subject-specific atlas generation[J]. IEEE Transactions on Medical Imaging, 2013, 32(9): 1723-1730. [5] Tong Tong, Wolz R, Wang Zehan, et al. Discriminative dictionary learning for abdominal multi-organ segmentation[J]. Medical Image Analysis, 2015, 23(1): 92-104. [6] Okada T, Linguraru MG, Hori M, et al. Abdominal multi-organ segmentation from CT images using conditional shape-location and unsupervised intensity priors[J]. Medical Image Analysis, 2015, 26(1): 1-18. [7] Iglesias JE, Sabuncu MR. Multi-atlas segmentation of biomedical images: a survey[J]. Medical Image Analysis, 2015, 24(1): 205-219. [8] Zhou Yuyin, Li Zhe, Bai Song, et al. Prior-aware neural network for partially-supervised multi-organ segmentation[C]//Proceedings of the 19th IEEE/CVF International Conference on Computer Vision. Seoul: IEEE, 2019: 10672-10681. [9] Fang Xi, Yan Pingkun. Multi-organ segmentation over partially labeled datasets with multi-scale feature abstraction[J]. IEEE Transactions on Medical Imaging, 2020, 39(11): 3619-3629. [10] Wang Yan, Zhou Yuyin, Shen Wei, et al. Abdominal multi-organ segmentation with organ-attention networks and statistical fusion[J]. Medical Image Analysis, 2019, 55(1): 88-102. [11] Huang Rui, Zheng Yuanjie, Hu Zhiqiang, et al. Multi-organ segmentation via co-training weight-averaged models from few-organ datasets[C]//Proceedings of the 23rd International Conference on Medical Image Computing and Computer-Assisted Intervention. Lima: Springer, 2020: 146-155. [12] Dmitriev K, Kaufman A E. Learning multi-class segmentations from single-class datasets[C]//Proceedings of the 19th IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach: IEEE, 2019: 9501-9511. [13] Heller N. The kits19 challenge data: 300 kidney tumor cases with clinical context, CT semantic segmentations, and surgical outcomes[EB/OL]. https://arxiv.org/abs/1904.00445, 2020-03-15/2022-05-24. [14] Simpson AL. A large annotated medical image dataset for the development and evaluation of segmentation algorithms[EB/OL]. https://arxiv.org/abs/1902.09063, 2019-02-25/2022-05-24. [15] Lee C, Xie S, Gallagher P, et al. Deeply-supervised nets[C]//Proceedings of the 18th International Conference on Artificial Intelligence and Statistics. Cadiz: PMLR, 2015: 562-570. [16] Woo S, Park J, Lee J, et al. CBAM: convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision (ECCV). San Jose: ECCV, 2018: 3-19. [17] Loshchilov I. Sgdr: Stochastic gradient descent with warm restarts[EB/OL]. https://arxiv.org/abs/1608.03983, 2017-05-03/2022-05-24. [18] Çiçek Ö, Abdulkadir A, Lienkamp S S, et al. 3D U-Net: learning dense volumetric segmentation from sparse annotation[C]//Proceedings of 19th International Conference on Medical Image Computing and Computer-assisted Intervention. Athens: Springer,2016:424-432. [19] Hu Jie, Shen Li, Sun Gang, et al. Squeeze-and-excitation networks[J]. IEEE Transaction on Pattern Analysis and Machine Intelligence, 2018, 42(8): 7132-7141. [20] Skorokhodov I, Ignatyev S, Elhoseiny M. Adversarial generation of continuous images[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Montreal: IEEE, 2021: 10753-10764. [21] Baek K, Choi Y, Uh Y, et al. Rethinking the truly unsupervised image-to-image translation[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision. Montreal: IEEE, 2021: 14154-14163. [22] 吉淑滢,肖志勇. 融合上下文和多尺度特征的胸部多器官分割[J]. 中国图象图形学报, 2021, 26(9): 2135-2145. [23] Mishra S, Zhang Yizhe, Chen DZ, et al. Data-driven deep supervision for medical image segmentation[J]. IEEE Transactions on Medical Imaging, 2022, 41(6): 1560-1574. [24] Nelms BE, Tomé WA, Robinson G, et al. Variations in the contouring of organs at risk: test case from a patient with oropharyngeal cancer[J]. International Journal of Radiation Oncology Biology Physics, 2012, 82(1): 368-378. [25] Peng Yinglin, Chen Li, Shen Guanzhu, et al. Interobserver variations in the delineation of target volumes and organs at risk and their impact on dose distribution in intensity-modulated radiation therapy for nasopharyngeal carcinoma[J]. Oral Oncology, 2018, 82(1): 1-7.