Lung cancer is the leading cause of cancer-related deaths globally, with early and accurate detection being crucial for patient survival. While computed tomography (CT) scans are instrumental in identifying pulmonary nodules, current deep learning models typically address segmentation and malignancy classification as separate tasks, potentially overlooking the mutual benefits of joint optimization. In this work, we propose MTL-LungNet, a novel multi-task learning framework that simultaneously performs nodule segmentation and malignancy classification using CT data. By leveraging shared representations and a segmentation-guided classification strategy, our model captures both spatial precision and semantic context, enhancing diagnostic accuracy. We validate our approach on the LIDC-IDRI dataset, employing standardized preprocessing techniques and balanced data splits. The proposed architecture integrates SwinUNETR for segmentation and a ResNet-based classifier with an attention-like mechanism that emphasizes clinically relevant regions. Experimental results demonstrate that MTL-LungNet achieves better segmentation performance than single-task baselines and provides malignancy prediction capabilities, emphasizing the promise of multi-task learning for robust, interpretable, and efficient lung cancer diagnosis.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

MTL-LungNet: A Unified Model for Nodule Segmentation and Malignancy Classification

  • Mihai Nan,
  • Cătălin–Mihail Chiru,
  • Adina Magda Florea

摘要

Lung cancer is the leading cause of cancer-related deaths globally, with early and accurate detection being crucial for patient survival. While computed tomography (CT) scans are instrumental in identifying pulmonary nodules, current deep learning models typically address segmentation and malignancy classification as separate tasks, potentially overlooking the mutual benefits of joint optimization. In this work, we propose MTL-LungNet, a novel multi-task learning framework that simultaneously performs nodule segmentation and malignancy classification using CT data. By leveraging shared representations and a segmentation-guided classification strategy, our model captures both spatial precision and semantic context, enhancing diagnostic accuracy. We validate our approach on the LIDC-IDRI dataset, employing standardized preprocessing techniques and balanced data splits. The proposed architecture integrates SwinUNETR for segmentation and a ResNet-based classifier with an attention-like mechanism that emphasizes clinically relevant regions. Experimental results demonstrate that MTL-LungNet achieves better segmentation performance than single-task baselines and provides malignancy prediction capabilities, emphasizing the promise of multi-task learning for robust, interpretable, and efficient lung cancer diagnosis.