机构:[1]School of Big Data and Software Engineering Chongqing University, Chongqing, China[2] The First Afliated Hospital of Army Medical University, Chongqing, China [3]Department of Radiology, Chongqing University Cancer Hospital, Chongqing, China
Computer-Aided Diagnosis (CAD) systems for lung nodule diagnosis based on deep learning have attracted much attention in recent years. However, most existing methods ignore the relationships between the segmentation and classification tasks, which leads to unstable performances. To address this problem, we propose a novel multi-task framework, which can provide lung nodule segmentation mask, malignancy prediction, and medical features for interpretable diagnosis at the same time. Our framework mainly contains two sub-network: (1) Multi-Channel Segmentation Sub-network (MSN) for lung nodule segmentation, and (2) Joint Classification Sub-network (JCN) for interpretable lung nodule diagnosis. In the proposed framework, we use U-Net down-sampling processes for extracting low-level deep learning features, which are shared by two sub-networks. The JCN forces the down-sampling processes to learn better low-level deep features, which lead to a better construct of segmentation masks. Meanwhile, two additional channels constructed by OTSU and super-pixel (SLIC) methods, are utilized as the guideline of the feature extraction. The proposed framework takes advantages of deep learning methods and classical methods, which can significantly improve the performances of all tasks. We evaluate the proposed framework on public dataset LIDC-IDRI. Our framework achieves a promising Dice score of 86.43% in segmentation, 87.07% in malignancy level prediction, and convincing results in interpretable medical feature predictions.
基金:
National Natural Science Foundation of ChinaNational Natural Science Foundation of China (NSFC) [61772093]; National Key R&D Project of China [2018YFB2101200]; Chongqing Major Theme Projects [cstc2018jszx-cyztzxX0017]
语种:
外文
被引次数:
WOS:
第一作者:
第一作者机构:[1]School of Big Data and Software Engineering Chongqing University, Chongqing, China
通讯作者:
推荐引用方式(GB/T 7714):
Chen Wei,Wang Qiuli,Yang Dan,et al.End-to-End Multi-Task Learning for Lung Nodule Segmentation and Diagnosis[J].2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR).2021,6710-6717.doi:10.1109/ICPR48806.2021.9412218.
APA:
Chen, Wei,Wang, Qiuli,Yang, Dan,Zhang, Xiaohong,Liu, Chen&Li, Yucong.(2021).End-to-End Multi-Task Learning for Lung Nodule Segmentation and Diagnosis.2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR),,
MLA:
Chen, Wei,et al."End-to-End Multi-Task Learning for Lung Nodule Segmentation and Diagnosis".2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR) .(2021):6710-6717