单位:[1]Health Science Center, School of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China[2]National-Regional Key Technology Engineering Laboratory for Medical Ultrasound, Shenzhen University, Shenzhen 518060, China[3]Marshall Laboratory of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China[4]Department of Radiology, Fuxing Hospital, Capital Medical University, Beijing 100038, China[5]Centre for Smart Health, School of Nursing, The Hong Kong Polytechnic University, Hong Kong[6]Department of Radiology, China-Japan Friendship Hospital, Beijing 100029, China[7]AI Research Center for Medical Image Analysis and Diagnosis, Shenzhen University, Shenzhen 518060, China
Automatic gastric tumor segmentation and lymph node (LN) classification not only can assist radiologists in reading images, but also provide image-guided clinical diagnosis and improve diagnosis accuracy. However, due to the inhomogeneous intensity distribution of gastric tumor and LN in CT scans, the ambiguous/missing boundaries, and highly variable shapes of gastric tumor, it is quite challenging to develop an automatic solution. To comprehensively address these challenges, we propose a novel 3D multi-attention guided multi-task learning network for simultaneous gastric tumor segmentation and LN classification, which makes full use of the complementary information extracted from different dimensions, scales, and tasks. Specifically, we tackle task correlation and heterogeneity with the convolutional neural network consisting of scale-aware attention-guided shared feature learning for refined and universal multi-scale features, and task-aware attention-guided feature learning for task-specific discriminative features. This shared feature learning is equipped with two types of scale-aware attention (visual attention and adaptive spatial attention) and two stage-wise deep supervision paths. The task-aware attention-guided feature learning comprises a segmentation-aware attention module and a classification-aware attention module. The proposed 3D multi-task learning network can balance all tasks by combining segmentation and classification loss functions with weight uncertainty. We evaluate our model on an in-house CT images dataset collected from three medical centers. Experimental results demonstrate that our method outperforms the state-of-the-art algorithms, and obtains promising performance for tumor segmentation and LN classification. Moreover, to explore the generalization for other segmentation tasks, we also extend the proposed network to liver tumor segmentation in CT images of the MICCAI 2017 Liver Tumor Segmentation Challenge. Our implementation is released at https://github.com/infinite-tao/MA-MTLN.
基金:
National Natural Science Foundation of ChinaNational Natural Science Foundation of China (NSFC) [81771922, 62071309, 61801305, 62006160, 81971585, 61871274]; National Natural Science Foundation of Guangdong ProvinceNational Natural Science Foundation of Guangdong Province [2019A1515111205]; Shenzhen Key Basic Research Project [JCYJ20170818094109846, CYJ20180507184647636, JCYJ20190808155618806, GJHZ20190822095414576, JCYJ20190808145011259]; SZU Medical Young Scientists Program [71201-000001]
第一作者单位:[1]Health Science Center, School of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China[2]National-Regional Key Technology Engineering Laboratory for Medical Ultrasound, Shenzhen University, Shenzhen 518060, China[3]Marshall Laboratory of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China
通讯作者:
通讯机构:[1]Health Science Center, School of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China[2]National-Regional Key Technology Engineering Laboratory for Medical Ultrasound, Shenzhen University, Shenzhen 518060, China[3]Marshall Laboratory of Biomedical Engineering, Shenzhen University, Shenzhen 518060, China[7]AI Research Center for Medical Image Analysis and Diagnosis, Shenzhen University, Shenzhen 518060, China
推荐引用方式(GB/T 7714):
Zhang Yongtao,Li Haimei,Du Jie,et al.3D Multi-Attention Guided Multi-Task Learning Network for Automatic Gastric Tumor Segmentation and Lymph Node Classification[J].IEEE TRANSACTIONS on MEDICAL IMAGING.2021,40(6):1618-1631.doi:10.1109/TMI.2021.3062902.
APA:
Zhang, Yongtao,Li, Haimei,Du, Jie,Qin, Jing,Wang, Tianfu...&Lei, Baiying.(2021).3D Multi-Attention Guided Multi-Task Learning Network for Automatic Gastric Tumor Segmentation and Lymph Node Classification.IEEE TRANSACTIONS on MEDICAL IMAGING,40,(6)
MLA:
Zhang, Yongtao,et al."3D Multi-Attention Guided Multi-Task Learning Network for Automatic Gastric Tumor Segmentation and Lymph Node Classification".IEEE TRANSACTIONS on MEDICAL IMAGING 40..6(2021):1618-1631