SSAL-Net: Semi-supervised Network Integrating Self-Supervised Adversarial Learning for Diagnosing Subtypes of Pulmonary Nodules

Yu Fu,Changli Liu,Shaoqiang Wang,Enqing Dong,Hui Xia
DOI: https://doi.org/10.1016/j.bspc.2024.106585
IF: 5.1
2024-01-01
Biomedical Signal Processing and Control
Abstract:Accurately diagnosing the subtype of pulmonary nodules is a crucial step in screening patients with early-stage lung cancer. However, most of existing researches can only diagnose benign and malignant pulmonary nodules but cannot give detailed subtypes. Besides, although convolutional neural networks (CNNs) have been widely used in the intelligent diagnosis of benign and malignant pulmonary nodules, it is difficult to further improve the diagnostic accuracy due to the small number of samples. In this paper, a semi-supervised network integrating self-supervised adversarial learning (SSAL-Net) suitable for small sample classification problem is proposed to diagnose adenocarcinoma, squamous cell carcinoma, inflammation, and other benign types of pulmonary nodules. SSAL-Net is mainly composed of a self-supervised classification network (SSC-Net), an unsupervised adversarial reconstruction network (UAR-Net) and a supervised classification network (SC-Net), where SSC-Net and UAR-Net share an encoder to provide representative image features to SC-Net. SSAL-Net takes three different forms of pulmonary nodule data generated based on prior knowledge as input, and enables end-to-end collaborative training through SSC-Net, UAR-Net and SC-Net performing their respective tasks. The experimental results show that SSAL-Net can identify adenocarcinoma, squamous cell carcinoma, inflammation and other benign types of pulmonary nodules with an average accuracy of 89.29% on our self-constructed lung CT image dataset, and can identify malignant pulmonary nodules with an average accuracy of 94.67% on the publicly available LIDC-IDRI dataset, obtaining the state-of-the-art results.
What problem does this paper attempt to address?