Multitask connected U-Net: automatic lung cancer segmentation from CT images using PET knowledge guidance
Lung cancer is a predominant cause of cancer-related mortality worldwide, necessitating precise tumor segmentation of medical images for accurate diagnosis and treatment. However, the intrinsic complexity and variability of tumor morphology pose substantial challenges to segmentation tasks. To address this issue, we propose a multitask connected U-Net model with a teacher-student framework to enhance the effectiveness of lung tumor segmentation. The proposed model and framework integrate PET knowledge into the segmentation process, leveraging complementary information from both CT and PET modalities to improve segmentation performance. Additionally, we implemented a tumor area detection method to enhance tumor segmentation performance. In extensive experiments on four datasets, the average Dice coefficient of 0.56, obtained using our model, surpassed those of existing methods such as Segformer (0.51), Transformer (0.50), and UctransNet (0.43). These findings validate the efficacy of the proposed method in lung tumor segmentation tasks.
Authors
- Chaoyong Wu (ORCID: https://orcid.org/0000-0003-0692-9075)
- Yiheng Chen (ORCID: https://orcid.org/0009-0009-8726-4589)
- Lu Zhou (ORCID: https://orcid.org/0000-0001-6804-3865)
- Zhicheng Zhang (ORCID: https://orcid.org/0000-0002-5454-9918)
Institutions
- Henan University (CN)
- Beijing University of Chinese Medicine (CN)
- Chinese Academy of Sciences (CN)
- Shenzhen Institutes of Advanced Technology (CN)
Publication Details
- Journal
- Frontiers in Artificial Intelligence
- Published
- 2024-08-23
- DOI
- https://doi.org/10.3389/frai.2024.1423535
- Citations
- 3
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 1.21
Funders
- Henan University of Chinese Medicine
- Henan University