Abstract
Lung cancer has long posed a severe threat to human life and health, and early detection as well as effective treatment can significantly improve the survival rates of patients. In recent years, deep learning has greatly propelled the advancement in image processing, yet its application in pulmonary nodule detection through deep learning is still at an early stage. Therefore, researching automatic pulmonary nodule detection technology based on deep learning, enhancing the performance of lung nodule detection, holds significant importance for improving the survival rates among lung cancer patients. This paper aims to address the above issues and propose a semi-supervised pulmonary nodule detection algorithm based on adversarial learning. A multi-task classification network is designed to distinguish between annotated and unannotated candidate pulmonary nodules while removing false-positive pulmonary nodules. The adversarial loss and maximum-minimum strategy are introduced into the training process, and the unannotated images and adversarial loss are used to update the 3D sliding window semantic segmentation network. To enable the adversarial loss to backpropagate, the algorithm makes some improvements to RoI pooling, connecting the 3D sliding window semantic segmentation network and multi-task classification network for end-to-end learning. In experiments, compared to the supervised lung nodule algorithm using only 50% labeled images, the semi-supervised lung nodule detection algorithm using 50% labeled images and 50% unlabeled images increased the average FROC from 0.677 to 0.717.
Similar content being viewed by others
Data availability
The data that support the findings of this study are available from the corresponding author upon reasonable request.
References
Miller KD, Nogueira L, Mariotto AB et al (2019) Cancer treatment and survivorship statistics, 2019[J]. CA: A Cancer J Clin 69(5):363–385
Alex K, Ilya S, Hinton GE (2017) ImageNet classification with deep convolutional neural networks[J]. Commun ACM 60(6):84–90
Park J, Yun J, Kim N et al (2019) Fully automated lung lobe segmentation in volumetric Chest CT with 3D U-Net: validation with intra- and extra-datasets[J]. J Digit Imaging 33(2). https://doi.org/10.1007/s10278-019-00223-1
Zhang Z, Wu C, Coleman S et al (2020) DENSE-INception U-net for medical image segmentation[J]. Comput Methods Programs Biomed 192:105395
Dong X, Lei Y, Wang T et al (2019) Automatic multiorgan segmentation in thorax CT images using U-Net-GAN[J]. Med Phys 46(5):2157–2168. https://doi.org/10.1002/mp.13458
Skourt BA, Hassani AE, Majda A (2018) Lung CT image segmentation using deep neural networks[J]. Procedia Comput Sci 127:109–113
Milletari F, Navab N, Ahmadi SA (2016) V-Net: fully convolutional neural networks for volumetric medical image segmentation[C]. IEEE. https://doi.org/10.1109/3DV.2016.79
Dutande P, Baid U, Talbar S (2021) LNCDS: A 2D–3D cascaded CNN approach for lung nodule classification, detection and segmentation[J]. Biomed Signal Process Control 67(6):102527
Lee S, Hussein R, Mckeown MJ (2019) A deep convolutional-recurrent neural network architecture for parkinson’s disease EEG classification[C]. IEEE. https://doi.org/10.1109/GlobalSIP45357.2019.8969309
Wang R, Sun H, Zhang J et al (2019) A transfer learning method for ct image classification of pulmonary nodules[J]. https://doi.org/10.1007/978-3-030-19156-6_15
Ramachandran S, George J, Skaria S et al (2018) Using YOLO based deep learning network for real time detection and localization of lung nodules from low dose CT scans[J]. https://doi.org/10.1117/12.2293699
Hx A, Dyab C, Ns C et al (2019) Automated pulmonary nodule detection in CT images using deepconvolutional neural networks[J]. Pattern Recogn 85:109–119
Zhu W, Liu C, Fan W et al (2017) DeepLung: 3D deep convolutional nets for automated pulmonary nodule detection and classification[J]. https://doi.org/10.1101/189928
Zheng S, Cornelissen LJ, Cui X et al (2021) Deep convolutional neural networks for multiplanar lung nodule detection: Improvement in small nodule identification[J]. Med Phys 48(2):733–744
Zhao C, Han J, Jia Y et al (2018) Lung nodule detection via 3D u-net and contextual convolutional neural network[C]. 2018 International conference on networking and network applications (NaNA). IEEE Comput Soc 356–361. https://doi.org/10.1109/NANA.2018.8648753
He Q, Gao P, Zhang F et al (2024) Healthcare entity recognition based on deep learning. Multimed Tools Appl 83:32739–32763. https://doi.org/10.1007/s11042-023-16900-x
Wang CY, Liao H, Wu YH et al (2020) CSPNet: a new backbone that can enhance learning capability of CNN[C]. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW). IEEE. https://doi.org/10.1109/CVPRW50498.2020.00203
Carion N, Massa F, Synnaeve G et al (2020) End-to-end object detection with transformers[J]. https://doi.org/10.1007/978-3-030-58452-8_13
Liu Z, Lin Y, Cao Y et al (2021) Swin transformer: hierarchical vision transformer using shifted windows[J]. https://doi.org/10.48550/arXiv.2103.14030
Liu J, Jiang H, He C et al (2017) An assisted diagnosis system for detection of early pulmonarynodule in computed tomography images[J]. J Med Syst 41(2):30
Zhang J, Xia Y, Cui H et al (2018) Pulmonary nodule detection in medical images: a survey[J]. Biomed Signal Process Control 43:138–147
Arjovsky M, Bottou L (2017) Towards principled methods for training generative adversarial networks. arXiv:1701.04862
Arjovsky M, Chintala S, Bottou L (2017) Wasserstein generative adversarial networks[J]. PMLR. https://doi.org/10.1088/1742-6596/2586/1/012157
Zhu W, Liu C, Fan W, Xie X (2018) Deeplung: deep 3D dual path nets for automated pulmonary nodule detection and classification. arXiv:1801.09555
Wu B, Zhou Z, Wang J, Wang Y (2018) Joint learning for pulmonary nodule segmentation, attributes and malignancy prediction[C]. In: 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018). IEEE. https://doi.org/10.1109/ISBI.2018.8363765
Khosravan N, Bagci U (2018) Semi-supervised multi-task learning for lung cancer diagnosis. arXiv:1802.06181
Acknowledgements
This work is supported by the National Natural Science Foundation of China (61872284); Key Research and Development Program of Shaanxi(2023-YBGY-203,2023-YBGY-021); Industrialization Project of Shaanxi Provincial Department of Education (21JC017); "Thirteenth Five-Year" National Key R&D Program Project (Project Number: 2019YFD1100901); Natural Science Foundation of Shannxi Province, China(2021JLM-16); Key R&D Plan of Xianyang City (L2023-ZDYF-QYCX-021).
Author information
Authors and Affiliations
Corresponding authors
Ethics declarations
Conflict of interest
The authors have no conflicts to disclose.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
He, Q., Gao, P., Zhang, F. et al. Semi-supervised lung nodule detection with adversarial learning. Multimed Tools Appl (2024). https://doi.org/10.1007/s11042-024-19074-2
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11042-024-19074-2